-module(viva_tensor@quant).
-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch, inline]).
-define(FILEPATH, "src/viva_tensor/quant.gleam").
-export([matmul_nf4/7]).

-if(?OTP_RELEASE >= 27).
-define(MODULEDOC(Str), -moduledoc(Str)).
-define(DOC(Str), -doc(Str)).
-else.
-define(MODULEDOC(Str), -compile([])).
-define(DOC(Str), -compile([])).
-endif.

?MODULEDOC(false).

-file("src/viva_tensor/quant.gleam", 21).
?DOC(false).
-spec matmul_nf4(
    viva_tensor@core@ffi:native_tensor_ref(),
    list(integer()),
    list(float()),
    integer(),
    integer(),
    integer(),
    integer()
) -> {ok, viva_tensor@core@ffi:native_tensor_ref()} | {error, binary()}.
matmul_nf4(A, B_indices, B_scales, M, N, K, Block_size) ->
    viva_tensor@core@ffi:nt_matmul_nf4(
        A,
        B_indices,
        B_scales,
        M,
        N,
        K,
        Block_size
    ).