diff --git a/cmake/TaichiCore.cmake b/cmake/TaichiCore.cmake
index 5b715d4bd3def..63cfd46911a98 100644
--- a/cmake/TaichiCore.cmake
+++ b/cmake/TaichiCore.cmake
@@ -237,6 +237,9 @@ if(TI_WITH_LLVM)
     if (TI_WITH_AMDGPU)
         llvm_map_components_to_libnames(llvm_amdgpu_libs AMDGPU)
         add_subdirectory(taichi/rhi/amdgpu)
+        add_subdirectory(taichi/codegen/amdgpu)
+
+        target_link_libraries(${CORE_LIBRARY_NAME} PRIVATE amdgpu_codegen)
         target_link_libraries(${CORE_LIBRARY_NAME} PRIVATE amdgpu_rhi)
     endif()