gpu.yaml

---

# This way of detecting GPUs is portable and doesn't require root
- name: check for Nvidia GPU
  shell:
    cmd: awk 'BEGIN{result="false"} $0 ~ /0x10de/ {gsub(/\/vendor/, "/class", FILENAME); getline pciclass < FILENAME; if(pciclass ~ /^0x03/) result="true"}  END{print result}'  /sys/bus/pci/devices/*/vendor
  register: has_nvidia_gpu

  - name: Disable Nouveau Driver
    include_role:
      name: nouveau-blacklist
    when:
      (gpu | bool) and
      (has_nvidia_gpu.stdout == "true")

  - name: Install Nvidia Drivers
    include_role:
      name: nvidia-driver-install
    when:
      (gpu | bool) and
      (has_nvidia_gpu.stdout == "true")

  # TODO: Do we need OCI hooks or something else to enable radeon containers like nvidia?
  # TODO: What is needed if using docker instead of podman
  - name: Install Nvidia OCI Hooks
    include_role:
      name: nvidia-container-runtime-hook
    when:
      (gpu | bool) and
      (has_nvidia_gpu.stdout == "true") and
      (gpu_driver == 'cuda') and
      (podman | bool)

  - name: Install OpenCL Drivers
    include_role:
      name: opencl-driver-install
    when:
      (gpu | bool) and
      (gpu_type == 'nvidia') and
      (gpu_driver == 'opencl')

  - name: Install Radeon Drivers
    include_role:
      name: radeon-driver-install
    when:
      (gpu | bool) and
      (gpu_type == 'radeon')