diff --git a/src/target/llvm/llvm_instance.cc b/src/target/llvm/llvm_instance.cc index 9e88222059..08ba34cc73 100644 --- a/src/target/llvm/llvm_instance.cc +++ b/src/target/llvm/llvm_instance.cc @@ -42,7 +42,11 @@ #include #include #include +#if TVM_LLVM_VERSION >= 180 +#include +#else #include +#endif #include #include #include diff --git a/src/target/llvm/llvm_module.cc b/src/target/llvm/llvm_module.cc index 4823f2c9ea..62ea797edd 100644 --- a/src/target/llvm/llvm_module.cc +++ b/src/target/llvm/llvm_module.cc @@ -41,7 +41,11 @@ #include #include #include +#if TVM_LLVM_VERSION >= 180 +#include +#else #include +#endif #include #include #include