add macro to control amdgpu-related header file

taichi-dev · Dec 31, 2022 · cc46700 · cc46700
1 parent 96e65ee
commit cc46700
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/taichi/rhi/amdgpu/amdgpu_caching_allocator.cpp b/taichi/rhi/amdgpu/amdgpu_caching_allocator.cpp
@@ -3,7 +3,6 @@
 namespace taichi {
 namespace lang {
 namespace amdgpu {
-
 AmdgpuCachingAllocator::AmdgpuCachingAllocator(LlvmDevice *device)
     : device_(device) {
 }

diff --git a/taichi/runtime/llvm/llvm_runtime_executor.cpp b/taichi/runtime/llvm/llvm_runtime_executor.cpp
@@ -6,14 +6,14 @@
 #include "taichi/rhi/cuda/cuda_device.h"
 #include "taichi/platform/cuda/detect_cuda.h"
 #include "taichi/rhi/cuda/cuda_driver.h"
-#include "taichi/rhi/amdgpu/amdgpu_driver.h"
-#include "taichi/rhi/amdgpu/amdgpu_device.h"
 
 #if defined(TI_WITH_CUDA)
 #include "taichi/rhi/cuda/cuda_context.h"
 #endif
 
 #if defined(TI_WITH_AMDGPU)
+#include "taichi/rhi/amdgpu/amdgpu_driver.h"
+#include "taichi/rhi/amdgpu/amdgpu_device.h"
 #include "taichi/rhi/amdgpu/amdgpu_context.h"
 #endif