GPUArrays.jl

GPUArrays.jl copied to clipboard

See here for a description of the operations involved.

I expect that pooling will be relatively easy, convolutions are also straightforward enough in principle but hard to make really fast. We probably want to port or wrap an existing high-quality implementation for Open CL, but I'm not sure what the benchmarks is here (perhaps DeepCL?).

I am interested to take this up! From what I interpret, do we require the GPUArray version of the convolutions and pooling in Julia? Please correct me if I am wrong.

Yes, that or some wrapper for an existing implementation. A generic kernel is probably a good idea either way though.

@tejank10 have you had any luck with this?