fix unit test to use no grad

Author: liangel-02

benchmarks/benchmark_uintx.py

@@ -6,11 +6,11 @@
 from copy import deepcopy
 
 import torch
-
 from torchao.prototype.uintx import (
     uintx_affine_weight_only,
     unpack_cpu,
 )
+
 from torchao.quantization.quant_api import quantize_
 
 

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(

test/test_low_bit_optim.py

@@ -30,6 +30,7 @@
     common_utils.SEED = 1234
 
 from packaging.version import Version
+
 from torchao import optim
 from torchao.optim.quant_utils import (
     _fp32_to_bf16_sr,