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

Remove duplicated section in DnnBase.

上级 0550874d
...@@ -133,20 +133,6 @@ class DnnBase(GpuOp, COp): ...@@ -133,20 +133,6 @@ class DnnBase(GpuOp, COp):
def c_libraries(self): def c_libraries(self):
return ['cudnn'] return ['cudnn']
def c_init_code(self):
if PY3:
error_out = "NULL"
else:
error_out = ""
return ["""{
cudnnStatus_t err;
if ((err = cudnnCreate(&_handle)) != CUDNN_STATUS_SUCCESS) {
PyErr_Format(PyExc_RuntimeError, "could not create cuDNN handle: %%s",
cudnnGetErrorString(err));
return %s;
}
}""" % (error_out,)]
class DnnVersion(GpuOp): class DnnVersion(GpuOp):
def c_compiler(self): def c_compiler(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论