diff --git a/src/arithmetic/ir_mutator_with_analyzer.h b/src/arithmetic/ir_mutator_with_analyzer.h index 475802020a0b4..9e3a86bb52801 100644 --- a/src/arithmetic/ir_mutator_with_analyzer.h +++ b/src/arithmetic/ir_mutator_with_analyzer.h @@ -26,6 +26,7 @@ #include #include +#include namespace tvm { namespace arith { diff --git a/src/pass/lower_intrin.cc b/src/pass/lower_intrin.cc index 3e95bce52f70d..bbd6c35725490 100644 --- a/src/pass/lower_intrin.cc +++ b/src/pass/lower_intrin.cc @@ -36,6 +36,7 @@ namespace ir { class IntrinInjecter : public tvm::arith::IRMutatorWithAnalyzer { public: using IRMutatorWithAnalyzer::VisitStmt_; + using IRMutatorWithAnalyzer::VisitExpr_; IntrinInjecter(arith::Analyzer* analyzer, std::string target) : IRMutatorWithAnalyzer(analyzer) {