add ci skips (#2547)

Author: t-vi

.lightning/workflows/all-tests.yaml

@@ -4,7 +4,7 @@ trigger:
   pull_request:
     branches: ["main"]
 
-timeout: "40" # minutes
+timeout: "60" # minutes
 parametrize:
   matrix:
     image:

thunder/tests/test_dynamo.py

@@ -1294,6 +1294,7 @@ def foo(x):
         run_script(file, cmd)
 
 
+@pytest.mark.skip(reason="https://github.com/Lightning-AI/lightning-thunder/issues/2546")
 @requiresCUDA
 def test_WallTime_KernelTime():
     from nvfuser import FusionDefinition, DataType
@@ -1404,6 +1405,7 @@ def foo(x):
         run_script(file, cmd)
 
 
+@pytest.mark.skip(reason="https://github.com/Lightning-AI/lightning-thunder/issues/2546")
 @requiresCUDA
 @given(file_indices=st.lists(st.integers(min_value=0, max_value=4), min_size=1, max_size=1, unique=True))
 @settings(max_examples=2, deadline=None, suppress_health_check=[HealthCheck.function_scoped_fixture])

thunder/tests/test_grad.py

@@ -670,6 +670,10 @@ def test_vjp_correctness_nll_loss_manual(op, device, dtype, executor, comp):
 
 @ops((get_opinfo("cross_entropy"),), supported_dtypes=(dtypes.float64,))
 def test_vjp_correctness_cross_entropy_manual(op, device, dtype, executor, comp):
+    from thunder.tests.framework import nvFuserTestExecutor
+
+    if type(executor) is nvFuserTestExecutor:
+        pytest.skip("https://github.com/Lightning-AI/lightning-thunder/issues/2535")
     for sample in op.sample_inputs(device, dtype, requires_grad=True, no_rhs_numbers=True):
         # Traced backwards function does not follow PyTorch cross_entropy behavior with zero element tensors
         if sample.args[0].numel() == 0: