diff --git a/setup.py b/setup.py index fa19cffff632dcc87608714aef33cc8e9d5c5e88..c21f3b923a7c79afe9cd4c1f6d8bf0bdfb6aef3d 100644 --- a/setup.py +++ b/setup.py @@ -40,7 +40,7 @@ setup(name="sumpy", packages=["sumpy", "sumpy.expansion"], install_requires=[ - "loo.py>=2013.1beta", + "loo.py>=2017.2", "pytools>=2013.5.6", "boxtree>=2013.1", "pytest>=2.3", diff --git a/sumpy/e2e.py b/sumpy/e2e.py index bff0f508c05ea378fb5c7cd31ce85f82c0ffdfcd..749585e0970a0176fbbb47eb6b25dd6f5dd9e5f9 100644 --- a/sumpy/e2e.py +++ b/sumpy/e2e.py @@ -211,9 +211,8 @@ class E2EFromCSR(E2EBase): name=self.name, assumptions="ntgt_boxes>=1", silenced_warnings="write_race(write_expn*)", - default_offset=lp.auto) - - loopy_knl = lp.fix_parameters(loopy_knl, dim=self.dim) + default_offset=lp.auto, + fixed_parameters=dict(dim=self.dim)) for expn in [self.src_expansion, self.tgt_expansion]: loopy_knl = expn.prepare_loopy_kernel(loopy_knl) @@ -319,11 +318,8 @@ class E2EFromChildren(E2EBase): ] + gather_loopy_arguments([self.src_expansion, self.tgt_expansion]), name=self.name, assumptions="ntgt_boxes>=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) @@ -413,11 +409,9 @@ class E2EFromParent(E2EBase): "..." ] + gather_loopy_arguments([self.src_expansion, self.tgt_expansion]), name=self.name, assumptions="ntgt_boxes>=1", - silenced_warnings="write_race(write_expn*)") + silenced_warnings="write_race(write_expn*)", + fixed_parameters=dict(dim=self.dim, nchildren=2**self.dim)) - loopy_knl = lp.fix_parameters(loopy_knl, - dim=self.dim, - nchildren=2**self.dim) for expn in [self.src_expansion, self.tgt_expansion]: loopy_knl = expn.prepare_loopy_kernel(loopy_knl) diff --git a/sumpy/e2p.py b/sumpy/e2p.py index 9a2a8bc1ea564e37cf4e2d248de6952cfd386067..f21473a69f52fb62d84bc97f570cb298d7ebbb1d 100644 --- a/sumpy/e2p.py +++ b/sumpy/e2p.py @@ -181,11 +181,8 @@ class E2PFromSingleBox(E2PBase): name=self.name, assumptions="ntgt_boxes>=1", silenced_warnings="write_race(write_result*)", - default_offset=lp.auto) - - loopy_knl = lp.fix_parameters(loopy_knl, - dim=self.dim, - nresults=len(result_names)) + default_offset=lp.auto, + 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) @@ -284,11 +281,10 @@ class E2PFromCSR(E2PBase): name=self.name, assumptions="ntgt_boxes>=1", silenced_warnings="write_race(write_result*)", - default_offset=lp.auto) - - loopy_knl = lp.fix_parameters(loopy_knl, - dim=self.dim, - nresults=len(result_names)) + default_offset=lp.auto, + fixed_parameters=dict( + dim=self.dim, + nresults=len(result_names))) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr") loopy_knl = lp.prioritize_loops(loopy_knl, "itgt_box,itgt,isrc_box") diff --git a/sumpy/p2e.py b/sumpy/p2e.py index b8c60a6423bc1dd704cf63cf915831c084480b7b..7bacba79bd654c1c4f2c641f42fd61459423f68a 100644 --- a/sumpy/p2e.py +++ b/sumpy/p2e.py @@ -153,9 +153,8 @@ class P2EFromSingleBox(P2EBase): name=self.name, assumptions="nsrc_boxes>=1", silenced_warnings="write_race(write_expn*)", - default_offset=lp.auto) - - loopy_knl = lp.fix_parameters(loopy_knl, dim=self.dim) + default_offset=lp.auto, + fixed_parameters=dict(dim=self.dim)) loopy_knl = self.expansion.prepare_loopy_kernel(loopy_knl) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr") @@ -253,9 +252,8 @@ class P2EFromCSR(P2EBase): name=self.name, assumptions="ntgt_boxes>=1", silenced_warnings="write_race(write_expn*)", - default_offset=lp.auto) - - loopy_knl = lp.fix_parameters(loopy_knl, dim=self.dim) + default_offset=lp.auto, + fixed_parameters=dict(dim=self.dim)) loopy_knl = self.expansion.prepare_loopy_kernel(loopy_knl) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr") diff --git a/sumpy/p2p.py b/sumpy/p2p.py index 9e07fbe01adffbe9a0732c46a80e87ee75b86e98..f8c925c871667c7d10b3c846f61f61a9b5bff0eb 100644 --- a/sumpy/p2p.py +++ b/sumpy/p2p.py @@ -163,13 +163,11 @@ class P2P(P2PBase): if self.exclude_self else [] ) + gather_loopy_source_arguments(self.kernels), name=self.name, - assumptions="nsources>=1 and ntargets>=1") - - loopy_knl = lp.fix_parameters( - loopy_knl, - dim=self.dim, - nstrengths=self.strength_count, - nresults=len(self.kernels)) + assumptions="nsources>=1 and ntargets>=1", + fixed_parameters=dict( + dim=self.dim, + nstrengths=self.strength_count, + nresults=len(self.kernels))) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr") @@ -296,13 +294,11 @@ class P2PFromCSR(P2PBase): [lp.GlobalArg("target_to_source", np.int32, shape=("ntargets",))] if self.exclude_self else [] ) + gather_loopy_source_arguments(self.kernels), - name=self.name, assumptions="ntgt_boxes>=1") - - loopy_knl = lp.fix_parameters( - loopy_knl, + name=self.name, assumptions="ntgt_boxes>=1", + fixed_parameters=dict( dim=self.dim, nstrengths=self.strength_count, - nkernels=len(self.kernels)) + nkernels=len(self.kernels))) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr") loopy_knl = lp.tag_array_axes(loopy_knl, "strength", "sep,C") diff --git a/sumpy/qbx.py b/sumpy/qbx.py index 8892739e3b9050d3d617e8f210b288b3da4b7e28..f1df3c38bfc979f6b81f0f240c94c2c311666f7b 100644 --- a/sumpy/qbx.py +++ b/sumpy/qbx.py @@ -175,10 +175,8 @@ class LayerPotentialBase(KernelComputation, KernelCacheWrapper): name=self.name, assumptions="nsources>=1 and ntargets>=1", default_offset=lp.auto, - silenced_warnings="write_race(write_lpot*)" - ) - - loopy_knl = lp.fix_parameters(loopy_knl, dim=self.dim) + silenced_warnings="write_race(write_lpot*)", + fixed_parameters=dict(dim=self.dim)) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr")