3 ms·
As far as I can tell, its optional dependency is Open MP, not CUDA. Doesn't seem directly dependent on CUDA.
by geph2021 2y ago
As far as I can tell, its optional dependency is Open MP, not CUDA. Doesn't seem directly dependent on CUDA.
- gpderetta 2y agoYes, a quick skim of the code only shows openmp dependency. The C/CUDA reference might have meant to be C/OMP . Although I wonder if it would work well with GCC PTX OMP offloading.
- dnlzro 2y agoThe plan is to eventually implement with CUDA: "Currently, I am working on [...] direct CUDA implementation, which will be significantly faster and probably come close to PyTorch."