提交 552fb9df authored 作者: --global's avatar --global

Set requestedCount to 1

上级 755429dd
...@@ -72,7 +72,7 @@ APPLY_SPECIFIC(conv_fwd)(CudaNdarray *input, CudaNdarray *kerns, ...@@ -72,7 +72,7 @@ APPLY_SPECIFIC(conv_fwd)(CudaNdarray *input, CudaNdarray *kerns,
if (CHOOSE_ALGO_TIME) if (CHOOSE_ALGO_TIME)
{ {
// Time the different implementations to choose the best one // Time the different implementations to choose the best one
int requestedCount = 2; int requestedCount = 1;
int count; int count;
cudnnConvolutionFwdAlgoPerf_t choosen_algo_perf; cudnnConvolutionFwdAlgoPerf_t choosen_algo_perf;
err = cudnnFindConvolutionForwardAlgorithm(_handle, err = cudnnFindConvolutionForwardAlgorithm(_handle,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论