fix version

pytorch · Sep 27, 2024 · b113eda · b113eda
1 parent ed9b82e
commit b113eda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchao/testing/utils.py b/torchao/testing/utils.py
@@ -320,7 +320,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
         y_d = dn_dist(up_dist(input_dtensor))
 
-        if TORCH_VERSION_AT_LEAST_2_5:
+        if not TORCH_VERSION_AT_LEAST_2_5:
             # Need torch 2.5 to support compiled tensor parallelism
             return