qiskit-aer
qiskit-aer copied to clipboard
Attempt to Add ROCm Support to Qiskit Aer for AMD GPUs (issue #2113)
Summary
I made changes to build.yml, deploy.yml and added tests for the ROCm backend.
The changes include:
- CI/CD updates to build and deploy ROCm-enabled wheels.
- Unit tests to validate ROCm functionality.
- New configurations to to CI.
This is a work-in-progress (WIP), my first time attempting to write workflows and I would highly appreciate any feedback and performance optimizations for ROCm.
Details and comments
Modified Files:
- .github/workflows/build.yml → Adds wheel-gpu-rocm job for ROCm builds.
- .github/workflows/deploy.yml → For automatic deployment of ROCm wheels to PyPI.
Changes:
Adds a new job (wheel-gpu-rocm) that builds Qiskit Aer using AER_THRUST_BACKEND=ROCM. Ensures that ROCm-enabled wheels are packaged and uploaded alongside CUDA and CPU builds. Runs automated CI/CD testing for ROCm compatibility.
Looking for feedback on:
Whether the CI/CD workflow properly integrates with ROCm and any optimizations that can improve the build process. Additionally, I do not have an AMD card so I would appreciate if people could test this out on actual hardware.