fix UT of RTN on HPU (#2098)

Signed-off-by: xin3he <xin3.he@intel.com> Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com>
intel · Dec 25, 2024 · fa8ad83 · fa8ad83
1 parent def933a
commit fa8ad83
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/.azure-pipelines/ut-3x-pt-fp8.yml b/.azure-pipelines/ut-3x-pt-fp8.yml
@@ -15,6 +15,8 @@ pr:
       - neural_compressor/torch
       - test/3x/torch/algorithms/fp8_quant
       - test/3x/torch/quantization/fp8_quant
+      - test/3x/torch/quantization/weight_only/test_rtn.py
+      - test/3x/torch/quantization/weight_only/test_load.py
       - setup.py
       - requirements_pt.txt
 

diff --git a/test/3x/torch/quantization/weight_only/test_rtn.py b/test/3x/torch/quantization/weight_only/test_rtn.py
@@ -16,6 +16,7 @@
 from neural_compressor.torch.utils import accelerator, is_hpex_available
 
 device = accelerator.current_device_name()
+torch.set_grad_enabled(False)
 
 
 class ModelConv1d(torch.nn.Module):