Update cudnn_algoreg-inl.h (#7988)

apache · Oct 21, 2017 · ad20d91 · ad20d91
1 parent 72b0051
commit ad20d91
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/cudnn_algoreg-inl.h b/src/operator/cudnn_algoreg-inl.h
@@ -102,7 +102,7 @@ class CuDNNAlgoReg {
     ParamKey key{param, in_shape[0], in_shape[1], out_shape[0], cudnn_data_type,
                  cudnn_forward_compute_type, cudnn_backward_compute_type, sm_arch};
     std::lock_guard<std::mutex> guard(lock_);
-    if (reg_.size() % 50 == 0) {
+    if (param.cudnn_tune.value() && reg_.size() % 50 == 0) {
       LOG(INFO) << "Running performance tests to find the best convolution "
                    "algorithm, "
                    "this can take a while... (setting env variable "