depot/third_party/nixpkgs/pkgs/development/rocm-modules/6/clr/test-rocm-smi.nix

{
  lib,
  makeImpureTest,
  clinfo,
  clr,
  rocm-smi,
}:

makeImpureTest {
  name = "rocm-smi";
  testedPackage = "rocmPackages_6.clr";
  nativeBuildInputs = [
    clinfo
    rocm-smi
  ];
  OCL_ICD_VENDORS = "${clr.icd}/etc/OpenCL/vendors";

  testScript = ''
    # Test fails if the number of platforms is 0
    clinfo | grep -E 'Number of platforms * [1-9]'
    rocm-smi | grep -A1 GPU
  '';

  meta = with lib; {
    maintainers = teams.rocm.members;
  };
}