[ROCm] add support to ROCm 6.0 and MI300 (#2274)

This commit is contained in:
Hongxia Yang
2024-01-26 15:41:10 -05:00
committed by GitHub
parent 5265631d15
commit 6b7de1a030
8 changed files with 96 additions and 13 deletions

View File

@@ -51,6 +51,8 @@ if _is_hip():
"Cannot find ROCM_HOME. ROCm must be available to build the package."
)
NVCC_FLAGS += ["-DUSE_ROCM"]
NVCC_FLAGS += [f"-U__HIP_NO_HALF_CONVERSIONS__"]
NVCC_FLAGS += [f"-U__HIP_NO_HALF_OPERATORS__"]
if _is_cuda() and CUDA_HOME is None:
raise RuntimeError(