- name: Add ROCm repository
  become: true
  ansible.builtin.yum_repository:
    name: rocm
    description: ROCm Repository
    baseurl: https://repo.radeon.com/rocm/el9/latest/main/
    gpgcheck: true
    gpgkey: https://repo.radeon.com/rocm/rocm.gpg.key
    enabled: true
    state: present

- name: Add AMD Graphics repository
  become: true
  ansible.builtin.yum_repository:
    name: rocmgraphics
    description: AMD Graphics Repository
    baseurl: https://repo.radeon.com/graphics/latest/el/9.6/main/x86_64/
    gpgkey: https://repo.radeon.com/rocm/rocm.gpg.key
    gpgcheck: true
    enabled: true
    state: present

- name: Install rocm-dkms and dependencies
  become: true
  ansible.builtin.package:
    name:
      - rocm
      - rocm-developer-tools
      - hipblas-devel
      - hip-devel
      - rocwmma-devel
      - rocm-opencl-devel
    state: present