Skip to content

Commit f4d8329

Browse files
committed
Enable group mode varlen kernel generation for PyTorch
1 parent 3f04d27 commit f4d8329

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1219,7 +1219,6 @@ def fit(problem_ctx: ProblemContext, kernel_ctx: KernelContext) -> bool:
12191219
cond &= kernel_ctx.pipeline.F_vlayout == "row"
12201220
cond &= kernel_ctx.pipeline.F_bias in ["no", "bias"]
12211221
cond &= kernel_ctx.pipeline.F_qscale == "no"
1222-
cond &= problem_ctx.mode == "batch"
12231222
cond &= kernel_ctx.pipeline.F_skip == "f"
12241223
cond &= kernel_ctx.pipeline.F_logits == "f"
12251224
return cond

0 commit comments

Comments
 (0)