Fix lint

apache · Apr 26, 2023 · f2a94ea · f2a94ea
1 parent 0a9859b
commit f2a94ea
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/target/opt/build_metal_off.cc b/src/target/opt/build_metal_off.cc
@@ -26,10 +26,11 @@
 namespace tvm {
 namespace runtime {
 
-Module MetalModuleCreate(std::string data, std::string fmt,
-                         std::unordered_map<std::string, FunctionInfo> fmap, std::string source) {
+Module MetalModuleCreate(std::unordered_map<std::string, std::string> smap,
+                         std::unordered_map<std::string, FunctionInfo> fmap, std::string fmt,
+                         std::string source) {
   LOG(WARNING) << "Metal runtime not enabled, return a source module...";
-  return codegen::DeviceSourceModuleCreate(data, fmt, fmap, "metal");
+  return codegen::DeviceSourceModuleCreate(source, fmt, fmap, "metal");
 }
 
 }  // namespace runtime

diff --git a/src/target/source/codegen_metal.cc b/src/target/source/codegen_metal.cc
@@ -25,6 +25,7 @@
 #include <algorithm>
 #include <string>
 #include <vector>
+#include <unordered_map>
 
 #include "../../runtime/metal/metal_module.h"
 #include "../../runtime/thread_storage_scope.h"