提交 95afba78 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make sure to initialize the cudnn descriptors.

上级 cb326d31
......@@ -46,6 +46,11 @@ cudnnConvolutionDescriptor_t op%(id)d;
def c_init_code_struct(self, node, struct_id, sub):
return """
handle%(id)d = NULL;
input%(id)d = NULL;
output%(id)d = NULL;
kerns%(id)d = NULL;
op%(id)d = NULL;
if (cudnnCreate(&handle%(id)d) != CUDNN_STATUS_SUCCESS) {
PyErr_SetString(PyExc_RuntimeError, "could not create cudnn handle");
%(fail)s
......@@ -185,7 +190,7 @@ if (err%(name)s != CUDNN_STATUS_SUCCESS) {
fail=sub['fail'], id=sub['struct_id'], name=name)
def c_code_cache_version(self):
return (0,)
return (1,)
from theano.sandbox.cuda.opt import (local_optimizer, gpu_contiguous,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论