diff --git a/test/torchaudio_unittest/sox_io_backend/load_test.py b/test/torchaudio_unittest/sox_io_backend/load_test.py index f2a514ea61..c1004f7865 100644 --- a/test/torchaudio_unittest/sox_io_backend/load_test.py +++ b/test/torchaudio_unittest/sox_io_backend/load_test.py @@ -83,7 +83,7 @@ def assert_mp3(self, sample_rate, num_channels, bit_rate, duration): data_ref = load_wav(ref_path)[0] # 5. Compare assert sr == sample_rate - self.assertEqual(data, data_ref, atol=3e-03, rtol=1.3e-06) + self.assertEqual(data, data_ref, atol=5e-05, rtol=1.3e-06) def assert_flac(self, sample_rate, num_channels, compression_level, duration): """`sox_io_backend.load` can load flac format. diff --git a/torchaudio/csrc/sox/utils.cpp b/torchaudio/csrc/sox/utils.cpp index 1ac6dd3faf..b200eec55b 100644 --- a/torchaudio/csrc/sox/utils.cpp +++ b/torchaudio/csrc/sox/utils.cpp @@ -169,7 +169,7 @@ torch::Tensor convert_to_tensor( // So make sure to create a new copy after processing samples. if (normalize || dtype == torch::kFloat32) { t = t.to(torch::kFloat32); - t *= (t > 0) / 2147483647. + (t < 0) / 2147483648.; + t.divide_(2147483648); } else if (dtype == torch::kInt32) { t = t.clone(); } else if (dtype == torch::kInt16) {