depot/third_party/nixpkgs/pkgs/tools/system/rocm-smi/default.nix

{ lib, buildPythonApplication, fetchFromGitHub }:

buildPythonApplication rec {
  pname = "rocm-smi";
  version = "3.10.0";

  src = fetchFromGitHub {
    owner = "RadeonOpenCompute";
    repo = "ROC-smi";
    rev = "rocm-${version}";
    hash = "sha256-0QqaBMkqRVEl89x3hvWQGAgt7LbtMZPhuf7KenQYHaQ=";
  };

  format = "other";

  dontConfigure = true;
  dontBuild = true;

  installPhase = ''
    install -Dm0755 rocm_smi.py $out/bin/rocm-smi
  '';

  meta = with lib; {
    description = "System management interface for AMD GPUs supported by ROCm";
    homepage = "https://github.com/RadeonOpenCompute/ROC-smi";
    license = with licenses; [ mit ];
    maintainers = with maintainers; [ danieldk ];
    platforms = platforms.linux;
  };
}