diff --git a/pkgs/development/rocm-modules/amdsmi/default.nix b/pkgs/development/rocm-modules/amdsmi/default.nix index af24c68b8fd2..16647a728bb8 100644 --- a/pkgs/development/rocm-modules/amdsmi/default.nix +++ b/pkgs/development/rocm-modules/amdsmi/default.nix @@ -6,6 +6,7 @@ cmake, pkg-config, libdrm, + python, wrapPython, autoPatchelfHook, }: @@ -67,6 +68,9 @@ stdenv.mkDerivation (finalAttrs: { ]; postInstall = '' + mkdir -p $out/${python.sitePackages} + ln -s $out/share/amd_smi/amdsmi $out/${python.sitePackages}/amdsmi + makeWrapperArgs=(--prefix LD_LIBRARY_PATH : ${lib.makeLibraryPath [ libdrm ]}) wrapPythonProgramsIn $out rm $out/bin/amd-smi diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 159bf67230b3..d9ae6e93952f 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -694,6 +694,12 @@ self: super: with self; { amcrest = callPackage ../development/python-modules/amcrest { }; + amdsmi = toPythonModule ( + pkgs.rocmPackages.amdsmi.override { + inherit python; + } + ); + ament-package = callPackage ../development/python-modules/ament-package { }; amply = callPackage ../development/python-modules/amply { };