Struct TropicalGemm 

pub struct TropicalGemm<T: TropicalSemiring> { /* private fields */ }

Builder for configuring tropical GEMM operations.

Provides a fluent API for setting options like transposition, alpha/beta scaling, and output preferences.

Example

use tropical_gemm::{TropicalGemm, TropicalMaxPlus, TropicalSemiring};

let a = vec![1.0f32; 6]; // 2x3
let b = vec![1.0f32; 6]; // 3x2
let mut c = vec![TropicalMaxPlus::tropical_zero(); 4]; // 2x2

TropicalGemm::<TropicalMaxPlus<f32>>::new(2, 2, 3)
    .execute(&a, 3, &b, 2, &mut c, 2);

Implementations

impl<T: TropicalSemiring + KernelDispatch> TropicalGemm<T>

pub fn new(m: usize, n: usize, k: usize) -> Self

Create a new GEMM builder.

pub fn trans_a(self) -> Self

Transpose matrix A.

pub fn trans_b(self) -> Self

Transpose matrix B.

pub fn execute( self, a: &[T::Scalar], lda: usize, b: &[T::Scalar], ldb: usize, c: &mut [T], ldc: usize, )

Execute the GEMM operation.

Arguments

• a: Matrix A data
• lda: Leading dimension of A
• b: Matrix B data
• ldb: Leading dimension of B
• c: Output matrix C (must be pre-allocated)
• ldc: Leading dimension of C