提交 fec62ac0 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Bump the C version for the modified Crossentropy op.

上级 c3f03aab
...@@ -270,7 +270,7 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias(Op): ...@@ -270,7 +270,7 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias(Op):
return sio.getvalue() return sio.getvalue()
def c_code_cache_version(self): def c_code_cache_version(self):
return (5,) return (6,)
def c_compiler(self): def c_compiler(self):
return NVCC_compiler return NVCC_compiler
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论