From 5bf0d0fa64be8908c45a31854589a134aae173f7 Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Wed, 20 Oct 2021 14:43:03 -0700
Subject: [PATCH] Remove some more variable access ordered checks (#88)
* Remove some more variable access ordered checks
* more checks removed
---
sumpy/e2e.py | 4 ++++
sumpy/e2p.py | 4 ++++
sumpy/p2e.py | 2 ++
sumpy/p2p.py | 6 ++++++
4 files changed, 16 insertions(+)
diff --git a/sumpy/e2e.py b/sumpy/e2e.py
index 2d13ba5a..7774b62a 100644
--- a/sumpy/e2e.py
+++ b/sumpy/e2e.py
@@ -649,6 +649,8 @@ class M2LGenerateTranslationClassesDependentData(E2EBase):
loopy_knl = knl.prepare_loopy_kernel(loopy_knl)
loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr")
+ loopy_knl = lp.set_options(loopy_knl,
+ enforce_variable_access_ordered="no_check")
return loopy_knl
@@ -983,6 +985,8 @@ class E2EFromParent(E2EBase):
loopy_knl = knl.prepare_loopy_kernel(loopy_knl)
loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr")
+ loopy_knl = lp.set_options(loopy_knl,
+ enforce_variable_access_ordered="no_check")
return loopy_knl
diff --git a/sumpy/e2p.py b/sumpy/e2p.py
index 2a4ff531..25e03242 100644
--- a/sumpy/e2p.py
+++ b/sumpy/e2p.py
@@ -202,6 +202,8 @@ class E2PFromSingleBox(E2PBase):
knl = self.get_kernel()
knl = lp.tag_inames(knl, dict(itgt_box="g.0"))
knl = self._allow_redundant_execution_of_knl_scaling(knl)
+ knl = lp.set_options(knl,
+ enforce_variable_access_ordered="no_check")
return knl
@@ -314,6 +316,8 @@ class E2PFromCSR(E2PBase):
knl = self.get_kernel()
knl = lp.tag_inames(knl, dict(itgt_box="g.0"))
knl = self._allow_redundant_execution_of_knl_scaling(knl)
+ knl = lp.set_options(knl,
+ enforce_variable_access_ordered="no_check")
return knl
def __call__(self, queue, **kwargs):
diff --git a/sumpy/p2e.py b/sumpy/p2e.py
index cb4b6616..177d0b58 100644
--- a/sumpy/p2e.py
+++ b/sumpy/p2e.py
@@ -131,6 +131,8 @@ class P2EBase(KernelComputation, KernelCacheWrapper):
knl = lp.tag_array_axes(knl, "centers", "sep,C")
knl = self._allow_redundant_execution_of_knl_scaling(knl)
+ knl = lp.set_options(knl,
+ enforce_variable_access_ordered="no_check")
return knl
def __call__(self, queue, **kwargs):
diff --git a/sumpy/p2p.py b/sumpy/p2p.py
index 4853a2b1..21dde61e 100644
--- a/sumpy/p2p.py
+++ b/sumpy/p2p.py
@@ -187,6 +187,8 @@ class P2PBase(KernelComputation, KernelCacheWrapper):
knl = lp.split_iname(knl, "itgt", 1024, outer_tag="g.0")
knl = self._allow_redundant_execution_of_knl_scaling(knl)
+ knl = lp.set_options(knl,
+ enforce_variable_access_ordered="no_check")
return knl
@@ -395,6 +397,8 @@ class P2PMatrixSubsetGenerator(P2PBase):
knl = lp.split_iname(knl, "imat", 1024, outer_tag="g.0")
knl = self._allow_redundant_execution_of_knl_scaling(knl)
+ knl = lp.set_options(knl,
+ enforce_variable_access_ordered="no_check")
return knl
def __call__(self, queue, targets, sources, tgtindices, srcindices, **kwargs):
@@ -534,6 +538,8 @@ class P2PFromCSR(P2PBase):
knl = lp.split_iname(knl, "itgt_box", 4, outer_tag="g.0")
knl = self._allow_redundant_execution_of_knl_scaling(knl)
+ knl = lp.set_options(knl,
+ enforce_variable_access_ordered="no_check")
return knl
--
GitLab