[FEA] use of COO to balance the `faster_dot kernel` across the blocks

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] use of COO to balance the `faster_dot kernel` across the blocks #2335

Open rhdong opened 1 month ago