Print testing operators

pytorch-labs · Nov 18, 2024 · e023e56 · e023e56
1 parent a886d5f
commit e023e56
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.ci/tritonbench/test-gpu.sh b/.ci/tritonbench/test-gpu.sh
@@ -11,4 +11,4 @@ fi
 # install deps
 pip install psutil tabulate
 
-python -m unittest test.test_gpu.main
+python -m unittest test.test_gpu.main -v
diff --git a/test/test_gpu/main.py b/test/test_gpu/main.py
@@ -37,6 +37,7 @@
 
 TEST_OPERATORS = set(list_operators_by_collection(op_collection="default")) - SKIP_OPS
 
+print(f"Testing operators: {TEST_OPERATORS}")
 
 def check_ci_output(op):
     from tritonbench.utils.triton_op import REGISTERED_BENCHMARKS