[FEA] add the support of `masked_matmul`

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

746 stars 189 forks source link

[FEA] add the support of `masked_matmul` #2362

Closed rhdong closed 2 months ago

rhdong commented 3 months ago

https://github.com/rapidsai/raft/issues/2336

cjnolet commented 2 months ago

/merge