This commit is contained in:
Cha Zhang 2017-02-05 08:54:03 -08:00
Родитель 8eec5ed546
Коммит b363e2269d
1 изменённых файлов: 4 добавлений и 1 удалений

Просмотреть файл

@ -421,7 +421,10 @@ private:
algo.Algo.memory = 0;
algo.Algo.status = CUDNN_STATUS_SUCCESS;
algo.NoWorkspaceAlgo = noMemAlgo;
algo.autotuningState = AutotuningState::runState;
if (algo.autotuningState == AutotuningState::initState)
algo.autotuningState = AutotuningState::pendingTuningState;
else
algo.autotuningState = AutotuningState::runState;
return;
}
CUDNN_CALL(err);