Mojo module
dispatch
comptime valuesβ
DISPATCH_HITβ
comptime DISPATCH_HIT = 1
DISPATCH_MISSβ
comptime DISPATCH_MISS = 0
loggerβ
comptime logger = Logger(stdout, prefix=String(""), source_location=False)
Functionsβ
- β
dispatch_gemv: Dispatch M=1 (or N=1) matmul to GEMV or SM100 GEMM based on (N, K). - β
dispatch_sm100_batched_matmul: Dispatch batched matmul to SM100 kernel. - β
heuristic_and_outliers_dispatch: - β
matmul_dispatch_sm100: - β
matmul_dispatch_sm100_bf16: - β
matmul_dispatch_sm100_fp8: - β
select_and_launch_sm100_config: - β
sm100_heuristic_and_outliers_dispatch: - β
small_MN_gemms:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!