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¶