qiskit-aer icon indicating copy to clipboard operation
qiskit-aer copied to clipboard

Attempt to Add ROCm Support to Qiskit Aer for AMD GPUs (issue #2113)

Open SyedMustafaAhmad opened this issue 8 months ago • 0 comments

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.

SyedMustafaAhmad avatar Feb 04 '25 22:02 SyedMustafaAhmad