replicate
/

quantization-eetq

Kernels

Model card Files Files and versions

xet

Community

quantization-eetq / torch-ext /quantization_eetq /__init__.py

danieldk HF Staff

Import EETQ kernels

1dc29e9 over 1 year ago

Raw History Blame

167 Bytes

from .custom_ops import w8_a16_gemm, w8_a16_gemm_, preprocess_weights, quant_weights

__all__ = ["w8_a16_gemm", "w8_a16_gemm_", "preprocess_weights", "quant_weights"]

	from .custom_ops import w8_a16_gemm, w8_a16_gemm_, preprocess_weights, quant_weights

	__all__ = ["w8_a16_gemm", "w8_a16_gemm_", "preprocess_weights", "quant_weights"]