From c3939fa2cff10cbeac9b5ccace8f3f43c340b4ea Mon Sep 17 00:00:00 2001 From: Yureka Date: Mon, 12 Jan 2026 11:26:59 +0100 Subject: [PATCH] onnxruntime: use migraphx for rocmSupport --- pkgs/by-name/on/onnxruntime/package.nix | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/pkgs/by-name/on/onnxruntime/package.nix b/pkgs/by-name/on/onnxruntime/package.nix index 327edd0b40bd..35d2570bf307 100644 --- a/pkgs/by-name/on/onnxruntime/package.nix +++ b/pkgs/by-name/on/onnxruntime/package.nix @@ -153,9 +153,6 @@ effectiveStdenv.mkDerivation (finalAttrs: { ] ++ lib.optionals isCudaJetson [ cudaPackages.autoAddCudaCompatRunpath - ] - ++ lib.optionals rocmSupport [ - perl # for tools/ci_build/hipify-perl ]; buildInputs = [ @@ -195,6 +192,7 @@ effectiveStdenv.mkDerivation (finalAttrs: { ++ lib.optionals rocmSupport [ rocmPackages.clr rocmPackages.hipblas + rocmPackages.hipblaslt rocmPackages.hipcub rocmPackages.hipfft rocmPackages.hiprand @@ -204,6 +202,7 @@ effectiveStdenv.mkDerivation (finalAttrs: { rocmPackages.rocrand rocmPackages.rocthrust rocmPackages.miopen + rocmPackages.migraphx rocmPackages.rccl rocmPackages.rocm-smi rocmPackages.roctracer @@ -255,13 +254,9 @@ effectiveStdenv.mkDerivation (finalAttrs: { (lib.cmakeBool "onnxruntime_USE_FULL_PROTOBUF" withFullProtobuf) (lib.cmakeBool "onnxruntime_USE_CUDA" cudaSupport) (lib.cmakeBool "onnxruntime_USE_NCCL" (cudaSupport && ncclSupport)) - (lib.cmakeBool "onnxruntime_USE_ROCM" rocmSupport) + (lib.cmakeBool "onnxruntime_USE_MIGRAPHX" rocmSupport) (lib.cmakeBool "onnxruntime_ENABLE_LTO" (!cudaSupport || cudaPackages.cudaOlder "12.8")) ] - ++ lib.optionals (effectiveStdenv.cc.isClang || rocmSupport) [ - # Disable -Werror from COMPILE_WARNING_AS_ERROR target property - "--compile-no-warning-as-error" - ] ++ lib.optionals pythonSupport [ (lib.cmakeBool "onnxruntime_ENABLE_PYTHON" true) ]