1 | #include "taichi/platform/amdgpu/detect_amdgpu.h" |
---|---|
2 | |
3 | #if defined(TI_WITH_AMDGPU) |
4 | #include "taichi/rhi/amdgpu/amdgpu_driver.h" |
5 | #endif |
6 | |
7 | namespace taichi { |
8 | |
9 | bool is_rocm_api_available() { |
10 | #if defined(TI_WITH_AMDGPU) |
11 | return lang::AMDGPUDriver::get_instance_without_context().detected(); |
12 | #else |
13 | return false; |
14 | #endif |
15 | } |
16 | |
17 | } // namespace taichi |
18 |