[Fix] Skip grid-based stage fallback for OTHER/MEMORY kernels to prevent small scheduling ops from misclassifying decode layers as prefill - #4

[Fix] Skip grid-based stage fallback for OTHER/MEMORY kernels to prevent small scheduling ops from misclassifying decode layers as prefill#4

yichiche merged 1 commit into

mainyichiche/agent-box:mainfrom

fix-grid-fallback-stage-misclassificationyichiche/agent-box:fix-grid-fallback-stage-misclassificationCopy head branch name to clipboard