add macro

taichi-dev · Jan 16, 2023 · 68f871e · 68f871e
1 parent fe6d019
commit 68f871e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/taichi/runtime/llvm/llvm_context.cpp b/taichi/runtime/llvm/llvm_context.cpp
@@ -576,6 +576,7 @@ void TaichiLLVMContext::link_module_with_cuda_libdevice(
 void TaichiLLVMContext::link_module_with_amdgpu_libdevice(
     std::unique_ptr<llvm::Module> &module) {
   TI_ASSERT(arch_ == Arch::amdgpu);
+#if defined(TI_WITH_AMDGPU)
   auto isa_version = AMDGPUContext::get_instance().get_mcpu().substr(3, 4);
   std::string libdevice_files[] = {"ocml.bc",
                                    "oclc_wavefrontsize64_off.bc",
@@ -621,6 +622,7 @@ void TaichiLLVMContext::link_module_with_amdgpu_libdevice(
         func->setLinkage(llvm::Function::InternalLinkage);
     }
   }
+#endif
 }
 
 void TaichiLLVMContext::add_struct_module(std::unique_ptr<Module> module,