#pragma once
// SPDX-License-Identifier: MIT
 
#include <torch/all.h>
#include <torch/extension.h>
torch::Tensor gemm_a8w8_blockscale(
    torch::Tensor &XQ,
    torch::Tensor &WQ,
    torch::Tensor &x_scale,
    torch::Tensor &w_scale,
    torch::Tensor &Y);

torch::Tensor gemm_a8w8_blockscale_tune(
    torch::Tensor &XQ,
    torch::Tensor &WQ,
    torch::Tensor &x_scale,
    torch::Tensor &w_scale,
    torch::Tensor &Y,
    int kernelId,
    int splitK);
