[torchaudio] Fix jitting error

pssrawat · web-flow · commit 20378a7f2448 · 2025-08-21T13:52:44.000-07:00
Differential Revision: D80573765 Pull Request resolved: #4053
diff --git a/src/torchaudio/functional/functional.py b/src/torchaudio/functional/functional.py
@@ -876,7 +876,7 @@ def mask_along_axis_iid(
     specgrams = specgrams.transpose(axis, -1)
     # this aims to avoid CPU-GPU sync from upstream
     specgrams = (
-        torch.where((mask >= mask_start) & (mask < mask_end), mask_value.repeat(*specgrams.shape), specgrams)
+        torch.where((mask >= mask_start) & (mask < mask_end), mask_value.repeat(specgrams.shape), specgrams)
         if isinstance(mask_value, Tensor)
         else specgrams.masked_fill((mask >= mask_start) & (mask < mask_end), mask_value)
     )

Original file line number	Diff line number	Diff line change
`@@ -876,7 +876,7 @@ def mask_along_axis_iid(`
`876`	`876`	`specgrams = specgrams.transpose(axis, -1)`
`877`	`877`	`# this aims to avoid CPU-GPU sync from upstream`
`878`	`878`	`specgrams = (`
`879`		`- torch.where((mask >= mask_start) & (mask < mask_end), mask_value.repeat(*specgrams.shape), specgrams)`
	`879`	`+ torch.where((mask >= mask_start) & (mask < mask_end), mask_value.repeat(specgrams.shape), specgrams)`
`880`	`880`	`if isinstance(mask_value, Tensor)`
`881`	`881`	`else specgrams.masked_fill((mask >= mask_start) & (mask < mask_end), mask_value)`
`882`	`882`	`)`