Add skips for missing inductor or bf16 support

Lightning-AI · Aug 15, 2024 · 76f57f9 · 76f57f9
1 parent aac7dc2
commit 76f57f9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/thunder/tests/test_torch_compile_executor.py b/thunder/tests/test_torch_compile_executor.py
@@ -46,7 +46,9 @@ def test_torch_compile_cat_nvfuser_phi2_tanh():
     logits.sum().backward()
 
 
+@pytest.mark.skipif(not is_inductor_supported(), reason="inductor unsupported")
 @requiresCUDA
+@pytest.mark.skipif(not device_supports_bf16(torch.device("cuda")), reason="bf16 is not supported")
 def test_torch_compile_cat_rope_single_fusion():
     from thunder.benchmarks import LlamaQKVSplitRopeBenchmark
     from thunder.examine import get_fusions