defined(USE_ROCM)"
Or if it is desired to take the code path for ROCm/HIP only for
specific HIP versions:
"#if (defined(CUDA_VERSION) && CUDA_VERSION >= 11000) ||
(defined(USE_ROCM) && ROCM_VERSION >= 40300)"
Refer to CUDA Semantics doc
For any sections not listed here, please refer to the CUDA semantics
doc: CUDA semantics
Enabling kernel asserts
Kernel asserts are supported on ROCm, but they are disabled due to
performance overhead. It can be enabled by recompiling the PyTorch
from source.
Please add below line as an argument to cmake command parameters:
-DROCM_FORCE_ENABLE_GPU_ASSERTS:BOOL=ON