Diseño de núcleos de GPU de alto rendimiento con TileLang: Tensor-Core GEMM, Fused Softmax, FlashAttention y Autotuning
@tilelang.jit(out_idx=) def make_matmul(M: int, N: int, K: int, block_M: int = 128, block_N: int = 128, block_K: int = 32, num_stages: int = 3, subprocesos: int = 128, use_swizzle: bool…