4 ms·
Hi, So I wrote my own CUDA wrapper. Turns out there was a bug. I'm not sure where and how, but it worked for the earlier versions of Gorgonia but not this rele
by chewxy 10y ago
Hi,
So I wrote my own CUDA wrapper. Turns out there was a bug. I'm not sure where and how, but it worked for the earlier versions of Gorgonia but not this release.
You may also note that a bunch of assembly for the float32 tensor are missing... they seem to be affected by the same bug, so what I'm doing now is I'm planning to rewrite from scratch
- gtani 10y agoDid you reach out to Nvidia? It seems like they should be willing to help with putting out tested/documented bindings for more languagues. Also, where in the repo is your wrapper? (I don't have enough knowledge to debug s.t. like that, just curious how they work)
- marmaduke 10y agoWhy not OpenCL? With AVX2, it's quite a performer even on Xeons.
- hobofan 10y agoIt's still lacking as highly optimized libraries like cuDNN last time I looked :/