提交 5472dd89 authored 作者: Frederic Bastien's avatar Frederic Bastien

Add in the old gpu back-end, local_log_softmax_dnn as stabilize and fast_compile…

Add in the old gpu back-end, local_log_softmax_dnn as stabilize and fast_compile to have it on the GPU.
上级 2a756afd
...@@ -2526,7 +2526,8 @@ if True: ...@@ -2526,7 +2526,8 @@ if True:
out = as_cuda_ndarray_variable(out.dimshuffle(0, 1)) out = as_cuda_ndarray_variable(out.dimshuffle(0, 1))
return [out] return [out]
@register_opt('cudnn') @register_opt('cudnn', 'stabilize', 'fast_compile')
# We put fast_compile as otherwise it won't be on the GPU.
@local_optimizer([GpuElemwise, LogSoftmax]) @local_optimizer([GpuElemwise, LogSoftmax])
def local_log_softmax_dnn(node): def local_log_softmax_dnn(node):
# The log-softmax implementation is only available starting at cuDNN V3 # The log-softmax implementation is only available starting at cuDNN V3
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论