Hugging Face's logo

kernels-community
/
triton-scaled-mm

Kernels
Model card Files
xet
Community
triton-scaled-mm / torch-ext /triton_scaled_mm /__init__.py
danieldk's picture
danieldk HF Staff
Add Triton scaled matmul kernel from vLLM
be26e8c
raw
history blame contribute delete
79 Bytes
 from .triton_scaled_mm import triton_scaled_mm
__all__ = ["triton_scaled_mm"]