rocmPackages_10: add rocPRIM, rocRAND, rocFFT

pkgs/development/compilers/llvm/common/openmp/default.nix

@@ -36,15 +36,28 @@ stdenv.mkDerivation (finalAttrs: {
           strictDeps = true;
           __structuredAttrs = true;
         }
-        ''
-          mkdir -p "$out"
-          cp -r ${monorepoSrc}/cmake "$out"
-          cp -r ${monorepoSrc}/openmp "$out"
-        ''
+        (
+          ''
+            mkdir -p "$out"
+            cp -r ${monorepoSrc}/cmake "$out"
+            cp -r ${monorepoSrc}/openmp "$out"
+          ''
+          + lib.optionalString (lib.versionAtLeast release_version "23") ''
+            mkdir -p "$out/llvm"
+            cp -r ${monorepoSrc}/llvm/cmake "$out/llvm"
+            cp -r ${monorepoSrc}/llvm/utils "$out/llvm"
+            cp -r ${monorepoSrc}/third-party "$out"
+            cp -r ${monorepoSrc}/runtimes "$out"
+          ''
+        )
     else
       src;
 
-  sourceRoot = "${finalAttrs.src.name}/openmp";
+  # LLVM 23 removed OpenMP's standalone build
+  # https://openmp.llvm.org/Building.html
+  sourceRoot =
+    "${finalAttrs.src.name}/"
+    + (if lib.versionAtLeast release_version "23" then "runtimes" else "openmp");
 
   outputs = [
     "out"
@@ -89,14 +102,19 @@ stdenv.mkDerivation (finalAttrs: {
     (lib.cmakeFeature "OPT_TOOL" "${llvm}/bin/opt")
     (lib.cmakeFeature "LINK_TOOL" "${llvm}/bin/llvm-link")
   ]
+  ++ lib.optionals (lib.versionAtLeast release_version "23") [
+    (lib.cmakeFeature "LLVM_ENABLE_RUNTIMES" "openmp")
+  ]
   ++ devExtraCmakeFlags;
 
   doCheck = false;
 
   checkTarget = "check-openmp";
 
   preCheck = ''
-    patchShebangs ../tools/archer/tests/deflake.bash
+    patchShebangs ${
+      if lib.versionAtLeast release_version "23" then "../../openmp" else ".."
+    }/tools/archer/tests/deflake.bash
   '';
 
   __structuredAttrs = true;

pkgs/development/rocm-modules/10/amdsmi/default.nix

@@ -0,0 +1,100 @@
+{
+  lib,
+  stdenv,
+  fetchFromGitHub,
+  rocmUpdateScript,
+  cmake,
+  pkg-config,
+  libdrm,
+  libnl,
+  libmnl,
+  python,
+  wrapPython,
+  autoPatchelfHook,
+  binutils,
+}:
+
+let
+  esmi_ib_src = fetchFromGitHub {
+    owner = "amd";
+    repo = "esmi_ib_library";
+    # The commit amdsmi pins (tag esmi_pkg_ver-5.2.1.1)
+    rev = "d494a3194ceb4cc4dbb2debf9fcbe8773c6d3bef";
+    hash = "sha256-StSYzyIujaH8UJIRpfJC0lf+oVT9fWflJgQONvRqu70=";
+  };
+in
+stdenv.mkDerivation (finalAttrs: {
+  pname = "amdsmi";
+  version = "10.1";
+  src = fetchFromGitHub {
+    owner = "ROCm";
+    repo = "rocm-systems";
+    tag = "therock-${finalAttrs.version}";
+    sparseCheckout = [
+      "projects/amdsmi"
+      "shared"
+    ];
+    hash = "sha256-CmCVPmqgU0HtqIBuJUs+Nr7LRKtngOaNNsWFek9Nino=";
+  };
+  sourceRoot = "${finalAttrs.src.name}/projects/amdsmi";
+
+  postPatch = ''
+    # Pre-populate esmi_ib_library so CMake does not FetchContent it.
+    cp -rf --no-preserve=mode ${esmi_ib_src} ./esmi_ib_library
+    mkdir -p ./esmi_ib_library/include/asm
+    cp ./include/amd_smi/impl/amd_hsmp.h ./esmi_ib_library/include/asm/amd_hsmp.h
+  '';
+
+  nativeBuildInputs = [
+    cmake
+    pkg-config
+    wrapPython
+    autoPatchelfHook
+    binutils # CMake resolves link-lib SONAMEs via `objdump -p`
+  ];
+
+  buildInputs = [
+    libdrm
+    libnl # pkg_check_modules libnl-3.0 + libnl-genl-3.0
+    libmnl # pkg_check_modules libmnl
+  ];
+
+  cmakeFlags = [
+    # Manually define CMAKE_INSTALL_<DIR>
+    # See: https://github.com/NixOS/nixpkgs/pull/197838
+    "-DCMAKE_INSTALL_BINDIR=bin"
+    "-DCMAKE_INSTALL_LIBDIR=lib"
+    "-DCMAKE_INSTALL_INCLUDEDIR=include"
+    # py-interface/CMakeLists.txt fails without a system python3 unless this is set.
+    "-DAMDSMI_SYSTEM_PYTHON_SITELIB=${placeholder "out"}/${python.sitePackages}"
+    # libamd_smi.so still exports rsmi_init, and its own call to it goes
+    # through the PLT. Once librocm_smi64 is loaded globally (torch's
+    # libtorch_hip needs it), that call binds to rocm_smi's copy, amdsmi's own
+    # state is never initialised, and enumeration returns no processors:
+    # torch.cuda.device_count() is then 0. Upstream links only the wheel's
+    # libamd_smi_python.so this way; do the same for the system library.
+    "-DCMAKE_SHARED_LINKER_FLAGS=-Wl,-Bsymbolic-functions"
+  ];
+
+  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
+    ln -sf $out/libexec/amdsmi_cli/amdsmi_cli.py $out/bin/amd-smi
+  '';
+
+  passthru.updateScript = rocmUpdateScript { inherit finalAttrs; };
+
+  meta = {
+    description = "System management interface for AMD GPUs supported by ROCm";
+    homepage = "https://github.com/ROCm/rocm-systems/tree/develop/projects/amdsmi";
+    license = lib.licenses.mit;
+    maintainers = with lib.maintainers; [ lovesegfault ];
+    teams = [ lib.teams.rocm ];
+    platforms = [ "x86_64-linux" ];
+    mainProgram = "amd-smi";
+  };
+})

pkgs/development/rocm-modules/10/aqlprofile/default.nix

@@ -0,0 +1,42 @@
+{
+  lib,
+  stdenv,
+  clr,
+  cmake,
+  fetchFromGitHub,
+  rocmUpdateScript,
+}:
+
+stdenv.mkDerivation (finalAttrs: {
+  pname = "aqlprofile";
+  version = "10.1";
+
+  src = fetchFromGitHub {
+    owner = "ROCm";
+    repo = "rocm-systems";
+    tag = "therock-${finalAttrs.version}";
+    sparseCheckout = [
+      "projects/aqlprofile"
+      "shared"
+    ];
+    hash = "sha256-7yeEJoyShSCwy1MTCeArYBabh5qTVm5uZL8N74rvxPA=";
+  };
+  sourceRoot = "${finalAttrs.src.name}/projects/aqlprofile";
+
+  env.CXXFLAGS = "-DROCP_LD_AQLPROFILE=1";
+
+  nativeBuildInputs = [
+    cmake
+    clr
+  ];
+
+  passthru.updateScript = rocmUpdateScript { inherit finalAttrs; };
+
+  meta = {
+    description = "AQLPROFILE library for AMD HSA runtime API extension support";
+    homepage = "https://github.com/ROCm/rocm-systems/tree/develop/projects/aqlprofile";
+    license = lib.licenses.mit;
+    teams = [ lib.teams.rocm ];
+    platforms = lib.platforms.linux;
+  };
+})

pkgs/development/rocm-modules/10/clr/cmake-find-x11-libgl.patch

@@ -0,0 +1,40 @@
+diff --git a/rocclr/cmake/ROCclr.cmake b/rocclr/cmake/ROCclr.cmake
+index 3f233b72f..67bdc62ee 100644
+--- a/rocclr/cmake/ROCclr.cmake
++++ b/rocclr/cmake/ROCclr.cmake
+@@ -44,6 +44,19 @@ find_package(Threads REQUIRED)
+
+ find_package(AMD_OPENCL)
+
++# Find X11 package
++find_package(X11 REQUIRED)
++if(NOT X11_FOUND)
++    message(FATAL_ERROR "X11 libraries not found")
++endif()
++
++# Find OpenGL package
++find_package(OpenGL REQUIRED)
++if(NOT OpenGL_FOUND)
++    message(FATAL_ERROR "OpenGL not found")
++endif()
++
++
+ add_library(rocclr STATIC)
+
+ include(ROCclrCompilerOptions)
+@@ -123,9 +136,14 @@ target_include_directories(rocclr PUBLIC
+   ${ROCCLR_SRC_DIR}/device
+   ${ROCCLR_SRC_DIR}/elf
+   ${ROCCLR_SRC_DIR}/include
++  ${X11_INCLUDE_DIR}
++  ${OPENGL_INCLUDE_DIR}
+   ${AMD_OPENCL_INCLUDE_DIRS})
+
+-target_link_libraries(rocclr PUBLIC Threads::Threads)
++target_link_libraries(rocclr PUBLIC
++  Threads::Threads
++  ${X11_LIBRARIES}
++  ${OPENGL_LIBRARIES})
+ # IPC on Windows is not supported
+ if(UNIX)
+   target_link_libraries(rocclr PUBLIC rt)