[incubator-mxnet] branch master updated: warning autotune show once (#9132)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new b59943e  warning autotune show once (#9132)
b59943e is described below

commit b59943e9ca9dd4d603ffd8b2464d431dbd1d9bdb
Author: Hu Shiwen <ya...@gmail.com>
AuthorDate: Wed Dec 20 09:50:35 2017 +0800

    warning autotune show once (#9132)
    
    * warning_autotune
    
    * fix
---
 src/operator/nn/cudnn/cudnn_algoreg-inl.h | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/operator/nn/cudnn/cudnn_algoreg-inl.h b/src/operator/nn/cudnn/cudnn_algoreg-inl.h
index 3eca4c2..e029c83 100644
--- a/src/operator/nn/cudnn/cudnn_algoreg-inl.h
+++ b/src/operator/nn/cudnn/cudnn_algoreg-inl.h
@@ -109,10 +109,14 @@ class CuDNNAlgoReg {
                    "this can take a while... (setting env variable "
                    "MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable)";
       if (reg_.size() >= 1000) {
-        LOG(INFO)
+        // Many people are very concerned about this warning, so change the warning once.
+        if (!is_warning_autotune_) {
+          LOG(INFO)
             << "If you see this message in the middle of training, you are "
-               "probably using bucketing. Consider setting env variable "
-               "MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable cudnn tuning.";
+            "probably using bucketing. Consider setting env variable "
+            "MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable cudnn tuning.";
+          is_warning_autotune_ = true;
+        }
       }
     }
     reg_[key].fwd = fwd;
@@ -166,6 +170,7 @@ class CuDNNAlgoReg {
 
   std::mutex lock_;
   std::unordered_map<ParamKey, CudnnAlgorithms, ParamHash> reg_;
+  bool is_warning_autotune_ = false;
 };
 
 typedef CuDNNAlgoReg<ConvolutionParam> CuDNNConvAlgoReg;

-- 
To stop receiving notification emails like this one, please contact
['"commits@mxnet.apache.org" <co...@mxnet.apache.org>'].