spmm!
is both available directly, by calling CUSPARSE.cusparseSpMM
; as a slightly higher-level CUSPARSE.mm!
, CUDA.jl/lib/cusparse/generic.jl at 5b470c4614f7388c7b5b0938a93f781294673e80 · JuliaGPU/CUDA.jl · GitHub; and via LinearAlgebra.mul!
as called on sparse arrays, CUDA.jl/lib/cusparse/interfaces.jl at 5b470c4614f7388c7b5b0938a93f781294673e80 · JuliaGPU/CUDA.jl · GitHub.
1 Like