mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm]: Updates hip headers path for ROCm 6.0

Browse Source
This commit is contained in:
Rahul Batra 2023-12-27 22:43:47 +00:00
parent b7a7f0bd80
commit f997609e76
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
jaxlib/gpu/vendor.h
View File

@ -305,9 +305,9 @@ constexpr uint32_t kNumThreadsPerWarp = 32;
#elif defined(JAX_GPU_HIP)
#include "rocm/include/hip/hip_runtime_api.h"
#include "rocm/include/hipblas.h"
#include "rocm/include/hipsolver.h"
#include "rocm/include/hipsparse.h"
#include "rocm/include/hipblas/hipblas.h"
#include "rocm/include/hipsolver/hipsolver.h"
#include "rocm/include/hipsparse/hipsparse.h"
#define JAX_GPU_NAMESPACE hip
#define JAX_GPU_PREFIX "hip"