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

MD_NAME = "module_custom"


@compile_ops("module_custom")
def wvSpltK(in_a: Tensor, in_b: Tensor, out_c: Tensor, N_in: int, CuCount: int): ...


@compile_ops("module_custom")
def LLMM1(in_a: Tensor, in_b: Tensor, out_c: Tensor, rows_per_block: int): ...
