Mark that a test requires nvFuser. (#1402)

Lightning-AI · Nov 6, 2024 · 2ef11f7 · 2ef11f7
1 parent f44fc2e
commit 2ef11f7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/thunder/tests/test_core.py b/thunder/tests/test_core.py
@@ -2833,6 +2833,7 @@ def fn(x: torch.Tensor, y: torch.Tensor):
 
 
 @requiresCUDA
+@thunder.tests.framework.requiresNVFuser
 def test_bound_symbol_sort_stability():
     class LlamaMLPLike(torch.nn.Module):
         def __init__(self) -> None: