Add pytest.mark.cpu to two more already-passing tests (#183)

triton-lang · Dec 6, 2024 · d0d4ccc · d0d4ccc
1 parent ee7aa42
commit d0d4ccc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py
@@ -4185,6 +4185,7 @@ def _kernel(dst, src, off, N, BLOCK_SIZE: tl.constexpr, HINT: tl.constexpr):
         assert "ld.global.v4.b32" not in ptx
 
 
+@pytest.mark.cpu
 @pytest.mark.interpreter
 def test_assume(device):
 
@@ -5185,6 +5186,7 @@ def nested_while(data, countPtr):
     assert data[0] == 40
 
 
+@pytest.mark.cpu
 def test_constexpr_if_return(device):
     # Reproducer for #4883, return statement in an if with a constexpr causes
     # errors when combined with non-trivial control flow graphs