Hugging Face's logo

kernels-community
/
triton-scaled-mm

kernel
Model card Files 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"]