fix

jerryzh168 · jerryzh168 · commit ed9b82e93d04 · 2024-09-26T17:28:52.000-07:00
diff --git a/torchao/testing/utils.py b/torchao/testing/utils.py
@@ -224,8 +224,6 @@ def test_linear_compile(self, device, dtype):
     NUM_DEVICES,
 )
 
-COMPILED_TENSOR_PARALLEL_REQUIRED_VERSION = version.Version("2.5.0dev")
-
 class TorchAOTensorParallelTestCase(DTensorTestBase):
     """Basic test case for tensor subclasses
     """

Original file line number	Diff line number	Diff line change
`@@ -224,8 +224,6 @@ def test_linear_compile(self, device, dtype):`
`224`	`224`	`NUM_DEVICES,`
`225`	`225`	`)`
`226`	`226`
`227`		`-COMPILED_TENSOR_PARALLEL_REQUIRED_VERSION = version.Version("2.5.0dev")`
`228`		`-`
`229`	`227`	`class TorchAOTensorParallelTestCase(DTensorTestBase):`
`230`	`228`	`"""Basic test case for tensor subclasses`
`231`	`229`	`"""`