Temporary Fix: Skip on H100

jainapurva · jainapurva · commit ce5273c1c37f · 2024-10-02T19:02:21.000-07:00
diff --git a/test/dtypes/test_affine_quantized_tensor_parallel.py b/test/dtypes/test_affine_quantized_tensor_parallel.py
@@ -8,10 +8,7 @@ class TestAffineQuantizedTensorParallel(TorchAOTensorParallelTestCase):
 
 is_H100 = torch.cuda.is_available() and torch.cuda.get_device_capability() >= (9, 0)
 
-if not is_H100:
-    copy_tests(TorchAOTensorParallelTestCase, TestAffineQuantizedTensorParallel, "aqt_tp")
-else:
-    print("Skipping TestAffineQuantizedTensorParallel because it doesn't run on H100")
+copy_tests(TorchAOTensorParallelTestCase, TestAffineQuantizedTensorParallel, "aqt_tp")
 
 if __name__ == "__main__":
     if not is_H100:
