Mojo module
bmm
comptime values
elementwise_epilogue_type
comptime elementwise_epilogue_type = def[c_type: DType, width: Int, rank: Int, *, alignment: Int = 1](IndexList[rank], SIMD[c_type, width]) capturing -> None
logger
comptime logger = Logger(stdout, prefix=String(""), source_location=False)
Functions
-
batched_matmul: TileTensor primary implementation ofbatched_matmul. -
batched_matmul_dynamic_scaled_fp8: -
batched_matmul_dynamic_scaled_fp8_naive: -
batched_matmul_kernel_gpu: -
batched_matmul_shape: Compute the output shape of abatch_matmuloperation, and assert the inputs are compatible. -
bmm_sm100_blockwise_scaled_fp8: -
naive_batched_matmul_kernel:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!