Unverified Commit ac5dd79f authored by Vincent QB's avatar Vincent QB Committed by GitHub
Browse files

dither jit test (#417)

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

* only change dtype.
parent add9495e
...@@ -1051,7 +1051,8 @@ def _apply_probability_distribution(waveform, density_function="TPDF"): ...@@ -1051,7 +1051,8 @@ def _apply_probability_distribution(waveform, density_function="TPDF"):
signal_scaled_dis = signal_scaled + gaussian signal_scaled_dis = signal_scaled + gaussian
else: 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) TPDF = TPDF.repeat((channel_size + 1), 1)
signal_scaled_dis = signal_scaled + TPDF signal_scaled_dis = signal_scaled + TPDF
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment