gemm_cuda.h

#pragma once

#include "common.h"
#include "Tensor.h"


Tensor awq_gemm_forward_cuda(
    Tensor _in_feats,
    Tensor _kernel,
    Tensor _scales,
    Tensor _zeros);