Move new test case from gpu to cpu

Signed-off-by: Kyunggeun Lee <quic_kyunggeu@quicinc.com>
quic · Oct 31, 2023 · d6e9ebc · d6e9ebc
1 parent 870efda
commit d6e9ebc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TrainingExtensions/torch/test/python/test_tensor_quantizer.py b/TrainingExtensions/torch/test/python/test_tensor_quantizer.py
@@ -402,7 +402,7 @@ def test_learned_grid_encoding_getter(self, dtype):
         quant_wrapper = LearnedGridQuantWrapper(conv, round_mode=libpymo.RoundingMode.ROUND_NEAREST,
                                                 quant_scheme=QuantScheme.training_range_learning_with_tf_init,
                                                 is_output_quantized=True, activation_bw=16,
-                                                weight_bw=8, device="cuda:0")
+                                                weight_bw=8, device="cpu")
         enc = libpymo.TfEncoding()
         enc.bw, enc.max, enc.min = 16, 0.4, -0.98