# SPDX-License-Identifier: MIT
import torch
from torch import Tensor
from typing import Optional
from ..jit.core import compile_ops


MD_NAME = "module_awq_gemm_asm"


@compile_ops("module_awq_gemm_asm")
def awq_gemm_asm(
    out: Tensor,
    mat1: Tensor,
    mat2: Tensor,
    zero: Optional[Tensor] = None,
    scalar: Optional[Tensor] = None,
)->None: ...

@compile_ops("module_awq_gemm_asm")
def awq_gemm_asm_tuning(
    out: Tensor,
    mat1: Tensor,
    mat2: Tensor,
    zero: Optional[Tensor] = None,
    scalar: Optional[Tensor] = None,
    solutionid: int = 0,
    jsonfile: str = None,
)->None: ...

