Skip advanced RNG tests on any setup that's missing cuRand

manopapad · Aug 2, 2023 · 30b79e1 · 30b79e1
1 parent d54a3dd
commit 30b79e1
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 10 deletions.
diff --git a/tests/integration/test_random_advanced.py b/tests/integration/test_random_advanced.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 #
 import os
-import sys
 
 import numpy as np
 import pytest
@@ -22,7 +21,7 @@
 import cunumeric as num
 
 LEGATE_TEST = os.environ.get("LEGATE_TEST", None) == "1"
-if sys.platform == "darwin":
+if not num.runtime.has_curand:
     pytestmark = pytest.mark.skip()
     BITGENERATOR_ARGS = []
 else:

diff --git a/tests/integration/test_random_beta.py b/tests/integration/test_random_beta.py
@@ -12,15 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import sys
 
 import numpy as np
 import pytest
 from utils.random import ModuleGenerator, assert_distribution
 
 import cunumeric as num
 
-if sys.platform == "darwin":
+if not num.runtime.has_curand:
     pytestmark = pytest.mark.skip()
     BITGENERATOR_ARGS = []
 else:

diff --git a/tests/integration/test_random_bitgenerator.py b/tests/integration/test_random_bitgenerator.py
@@ -12,15 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import sys
 
 import numpy as np
 import pytest
 from utils.random import ModuleGenerator, assert_distribution
 
 import cunumeric as num
 
-if sys.platform == "darwin":
+if not num.runtime.has_curand:
     pytestmark = pytest.mark.skip()
     BITGENERATOR_ARGS = []
 else:

diff --git a/tests/integration/test_random_gamma.py b/tests/integration/test_random_gamma.py
@@ -12,15 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import sys
 
 import numpy as np
 import pytest
 from utils.random import ModuleGenerator, assert_distribution
 
 import cunumeric as num
 
-if sys.platform == "darwin":
+if not num.runtime.has_curand:
     pytestmark = pytest.mark.skip()
     BITGENERATOR_ARGS = []
 else:

diff --git a/tests/integration/test_random_straightforward.py b/tests/integration/test_random_straightforward.py
@@ -13,15 +13,14 @@
 # limitations under the License.
 #
 import math
-import sys
 
 import numpy as np
 import pytest
 from utils.random import ModuleGenerator, assert_distribution
 
 import cunumeric as num
 
-if sys.platform == "darwin":
+if not num.runtime.has_curand:
     pytestmark = pytest.mark.skip()
     BITGENERATOR_ARGS = []
 else: