提交 0d640a74 authored 作者: Frederic Bastien's avatar Frederic Bastien 提交者: Mathieu Germain

Make compile with v4

上级 49c79247
......@@ -54,7 +54,10 @@ c_set_filterNd(CudaNdarray *var, cudnnFilterDescriptor_t desc) {
return -1;
}
int dim = CudaNdarray_NDIM(var);
cudnnStatus_t err = cudnnSetFilterNdDescriptor_v3(desc, CUDNN_DATA_FLOAT, dim,
cudnnStatus_t err = cudnnSetFilterNdDescriptor_v4(desc,
CUDNN_DATA_FLOAT,
CUDNN_TENSOR_NCHW,
dim,
CudaNdarray_HOST_DIMS(var));
if (err != CUDNN_STATUS_SUCCESS) {
PyErr_Format(PyExc_RuntimeError,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论