remove GPU dependencies in unittests

Author: mark14wu

tests/test_config.py

@@ -1,4 +1,5 @@
-import pytest
+import pytest, os
+os.environ["TRITON_SANITIZER_BACKEND"] = "off"
 import triton_viz.core.config as cfg
 
 

tests/test_print_traceback.py

@@ -27,7 +27,7 @@ def kernel_A(ptr, n):
 
 
 def test_print_nested_functions():
-    x = torch.arange(4, device="cuda", dtype=torch.float32)
+    x = torch.arange(4, dtype=torch.float32)
     print("Input:", x)
 
     # We'll launch a grid bigger than x.numel() to force a out-of-bounds error

tests/test_wrapper.py

@@ -59,6 +59,7 @@ def _decorator(fn):
         env = os.environ.copy()
         env["PYTHONPATH"] = str(tmp_path) + os.pathsep + env.get("PYTHONPATH", "")
         env["TRITON_SANITIZER_BACKEND"] = "symexec"
+        env["TRITON_INTERPRET"] = "1"
 
         # run the dummy program using triton-sanitizer
         cmd = ["triton-sanitizer", str(tmp_path / "dummy_program.py")]