From bfae1bafcac9e8840c0b735a3ad7dd44b0797b3e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Robert=20Sch=C3=BCtz?= Date: Sat, 14 Jun 2025 16:28:04 -0700 Subject: [PATCH] rocmPackages.migraphx: override oneDNN --- .../development/rocm-modules/6/migraphx/default.nix | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/pkgs/development/rocm-modules/6/migraphx/default.nix b/pkgs/development/rocm-modules/6/migraphx/default.nix index 5501e6044b61..54620d320076 100644 --- a/pkgs/development/rocm-modules/6/migraphx/default.nix +++ b/pkgs/development/rocm-modules/6/migraphx/default.nix @@ -20,7 +20,7 @@ nlohmann_json, msgpack, sqlite, - oneDNN_2, + oneDNN, blaze, texliveSmall, doxygen, @@ -51,6 +51,15 @@ let ] ) ); + oneDNN' = oneDNN.overrideAttrs rec { + version = "2.7.5"; + src = fetchFromGitHub { + owner = "oneapi-src"; + repo = "oneDNN"; + tag = "v${version}"; + hash = "sha256-oMPBORAdL2rk2ewyUrInYVHYBRvuvNX4p4rwykO3Rhs="; + }; + }; in stdenv.mkDerivation (finalAttrs: { pname = "migraphx"; @@ -105,7 +114,7 @@ stdenv.mkDerivation (finalAttrs: { nlohmann_json msgpack sqlite - oneDNN_2 + oneDNN' blaze python3Packages.pybind11 python3Packages.onnx