From 1fa125370e925c927a5bd16564be8cba848beac4 Mon Sep 17 00:00:00 2001 From: Angel Li Date: Mon, 3 Nov 2025 10:54:45 -0800 Subject: [PATCH] fix unit test to use no grad --- .../quantize_/workflows/float8/test_float8_tensor.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/quantization/quantize_/workflows/float8/test_float8_tensor.py b/test/quantization/quantize_/workflows/float8/test_float8_tensor.py index 9f77ad020d..4871b48849 100644 --- a/test/quantization/quantize_/workflows/float8/test_float8_tensor.py +++ b/test/quantization/quantize_/workflows/float8/test_float8_tensor.py @@ -80,6 +80,7 @@ def forward(self, x): class TestFloat8Tensor(TorchAOIntegrationTestCase): def setUp(self): self.GPU_DEVICES = ["cuda"] if torch.cuda.is_available() else [] + torch.set_grad_enabled(False) @unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available") @unittest.skipIf(