diff --git a/pytential/qbx/interactions.py b/pytential/qbx/interactions.py index 0d78dcdb2d4562fbc5f1fc515434bdb9a6eea4ca..6105472db63a1aad41256798880ecef91e852c49 100644 --- a/pytential/qbx/interactions.py +++ b/pytential/qbx/interactions.py @@ -112,9 +112,9 @@ class P2QBXLFromCSR(P2EBase): """], arguments, name=self.name, assumptions="ntgt_centers>=1", - silenced_warnings="write_race(write_expn*)") + silenced_warnings="write_race(write_expn*)", + fixed_parameters=dict(dim=self.dim)) - loopy_knl = lp.fix_parameters(loopy_knl, dim=self.dim) loopy_knl = self.expansion.prepare_loopy_kernel(loopy_knl) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr") @@ -208,9 +208,8 @@ class M2QBXL(E2EBase): "..." ] + gather_loopy_arguments([self.src_expansion, self.tgt_expansion]), name=self.name, assumptions="ncenters>=1", - silenced_warnings="write_race(write_expn*)") - - loopy_knl = lp.fix_parameters(loopy_knl, dim=self.dim) + silenced_warnings="write_race(write_expn*)", + fixed_parameters=dict(dim=self.dim)) for expn in [self.src_expansion, self.tgt_expansion]: loopy_knl = expn.prepare_loopy_kernel(loopy_knl) @@ -309,11 +308,8 @@ class L2QBXL(E2EBase): ] + gather_loopy_arguments([self.src_expansion, self.tgt_expansion]), name=self.name, assumptions="ncenters>=1", - silenced_warnings="write_race(write_expn*)") - - loopy_knl = lp.fix_parameters(loopy_knl, - dim=self.dim, - nchildren=2**self.dim) + silenced_warnings="write_race(write_expn*)", + fixed_parameters=dict(dim=self.dim, nchildren=2**self.dim)) for expn in [self.src_expansion, self.tgt_expansion]: loopy_knl = expn.prepare_loopy_kernel(loopy_knl) @@ -408,11 +404,8 @@ class QBXL2P(E2PBase): ] + [arg.loopy_arg for arg in self.expansion.get_args()], name=self.name, assumptions="nglobal_qbx_centers>=1", - silenced_warnings="write_race(write_result*)") - - loopy_knl = lp.fix_parameters(loopy_knl, - dim=self.dim, - nresults=len(result_names)) + silenced_warnings="write_race(write_result*)", + fixed_parameters=dict(dim=self.dim, nresults=len(result_names))) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr") loopy_knl = self.expansion.prepare_loopy_kernel(loopy_knl) diff --git a/setup.py b/setup.py index 52d84d347139cb3ac21df485e3b6c1add0bcb692..d8d49a9cb1d30b0ac134f308ce983cd8dee0e7b1 100644 --- a/setup.py +++ b/setup.py @@ -51,7 +51,7 @@ def main(): "pyopencl>=2013.1", "boxtree>=2013.1", "pymbolic>=2013.2", - "loo.py>=2013.1beta", + "loo.py>=2017.2", "sumpy>=2013.1", "cgen>=2013.1.2",