Skip to content

Commit 06cc355

Browse files
committed
fix: disable variance fusion by default (small backbones, no benefit)
1 parent 99dad24 commit 06cc355

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

training/variance_task.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def __init__(self):
124124
# NOTE: Variance model backbones use num_channels=384/512, which are too
125125
# small for Triton fusion to benefit. The compilation overhead outweighs
126126
# the HBM savings. Fused kernels for acoustic model only (K=1024).
127-
if hparams.get('use_fused_kernels', False):
127+
if hparams.get('use_fused_kernels_variance', False):
128128
from modules.kernels.integration import patch_variance_model
129129
from lightning.pytorch.utilities.rank_zero import rank_zero_info
130130
n = patch_variance_model(

0 commit comments

Comments
 (0)