Struct OnnxRuntimeInstaller 

pub struct OnnxRuntimeInstaller { /* private fields */ }

The ONNX Runtime installer.

Implementations

impl OnnxRuntimeInstaller

pub fn new(config: OnnxRuntimeConfig) -> Self

Create a new ONNX Runtime installer with the given config.

pub fn with_defaults() -> Self

Create with default config.

pub fn dependencies(&self) -> &[&str]

Get the list of required dependencies.

ONNX Runtime depends on ROCm.

pub fn format_rocm_version_for_ort(&self, rocm_version: &str) -> String

Format ROCm version for ONNX Runtime (e.g., “7.2.0” -> “70200”).

The original script formats as Mmmpp: printf "%d%02d%02d" major minor patch.

pub fn build_prebuilt_install_command(&self) -> ShellCommand

Construct the pip install command for prebuilt onnxruntime-rocm wheel.

The original script tries this first: pip install --upgrade --prefer-binary onnxruntime-rocm

pub fn build_migraphx_wheel_url(&self) -> String

Construct the AMD manylinux repo URL for onnxruntime-migraphx.

Pattern: https://repo.radeon.com/rocm/manylinux/rocm-rel-{release}/

The URL points to a specific wheel matching the Python version and ROCm release. ROCm 7.2.4 ships onnxruntime_migraphx 1.23.2.

pub fn build_migraphx_install_command(&self) -> ShellCommand

Construct the pip install command for onnxruntime-migraphx from AMD repo.

This is the default install method — the AMD-provided wheel includes the MIGraphX execution provider and is built against the matching ROCm version, avoiding ABI incompatibility issues with PyPI’s onnxruntime-rocm.

pub fn build_model_optimizer_command(&self, model_path: &str) -> ShellCommand

Construct the Python command to run ORT graph optimization on an ONNX model.

Applies ORT_ENABLE_ALL optimization level which fuses quantized ops (DynamicQuantizeLinear, MatMulInteger, etc.) into custom ops that bypass MIGraphX’s broken kernels. The optimized model is saved alongside the original.

pub fn build_git_clone_command(&self) -> ShellCommand

Construct the git clone command for ONNX Runtime.

The original script clones: git clone --recursive https://github.com/microsoft/onnxruntime.git

pub fn build_git_checkout_command(&self) -> ShellCommand

Construct the git checkout command for the stable tag.

The original script checks out tag v1.20.1.

pub fn build_git_submodule_command(&self) -> ShellCommand

Construct the git submodule update command.

pub fn build_build_command(&self, rocm_env: &RocmEnv) -> ShellCommand

Construct the ONNX Runtime build.sh command with ROCm flags.

The original script runs ./build.sh with many flags:

• --config Release
• --build_wheel
• --parallel $(nproc)-1
• --use_rocm --rocm_home /opt/rocm
• --rocm_version 70200
• --use_migraphx --migraphx_home /opt/rocm
• --cmake_extra_defines CMAKE_HIP_ARCHITECTURES=...
• Various CMAKE isolation defines
• --allow_running_as_root

pub fn build_uninstall_command(&self) -> ShellCommand

Construct the pip uninstall command to remove all onnxruntime variants.

pub fn build_wheel_install_command(&self) -> ShellCommand

Construct the pip install command for the built wheel.

The original script installs from: build/Linux/Release/dist/*.whl

pub fn check_build_output( &self, stdout: &str, stderr: &str, exit_code: i32, ) -> Result<(), String>

Check build output for errors.

Detects common build failure patterns and propagates as an error.