Skip to content

Commit 29c230b

Browse files
authored
fix skipped kernel while training (#14365)
1 parent b97e42b commit 29c230b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nemo/collections/llm/gpt/model/megatron/hyena/hyena_mixer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,7 @@ def pad_to_multiple(x, multiple=16):
309309
if (
310310
self.use_b2b_causal_conv1d
311311
and self.operator_type in ["hyena_short_conv", "hyena_medium_conv"]
312-
and inference_context is not None
312+
and inference_context is None
313313
):
314314
# todo: support inference_context for b2b_kernel
315315
# Use the B2BCausalConv1dModule wrapper with the existing weights from the original model

0 commit comments

Comments
 (0)