awq_dq_asm.h 290 Bytes
Newer Older
Xiaowei.zhang's avatar
Xiaowei.zhang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
#pragma once
// SPDX-License-Identifier: MIT
 
#include <torch/extension.h>
#include "aiter_enum.h"


void awq_dq_asm(torch::Tensor &out,
                  torch::Tensor &mat1,
                  std::optional<torch::Tensor> &zero,
                  std::optional<torch::Tensor> &scalar
);