skip tests if customized ops are not enables

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jun 17, 2024 · 9920e57 · 9920e57
1 parent 22ae3b7
commit 9920e57
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/source/tests/pt/test_tabulate_fusion_se_a.py b/source/tests/pt/test_tabulate_fusion_se_a.py
@@ -3,6 +3,9 @@
 
 import torch
 
+from deepmd.pt.cxx_op import (
+    ENABLE_CUSTOMIZED_OP,
+)
 from deepmd.pt.utils import (
     env,
 )
@@ -13,6 +16,7 @@
 
 
 @parameterized((torch.float64, torch.float32))
+@unittest.skipIf(not ENABLE_CUSTOMIZED_OP, "PyTorch customized OPs are not built")
 class TestTabulateFusionSeAOp(unittest.TestCase):
     def setUp(self):
         (dtype,) = self.param

diff --git a/source/tests/pt/test_tabulate_fusion_se_atten.py b/source/tests/pt/test_tabulate_fusion_se_atten.py
@@ -3,6 +3,9 @@
 
 import torch
 
+from deepmd.pt.cxx_op import (
+    ENABLE_CUSTOMIZED_OP,
+)
 from deepmd.pt.utils import (
     env,
 )
@@ -13,6 +16,7 @@
 
 
 @parameterized((torch.float64, torch.float32))
+@unittest.skipIf(not ENABLE_CUSTOMIZED_OP, "PyTorch customized OPs are not built")
 class TestTabulateFusionSeAttenOp(unittest.TestCase):
     def setUp(self):
         (dtype,) = self.param

diff --git a/source/tests/pt/test_tabulate_fusion_se_r.py b/source/tests/pt/test_tabulate_fusion_se_r.py
@@ -3,6 +3,9 @@
 
 import torch
 
+from deepmd.pt.cxx_op import (
+    ENABLE_CUSTOMIZED_OP,
+)
 from deepmd.pt.utils import (
     env,
 )
@@ -13,6 +16,7 @@
 
 
 @parameterized((torch.float64, torch.float32))
+@unittest.skipIf(not ENABLE_CUSTOMIZED_OP, "PyTorch customized OPs are not built")
 class TestTabulateFusionSeAOp(unittest.TestCase):
     def setUp(self):
         (dtype,) = self.param

diff --git a/source/tests/pt/test_tabulate_fusion_se_t.py b/source/tests/pt/test_tabulate_fusion_se_t.py
@@ -3,6 +3,9 @@
 
 import torch
 
+from deepmd.pt.cxx_op import (
+    ENABLE_CUSTOMIZED_OP,
+)
 from deepmd.pt.utils import (
     env,
 )
@@ -13,6 +16,7 @@
 
 
 @parameterized((torch.float64, torch.float32))
+@unittest.skipIf(not ENABLE_CUSTOMIZED_OP, "PyTorch customized OPs are not built")
 class TestTabulateFusionSeTOp(unittest.TestCase):
     def setUp(self):
         (dtype,) = self.param