Skip to content

Implement HIP build version using TORCH_HIP_VERSION#15

Open
rraminen wants to merge 2 commits into
ROCm:mainfrom
rraminen:implement_TORCH_HIP_VERSION
Open

Implement HIP build version using TORCH_HIP_VERSION#15
rraminen wants to merge 2 commits into
ROCm:mainfrom
rraminen:implement_TORCH_HIP_VERSION

Conversation

@rraminen

Copy link
Copy Markdown
  • Use PyTorch’s TORCH_HIP_VERSION in cuda_version() on ROCm.
  • Extend _check_cuda_version() to compare that value to torch.version.hip and raise a clear error if PyTorch and TorchAudio were built for different ROCm HIP versions, mirroring the existing CUDA version check.
  • Relies on TORCH_HIP_VERSION being defined at compile time (see pytorch/pytorch#182810 for TORCH_HIP_VERSION support).

@rraminen rraminen marked this pull request as ready for review May 19, 2026 18:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant