dither jit test (#417)

* workaround for bartlett_window https://github.com/pytorch/pytorch/issues/32358#issuecomment-576909755 * only change dtype.

dither jit test (#417)
* workaround for bartlett_window https://github.com/pytorch/pytorch/issues/32358#issuecomment-576909755 * only change dtype.
ac5dd79f · Vincent QB · GitHub · add9495e · ac5dd79f
Unverified Commit ac5dd79f authored Jan 29, 2020 by Vincent QB Committed by GitHub Jan 29, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

torchaudio/functional.py torchaudio/functional.py +2 -1

No files found.
--- a/torchaudio/functional.py
+++ b/torchaudio/functional.py
@@ -1051,7 +1051,8 @@ def _apply_probability_distribution(waveform, density_function="TPDF"):

        signal_scaled_dis = signal_scaled + gaussian
    else:
-        TPDF = torch.bartlett_window(time_size + 1)
+        # dtype needed for https://github.com/pytorch/pytorch/issues/32358
+        TPDF = torch.bartlett_window(time_size + 1, dtype=torch.float)
        TPDF = TPDF.repeat((channel_size + 1), 1)
        signal_scaled_dis = signal_scaled + TPDF