OpenCL counterpart of cuDNN

naibaf7 / caffe

Caffe: a fast open framework for deep learning. With OpenCL and CUDA support.

http://caffe.berkeleyvision.org/

Other

85 stars 20 forks source link

OpenCL counterpart of cuDNN #34

Open dagamayank opened 8 years ago

dagamayank commented 8 years ago

I came across your post on the Tensorflow thread that you are developing an OpenCL counterpart for cuDNN. I would like to help/contribute on that project. Let me know where and how can I help. I have extensive OpenCL programming experience and am currently focused on ML activities at AMD.

naibaf7 commented 7 years ago

No, this is just the latest point in time I project being done with this step :) It could be from anywhere 2 to 4 weeks until LibDNN is on the newest kernel versions and gets pooling support.

bhack commented 7 years ago

Ok ;)

naibaf7 commented 7 years ago

After that I don't know what the next optimization is going to be. Either mobile ARM chips with integrated GPUs or AMD's Vega and FP16, depending on what I can get my hands on first.