optimum/quanto/library/extensions/cuda/awq/v2/gemm_cuda.h (2 lines of code) (raw):

#include <torch/extension.h> torch::Tensor awq_v2_gemm_f16i4(torch::Tensor _in_feats, torch::Tensor _kernel, torch::Tensor _scales, torch::Tensor _zeros);