diff --git a/sumpy/e2e.py b/sumpy/e2e.py index b5782077c1bbf88db57b1b291723853adbea4122..74b3edd92d55f36bb3bb1a70559cade19aa52fd3 100644 --- a/sumpy/e2e.py +++ b/sumpy/e2e.py @@ -616,7 +616,7 @@ class M2LGenerateTranslationClassesDependentData(E2EBase): [""" for itr_class <> d[idim] = m2l_translation_vectors[idim, \ - itr_class + translation_classes_level_start] {dup=idim} + itr_class + translation_classes_level_start] """] + self.get_translation_loopy_insns(result_dtype) + [""" m2l_translation_classes_dependent_data[itr_class, {idx}] = \ @@ -649,7 +649,6 @@ class M2LGenerateTranslationClassesDependentData(E2EBase): loopy_knl = knl.prepare_loopy_kernel(loopy_knl) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr") - loopy_knl = lp.tag_inames(loopy_knl, dict(idim="unr")) return loopy_knl diff --git a/sumpy/p2p.py b/sumpy/p2p.py index c2aa895d5dec0c250149fe5166ede72cf8534e68..4853a2b1de94601978238f1ff4a81906dace713c 100644 --- a/sumpy/p2p.py +++ b/sumpy/p2p.py @@ -481,7 +481,7 @@ class P2PFromCSR(P2PBase): for itgt for isrc <> d[idim] = \ - targets[idim, itgt] - sources[idim, isrc] {dup=idim} + targets[idim, itgt] - sources[idim, isrc] """] + [""" <> is_self = (isrc == target_to_source[itgt]) """ if self.exclude_self else ""] diff --git a/sumpy/qbx.py b/sumpy/qbx.py index 30d3254f1a65f93074fdfd586ade54b3b9907b37..5193c3b7736235d9c9204a2360b6b0ee60c3b558 100644 --- a/sumpy/qbx.py +++ b/sumpy/qbx.py @@ -248,7 +248,7 @@ class LayerPotential(LayerPotentialBase): """], self.get_kernel_scaling_assignments() + ["for itgt, isrc"] - + ["<> a[idim] = center[idim, itgt] - sources[idim, isrc] {dup=idim}"] + + ["<> a[idim] = center[idim, itgt] - sources[idim, isrc]"] + ["<> b[idim] = targets[idim, itgt] - center[idim, itgt] {dup=idim}"] + ["<> rscale = expansion_radii[itgt]"] + [f"<> strength_{i}_isrc = strength_{i}[isrc]" for i in @@ -324,7 +324,7 @@ class LayerPotentialMatrixGenerator(LayerPotentialBase): """], self.get_kernel_scaling_assignments() + ["for itgt, isrc"] - + ["<> a[idim] = center[idim, itgt] - sources[idim, isrc] {dup=idim}"] + + ["<> a[idim] = center[idim, itgt] - sources[idim, isrc]"] + ["<> b[idim] = targets[idim, itgt] - center[idim, itgt] {dup=idim}"] + ["<> rscale = expansion_radii[itgt]"] + loopy_insns + kernel_exprs @@ -399,7 +399,7 @@ class LayerPotentialMatrixSubsetGenerator(LayerPotentialBase): <> itgt = tgtindices[imat] <> isrc = srcindices[imat] - <> a[idim] = center[idim, itgt] - sources[idim, isrc] {dup=idim} + <> a[idim] = center[idim, itgt] - sources[idim, isrc] <> b[idim] = targets[idim, itgt] - center[idim, itgt] {dup=idim} <> rscale = expansion_radii[itgt] """]