---
- name: Check nvidia-smi status
  ansible.builtin.command: nvidia-smi --query-gpu=name,memory.total,driver_version --format=csv,noheader
  register: nvidia_runtime_smi
  changed_when: false
  failed_when: false

- name: Fail when post-install validation is required and nvidia-smi is unavailable
  ansible.builtin.fail:
    msg: >-
      nvidia-smi is unavailable after installation.
      This usually means a reboot is still required or the selected driver is incompatible.
  when:
    - nvidia_runtime_validate_after_install | bool
    - nvidia_runtime_smi.rc != 0

- name: Capture GPU info lines
  ansible.builtin.set_fact:
    nvidia_runtime_gpu_lines: "{{ nvidia_runtime_smi.stdout_lines | default([]) }}"
  when: nvidia_runtime_smi.rc == 0