Correction

FluxML · Jun 7, 2024 · 7927d93 · 7927d93
1 parent ef9b3d5
commit 7927d93
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/testsuite/spectral.jl b/test/testsuite/spectral.jl
@@ -26,10 +26,10 @@ function spectral_testsuite(Backend)
             @testset "Grads" begin
                 if Backend != CPU
                     x = rand(Float32, 16, batch...)
-                    gradtest_fn(s -> abs.(stft(s; n_fft=16)), x)
+                    window = hann_window(16)
 
-                    window = device(hann_window(16))
-                    gradtest_fn(s -> abs.(stft(s; n_fft=16, window)), x)
+                    gradtest_fn(s -> abs.(stft(s; n_fft=16)), x)
+                    gradtest_fn((s, w) -> abs.(stft(s; n_fft=16, window=w)), x, window)
                 end
             end