skip nf4 fsdp2

zxd1997066 · zxd1997066 · commit 243bd6f9e7ca · 2025-12-19T13:42:14.000+08:00
diff --git a/test/dtypes/test_nf4.py b/test/dtypes/test_nf4.py
@@ -634,7 +634,7 @@ def world_size(self) -> int:
         return 2
 
     @pytest.mark.skipif(
-        version.parse(torch.__version__) < version.parse("2.4.0"),
+        version.parse(torch.__version__).base_version < "2.4.0",
         reason="torch >= 2.4 required",
     )
     @skip_if_lt_x_gpu(2)

Original file line number	Diff line number	Diff line change
`@@ -634,7 +634,7 @@ def world_size(self) -> int:`
`634`	`634`	`return 2`
`635`	`635`
`636`	`636`	`@pytest.mark.skipif(`
`637`		`- version.parse(torch.__version__) < version.parse("2.4.0"),`
	`637`	`+ version.parse(torch.__version__).base_version < "2.4.0",`
`638`	`638`	`reason="torch >= 2.4 required",`
`639`	`639`	`)`
`640`	`640`	`@skip_if_lt_x_gpu(2)`