Fix

Author: mthrok

torchaudio/functional.py

@@ -108,10 +108,8 @@ def spectrogram(
     if power is not None:
         if power == 1.0:
             return spec_f.abs()
-        if power == 2.0:
-            if spec_f.is_cuda:
-                return torch.view_as_real(spec_f).pow(power).sum(-1)
-            return spec_f.pow(power)
+        if power == 2.0 and spec_f.is_cuda:
+            return torch.view_as_real(spec_f).pow(power).sum(-1)
         return spec_f.abs().pow(power)
     if not return_complex:
         return torch.view_as_real(spec_f)