Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions include/tritonblas/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,3 +2,6 @@
from .matmul import matmul_lt, matmul_a8w8_lt
from .matmul import matmul_fp4
from .origami import MatmulHeuristicResult
from .hadamard import hadamard_blocked_fast
from .fused_mxfp4_quant import fused_rms_mxfp4_quant, fused_rms_hadamard_mxfp4_quant, fused_mxfp4_quant
from .rmsnorm import rms_norm, rmsnorm2d_fwd_with_dynamicquant
Loading
Loading