feat(compile): CustomKernel and GatherQMM implement output_shapes for shapeless compile#3485

Open

dexwritescode wants to merge 8 commits into

ml-explore:mainfrom

dexwritescode:fix/custom-kernel-output-shapes

Commits on May 5, 2026

feat(compile): CustomKernel stores and returns output shapes
dexwritescode
committed
feat(compile): GatherQMM implements output_shapes for shapeless compile
dexwritescode
committed

Commits on May 6, 2026

fix(compile): remove std::move on const& and fix comment alignment
dexwritescode
committed
test(compile): add shapeless compile tests for CustomKernel and GatherQMM
dexwritescode
committed

Commits on May 7, 2026

Commits on May 8, 2026