Function tropical_gemm_with_argmax_inner 

pub unsafe fn tropical_gemm_with_argmax_inner<T: TropicalWithArgmax<Index = u32>, K: MicrokernelWithArgmax<T> + Sync>(
    m: usize,
    n: usize,
    k: usize,
    a: *const T::Scalar,
    lda: usize,
    trans_a: Transpose,
    b: *const T::Scalar,
    ldb: usize,
    trans_b: Transpose,
    result: &mut GemmWithArgmax<T>,
    params: &TilingParams,
    kernel: &K,
)

Tropical GEMM with argmax tracking and custom kernel.

Large outputs use disjoint Rayon tasks when parallel is enabled. The shared microkernel must implement Sync; K order and first-winner ties are unchanged.

Safety

Same requirements as tropical_gemm_portable