windows
matmul_kernel[grid]( a, b, c, M, N, K, stride_am, stride_an, stride_bn, stride_bk, stride_cm, stride_ck, BLOCK_SIZE_M=128, BLOCK_SIZE_K=64, BLOCK_SIZE_N=64, arch="sm90" )
查看全文