torch-ext/torch_binding.h

#pragma once

#include <torch/torch.h>

void bitlinear_int8xint2_cpp(torch::Tensor const &input0, torch::Tensor const &input1, torch::Tensor const &s, torch::Tensor const &ws);