提交 e51b6a22 authored 作者: notoraptor's avatar notoraptor

Fix typos

上级 bf4db351
...@@ -3,7 +3,7 @@ cudnnTensorDescriptor_t APPLY_SPECIFIC(input); ...@@ -3,7 +3,7 @@ cudnnTensorDescriptor_t APPLY_SPECIFIC(input);
cudnnTensorDescriptor_t APPLY_SPECIFIC(output); cudnnTensorDescriptor_t APPLY_SPECIFIC(output);
cudnnFilterDescriptor_t APPLY_SPECIFIC(kerns); cudnnFilterDescriptor_t APPLY_SPECIFIC(kerns);
static int c_check_groups_for_conv(cudnnConvolutionDescriptor_t desc, int groups) { static int c_get_groups_for_conv(cudnnConvolutionDescriptor_t desc, int groups) {
#if CUDNN_MAJOR >= 7 #if CUDNN_MAJOR >= 7
int desc_groups; int desc_groups;
if (groups > 1) { if (groups > 1) {
...@@ -83,7 +83,7 @@ typedef std::unordered_map<std::string, AlgoRec> AlgoCache; ...@@ -83,7 +83,7 @@ typedef std::unordered_map<std::string, AlgoRec> AlgoCache;
#endif #endif
#include "pthread.h" #include "pthread.h"
#line 73 "dnn_conv_base.c" #line 87 "dnn_conv_base.c"
pthread_mutex_t algoMutex; pthread_mutex_t algoMutex;
AlgoCache algoCache; AlgoCache algoCache;
......
...@@ -167,7 +167,7 @@ APPLY_SPECIFIC(conv_fwd)(PyGpuArrayObject *input, PyGpuArrayObject *kerns, ...@@ -167,7 +167,7 @@ APPLY_SPECIFIC(conv_fwd)(PyGpuArrayObject *input, PyGpuArrayObject *kerns,
return 0; return 0;
} }
int groups = c_check_groups_for_conv(desc, params->num_groups); int groups = c_get_groups_for_conv(desc, params->num_groups);
if (groups == -1) if (groups == -1)
return 1; return 1;
if (c_set_tensor_for_conv(input, APPLY_SPECIFIC(input), groups) == -1) if (c_set_tensor_for_conv(input, APPLY_SPECIFIC(input), groups) == -1)
......
...@@ -132,7 +132,7 @@ APPLY_SPECIFIC(conv_gi)(PyGpuArrayObject *kerns, PyGpuArrayObject *output, ...@@ -132,7 +132,7 @@ APPLY_SPECIFIC(conv_gi)(PyGpuArrayObject *kerns, PyGpuArrayObject *output,
return 0; return 0;
} }
int groups = c_check_groups_for_conv(desc, params->num_groups); int groups = c_get_groups_for_conv(desc, params->num_groups);
if (groups == -1) if (groups == -1)
return 1; return 1;
if (c_set_tensor_for_conv(output, APPLY_SPECIFIC(output), groups) == -1) if (c_set_tensor_for_conv(output, APPLY_SPECIFIC(output), groups) == -1)
......
...@@ -119,7 +119,7 @@ APPLY_SPECIFIC(conv_gw)(PyGpuArrayObject *input, PyGpuArrayObject *output, ...@@ -119,7 +119,7 @@ APPLY_SPECIFIC(conv_gw)(PyGpuArrayObject *input, PyGpuArrayObject *output,
return 0; return 0;
} }
int groups = c_check_groups_for_conv(desc, params->num_groups); int groups = c_get_groups_for_conv(desc, params->num_groups);
if (groups == -1) if (groups == -1)
return 1; return 1;
if (c_set_tensor_for_conv(input, APPLY_SPECIFIC(input), groups) == -1) if (c_set_tensor_for_conv(input, APPLY_SPECIFIC(input), groups) == -1)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论