search

crystal-data / num.cr

Scientific computing in pure Crystal
MIT License
151 stars 12 forks source link

CUDA support #78

Open watzon opened 1 year ago

watzon commented 1 year ago

It would be really cool if this could support CUDA as a backend

christopherzimmerman commented 1 year ago

A CUDA backend would be way easier than the current OpenCL backend due to much more documentation and support. The only reason I haven’t done it is because I don’t have a CUDA compatible graphics card, although with prices coming down maybe I should snag one!

watzon commented 1 year ago

Do it! I'd look into implementing it myself, but as many projects as I've tried to start in the past couple weeks I'm realizing I just don't have the bandwidth. I would, however, like to migrate cadmiumcr/glove to use num.cr for GPU support. It would be so much quicker to train the GLOVE models using the GPU.