cpplint

ylc · Jan 17, 2022 · ebed032 · ebed032
1 parent 834f54a
commit ebed032
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/runtime/contrib/cudnn/cudnn_utils.cc b/src/runtime/contrib/cudnn/cudnn_utils.cc
@@ -20,12 +20,16 @@
 /*!
  * \file Use external cudnn utils function
  */
+
 #include "cudnn_utils.h"
 
 #include <dmlc/thread_local.h>
 #include <tvm/runtime/data_type.h>
 #include <tvm/runtime/registry.h>
 
+#include <string>
+#include <vector>
+
 namespace tvm {
 namespace contrib {
 

diff --git a/src/runtime/contrib/cudnn/cudnn_utils.h b/src/runtime/contrib/cudnn/cudnn_utils.h
@@ -28,6 +28,8 @@
 #include <tvm/runtime/device_api.h>
 #include <tvm/runtime/logging.h>
 
+#include <string>
+
 #include "../../cuda/cuda_common.h"
 
 namespace tvm {