[MSVC] Make able to compile with MSVC (#6341)

* fix: make suitable for msvc, clang * clang-format * refactor: use DMLC_ATTRIBUTE
apache · Aug 26, 2020 · 4910c8c · 4910c8c
1 parent 942c90b
commit 4910c8c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/target/llvm/codegen_hexagon.cc b/src/target/llvm/codegen_hexagon.cc
@@ -636,7 +636,7 @@ bool UsesExportABI(const PrimFunc& f) {
   return false;
 }
 
-__attribute__((unused)) std::ostream& operator<<(std::ostream& os, const llvm::Module& m) {
+DMLC_ATTRIBUTE_UNUSED std::ostream& operator<<(std::ostream& os, const llvm::Module& m) {
   std::string ms;
   llvm::raw_string_ostream sos(ms);
   sos << m;

diff --git a/src/te/autodiff/ad_simplify.cc b/src/te/autodiff/ad_simplify.cc
@@ -52,6 +52,7 @@
 #include <tvm/tir/analysis.h>
 #include <tvm/tir/stmt_functor.h>
 
+#include <iterator>
 #include <memory>
 #include <utility>