pytorch / glow

Compiler for Neural Network hardware accelerators

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Glow with CUDA backend

teena3 opened this issue · comments

commented

I am curios to know, if it is possible (or there are any plans in future) to have pytorch/glow with CUDA backend.

pytorch/glow seems to have support only for OpenCL backend and building pytorch with OpenCL is complicated.
It is not as easy as building pytorch with CUDA.

It is quite difficult to do any proper comparison, of pytorch with glow enabled/disabled.