Requires at least nvfuser 0.1.7.

Lightning-AI · Mar 21, 2024 · 1e8e7d2 · 1e8e7d2
1 parent 7f7113f
commit 1e8e7d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/thunder/executors/nvfuserex_impl.py b/thunder/executors/nvfuserex_impl.py
@@ -1091,7 +1091,7 @@ def broadcast_in_dim(
 
 
 def _cat_check(tensors: list[TensorProxy], dim: int) -> bool:
-    if nv_version < LooseVersion("0.1.3"):
+    if nv_version < LooseVersion("0.1.7"):
         return False
 
     # Validates tensors and concatenated dimension lengths

diff --git a/thunder/tests/opinfos.py b/thunder/tests/opinfos.py
@@ -2802,12 +2802,9 @@ def cat_wrapper(*args, dim):
             active_if=(LooseVersion(torch.__version__) < "2.2.0"),
             executors=("torchcompile",),
         ),
-        # cat op was introduced in nvFuser 0.0.5, but was disabled because of
-        # this failing test
-        # https://github.com/Lightning-AI/lightning-thunder/issues/1071
         DecorateInfo(
             pytest.mark.xfail(strict=True),
-            active_if=(nvfuser_version < "0.1.3"),
+            active_if=(nvfuser_version < "0.1.7"),
             executors=("nvFuser",),
         ),
     ),