Packages

viva_tensor

2.2.100
2.2.104 2.2.103 2.2.102 2.2.100 2.1.0 retired 1.3.0 retired 1.2.1 retired 1.2.0 retired 1.1.0 retired 1.0.0 retired

Tensor library for Gleam/BEAM with a pure Gleam API, zero-copy views, and optional native acceleration

HexDocs

Current section

Files

Jump to
Readme
10 Versions
2 Dependencies
Dependants
Files
Activity
viva_tensor src viva_tensor@quant.erl
Raw

src/viva_tensor@quant.erl 

-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

    ).