提交 ee0ab1cb authored 作者: Frederic's avatar Frederic

Fix compilation crash with cudnn on python3

上级 2f612611
......@@ -149,6 +149,13 @@ class DnnVersion(GpuOp):
def c_libraries(self):
return ['cudnn']
def c_support_code(self):
return """
#if PY_MAJOR_VERSION >= 3
#define PyInt_FromLong PyLong_FromLong
#endif
"""
def make_node(self):
return Apply(self, [], [Generic()()])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论