From 0074a31d5e80d17241f5c1ae37beffa56ac3866c Mon Sep 17 00:00:00 2001
From: Pui Yong Qing <puiyongqing@gmail.com>
Date: Mon, 20 Oct 2025 17:03:16 +0800
Subject: [PATCH] rocmPackages.amdsmi: fix libdrm runtime loading and set
 mainProgram

---
 pkgs/development/rocm-modules/6/amdsmi/default.nix | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pkgs/development/rocm-modules/6/amdsmi/default.nix b/pkgs/development/rocm-modules/6/amdsmi/default.nix
index 6f26889ab01b..a145e4d61ded 100644
--- a/pkgs/development/rocm-modules/6/amdsmi/default.nix
+++ b/pkgs/development/rocm-modules/6/amdsmi/default.nix
@@ -70,6 +70,7 @@ stdenv.mkDerivation (finalAttrs: {
   ];
 
   postInstall = ''
+    makeWrapperArgs=(--prefix LD_LIBRARY_PATH : ${lib.makeLibraryPath [ libdrm ]})
     wrapPythonProgramsIn $out
     rm $out/bin/amd-smi
     ln -sf $out/libexec/amdsmi_cli/amdsmi_cli.py $out/bin/amd-smi
@@ -88,5 +89,6 @@ stdenv.mkDerivation (finalAttrs: {
     maintainers = with maintainers; [ lovesegfault ];
     teams = [ teams.rocm ];
     platforms = [ "x86_64-linux" ];
+    mainProgram = "amd-smi";
   };
 })