jit/cuda test for complex norm. (#421)

f4f71436 · Vincent QB · GitHub · 5894928d · f4f71436
Unverified Commit f4f71436 authored Jan 22, 2020 by Vincent QB Committed by GitHub Jan 22, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

test/test_transforms.py test/test_transforms.py +4 -0

No files found.
--- a/test/test_transforms.py
+++ b/test/test_transforms.py
@@ -326,6 +326,10 @@ class Tester(unittest.TestCase):

        _test_script_module(transforms.Spectrogram, tensor, sample_rate, sample_rate_2)

+    def test_scriptmodule_ComplexNorm(self):
+        tensor = torch.rand((1, 2, 201, 2))
+        _test_script_module(transforms.ComplexNorm, tensor)
+
    def test_resample_size(self):
        input_path = os.path.join(self.test_dirpath, 'assets', 'sinewave.wav')
        waveform, sample_rate = torchaudio.load(input_path)