onnxruntime: use migraphx for rocmSupport

This commit is contained in:
Yureka
2026-01-12 11:26:59 +01:00
parent 4262a1f3bd
commit c3939fa2cf
+3 -8
View File
@@ -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)
]