tilelang.metal.transform.metal_fragment_to_simdgroup¶
Rewrite local.fragment to metal.simdgroup for legacy Metal GEMM accumulators.
M5 cooperative-tensor GEMM keeps fragment/local buffers in regular scopes until Metal codegen sees explicit tl.cooperative_tensor_* builtins. Only legacy simdgroup GEMM requires changing the accumulator scope before layout inference.
Attributes¶
Module Contents¶
- tilelang.metal.transform.metal_fragment_to_simdgroup.MetalFragmentToSimdgroup¶