From d6e9ebcb594a73d06b092aab90fa06eac326ab45 Mon Sep 17 00:00:00 2001 From: Kyunggeun Lee Date: Mon, 30 Oct 2023 11:18:47 -0700 Subject: [PATCH] Move new test case from gpu to cpu Signed-off-by: Kyunggeun Lee --- TrainingExtensions/torch/test/python/test_tensor_quantizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TrainingExtensions/torch/test/python/test_tensor_quantizer.py b/TrainingExtensions/torch/test/python/test_tensor_quantizer.py index 1917944d190..4fa19f7bb0c 100644 --- 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