search

kakao / buffalo

TOROS Buffalo: A fast and scalable production-ready open source project for recommender systems
Apache License 2.0
576 stars 106 forks source link

extend cuda compatiblity #43

Closed js1010 closed 1 year ago

js1010 commented 3 years ago

currently, buffalo does not work for cuda ver11.x. this PR may resolve the cuda compatibility issue.

ita9naiwa commented 1 year ago

https://github.com/kakao/buffalo/blob/6fb01c8e807eeb700000b9b46e8c412f3cd115e4/cuda_extension.py#L49-L55 these lines of codes resolve this i guess.