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
784 stars 195 forks source link

Allow compilation with CUDA 12.6.1 #2469

Closed robertmaynard closed 1 month ago

robertmaynard commented 1 month ago

The 12.6.1 CUDA compiler has issues with enable_if inside the template arguments of some kernels. We can simplify kernel logic and remove the usage of enable_if.

robertmaynard commented 1 month ago

/merge