For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo module
quantization
Registers quantization and dequantization graph ops backed by the quantization and linalg kernels.
comptime valuesβ
loggerβ
comptime logger = Logger(stdout, prefix=String(""), source_location=False)
Structsβ
- β
GGMLQ40Dequantize: Registers theggml_q4_0_dequantizegraph op with the graph compiler. - β
GGMLQ4KDequantize: Registers theggml_q4_k_dequantizegraph op with the graph compiler. - β
GGMLQ6KDequantize: Registers theggml_q6_k_dequantizegraph op with the graph compiler. - β
QMatmulGPU_b4_g128: Registers theqmatmul_b4_g128graph op with the graph compiler. - β
QMatmulGPU_b4_g32: Registers theqmatmul_b4_g32graph op with the graph compiler. - β
QMatmulGPURepackGGUF: Registers theGGUF_gpu_repack_q4_0graph op with the graph compiler. - β
QMatmulGPURepackGPTQ_b4_g128: Registers theGPTQ_gpu_repack_b4_g128graph op with the graph compiler. - β
QMatmulGPURepackGPTQ_b4_g128_desc_act: Registers theGPTQ_gpu_repack_b4_g128_desc_actgraph op with the graph compiler. - β
QuantizeDynamicScaledFloat8: Registers themo.quantize_dynamic_scaled_float8graph op with the graph compiler. - β
QuantizeStaticScaledFloat8: Registers themo.quantize_static_scaled_float8graph op with the graph compiler. - β
QuantizeTensorDynamicScaledFloat8: Registers themo.quantize_tensor_dynamic_scaled_float8graph op with the graph compiler. - β
ResizeBicubic: Registers themo.resize.bicubicgraph op with the graph compiler. - β
ResizeLinear: Registers themo.resize.lineargraph op with the graph compiler. - β
ResizeNearest: Registers themo.resize.nearestgraph op with the graph compiler. - β
RMSNormFusedQuantizeDynamicScaledFP8: Registers themo.composite.rms_norm_fused_quantize_dynamic_scaled_fp8graph op with the graph compiler. - β
Struct_dequant_mxfp4: Registers themo.dequant.mxfp4graph op with the graph compiler. - β
Struct_grouped_quantize_dynamic_block_scaled: Registers themo.grouped.quantize.dynamic.block.scaledgraph op with the graph compiler. - β
Struct_interleave_block_scales: Registers themo.interleave.block.scalesgraph op with the graph compiler. - β
Struct_mxfp4_preshuffle_b_5d: Run the AMD CDNA4 MXFP4 B 5D preshuffle as a custom op. - β
Struct_mxfp4_preshuffle_scale_4d_per_expert: Per-step A-scale preshuffle for the AMD CDNA4 preb grouped matmul. - β
Struct_quantize_dynamic_block_scaled: Registers themo.quantize.dynamic.block.scaledgraph op with the graph compiler. - β
Struct_quantize_dynamic_block_scaled_mxfp4: Registers themo.quantize.dynamic.block.scaled.mxfp4graph op with the graph compiler. - β
Struct_unfused_qkv_matmul_ragged_paged_gguf_quantized: Registers themo.unfused_qkv_matmul.ragged.paged.gguf_quantizedgraph op with the graph compiler. - β
VroomQ40Matmul: Registers thevroom_q4_0_matmulgraph op with the graph compiler. - β
VroomQ40RepackWeights: Registers thevroom_q4_0_repack_weightsgraph op with the graph compiler. - β
VroomQ4KMatmul: Registers thevroom_q4_k_matmulgraph op with the graph compiler. - β
VroomQ4KRepackWeights: Registers thevroom_q4_k_repack_weightsgraph op with the graph compiler. - β
VroomQ6KMatmul: Registers thevroom_q6_k_matmulgraph op with the graph compiler. - β
VroomQ6KRepackWeights: Registers thevroom_q6_k_repack_weightsgraph op with the graph compiler.
Functionsβ
- β
composite_rms_norm_fused_quantize_dynamic_scaled_fp8_shape: Computes the output shapes for the fused RMS norm and dynamic scaled FP8 quantization op. - β
ggml_q4_0_dequantize_shape: Computes the output shape for theggml_q4_0_dequantizegraph op. - β
ggml_q4_k_dequantize_shape: Computes the output shape for theggml_q4_k_dequantizegraph op. - β
ggml_q6_k_dequantize_shape: Computes the output shape for theggml_q6_k_dequantizegraph op. - β
GGUF_gpu_repack_q4_0_shape: Computes the output shape for theGGUF_gpu_repack_q4_0graph op. - β
GPTQ_gpu_repack_b4_g128_desc_act_shape: Computes the output shape for theGPTQ_gpu_repack_b4_g128_desc_actgraph op. - β
GPTQ_gpu_repack_b4_g128_shape: Computes the output shape for theGPTQ_gpu_repack_b4_g128graph op. - β
qmatmul_b4_g128_shape: Computes the output shape for theqmatmul_b4_g128graph op. - β
qmatmul_b4_g32_shape: Computes the output shape for theqmatmul_b4_g32graph op. - β
resize_bicubic_shape: Computes the output shape for themo.resize.bicubicgraph op. - β
resize_linear_shape: Computes the output shape for themo.resize.lineargraph op. - β
resize_nearest_shape: Computes the output shape for themo.resize.nearestgraph op. - β
vroom_q4_0_matmul_shape: Computes the output shape for thevroom_q4_0_matmulgraph op. - β
vroom_q4_0_repack_weights_shape: Computes the output shape for thevroom_q4_0_repack_weightsgraph op. - β
vroom_q4_k_matmul_shape: Computes the output shape for thevroom_q4_k_matmulgraph op. - β
vroom_q4_k_repack_weights_shape: Computes the output shape for thevroom_q4_k_repack_weightsgraph op. - β
vroom_q6_k_matmul_shape: Computes the output shape for thevroom_q6_k_matmulgraph op. - β
vroom_q6_k_repack_weights_shape: Computes the output shape for thevroom_q6_k_repack_weightsgraph op.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!