From 75e9330ce1ad1689f87114ed49e3ac3fdf363e92 Mon Sep 17 00:00:00 2001
From: Pian Pawakapan <pianpwk@meta.com>
Date: Wed, 21 May 2025 18:11:56 -0700
Subject: [PATCH] Update functional.py

---
 src/torchaudio/functional/functional.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/torchaudio/functional/functional.py b/src/torchaudio/functional/functional.py
index af34e707e5..d4175b8af9 100644
--- a/src/torchaudio/functional/functional.py
+++ b/src/torchaudio/functional/functional.py
@@ -580,7 +580,7 @@ def melscale_fbanks(
         enorm = 2.0 / (f_pts[2 : n_mels + 2] - f_pts[:n_mels])
         fb *= enorm.unsqueeze(0)
 
-    if (fb.max(dim=0).values == 0.0).any():
+    if not torch.compiler.is_compiling() and (fb.max(dim=0).values == 0.0).any():
         warnings.warn(
             "At least one mel filterbank has all zero values. "
             f"The value for `n_mels` ({n_mels}) may be set too high. "