提交 63b6b80b authored 作者: Frederic's avatar Frederic

Add support of GpuGer in GpuAlloc.do_constant_folding.

上级 d57b103b
......@@ -672,7 +672,7 @@ class GpuAlloc(HideC, Alloc):
#theano.sandbox.gpuarray.subtensor.GpuAdvancedIncSubtensor1,
theano.sandbox.gpuarray.blas.GpuGemm,
theano.sandbox.gpuarray.blas.GpuGemv,
#theano.sandbox.gpuarray.blas.GpuGer, Not Yet implemented
theano.sandbox.gpuarray.blas.GpuGer,
))):
return False
#If the clients is a transfer, we don't want to fold. We
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论