[theano-users] GPU convolutions implementation

[V]

Hi,
are *any of the several implementations* of the convolutional operations
for GPU completely open source? Where exactly in the code does the
convolutional operation happen? (probably written in CUDA)
Thanks!

[Frédéric Bastien]

Hi,

the fastest implementation is from cudnn that is closed source. But we have
one implementation based on GEMM that is open source. See the class
BaseGpuCorrMM
and its chield classes:

https://github.com/Theano/Theano/blob/master/theano/gpuarray/blas.py#L444

Frédéric

Post by V
Hi,
are *any of the several implementations* of the convolutional operations
for GPU completely open source? Where exactly in the code does the
convolutional operation happen? (probably written in CUDA)
Thanks!
--
---
You received this message because you are subscribed to the Google Groups
"theano-users" group.
To unsubscribe from this group and stop receiving emails from it, send an
For more options, visit https://groups.google.com/d/optout.