Function tropical_gemm_with_argmax_dispatch 

pub unsafe fn tropical_gemm_with_argmax_dispatch<T: KernelDispatch + TropicalWithArgmax<Index = u32>>(
    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>,
)

Runtime-dispatched GEMM with first-winner indices.

Safety

Same pointer and output storage requirements as tropical_gemm_with_argmax_portable.