提交 24e33a70 authored 作者: Frederic's avatar Frederic 提交者: Arnaud Bergeron

[CRASH] fix cudnn r2 softmax compilation

上级 25c9209d
......@@ -1003,6 +1003,7 @@ class GpuDnnSoftmaxBase(DnnBase):
def __init__(self, tensor_format, algo, mode):
assert(tensor_format in ('bc01', 'b01c'))
DnnBase.__init__(self)
self.tensor_format = tensor_format
assert(algo in ('fast', 'accurate'))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论