[FEA]creating the masked_matmul function to front the SDDMM+custom kernel.

rapidsai / raft

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

https://docs.rapids.ai/api/raft/stable/

Apache License 2.0

680 stars 180 forks source link

[FEA]creating the masked_matmul function to front the SDDMM+custom kernel. #2336

Open rhdong opened 1 month ago