Skip to main content

Module onnxruntime

Module onnxruntime 

Source
Expand description

ONNX Runtime installer — ports scripts/build_onnxruntime_multi.sh.

Constructs correct cmake commands with ROCm integration and Python bindings, builds from source using ONNX Runtime’s build.sh, and pip installs from the built wheel. Falls back to prebuilt wheel if available.

ONNX Runtime depends on ROCm.

§Validation Assertions

  • VAL-INSTALL-010: ONNX Runtime correct cmake command
  • VAL-INSTALL-011: ONNX Runtime pip install from build output
  • VAL-INSTALL-014: Source builds parse build output for errors
  • VAL-INSTALL-045: ONNX Runtime declares dependency on ROCm

Structs§

OnnxRuntimeConfig
Configuration for the ONNX Runtime installer.
OnnxRuntimeInstaller
The ONNX Runtime installer.
ShellCommand
A constructed shell command.

Enums§

HipArchs
GPU architecture specification for HIP builds.
OnnxInstallMethod
ONNX Runtime install method.

Constants§

DEFAULT_ONNXRUNTIME_VERSION
Current ROCm-compatible ONNX Runtime release: the source-build target and the version the bundled manifest proposes.
ONNX_INSTALL_METHOD_ENV
Env var selecting the ONNX install method (migraphx/source/prebuilt). Default: OnnxInstallMethod::MigraphxWheel.
ONNX_VERSION_ENV
Env var overriding the prebuilt onnxruntime-migraphx version (else the pinned PREBUILT_MIGRAPHX_VERSION is used).
PREBUILT_MIGRAPHX_VERSION
Pinned prebuilt onnxruntime-migraphx wheel version installed from PyPI by the default (OnnxInstallMethod::MigraphxWheel) path — the primary, production-proven ROCm build that ships the MIGraphXExecutionProvider.

Functions§

install_method_from_env
Resolve the ONNX install method from ONNX_INSTALL_METHOD_ENV.
prebuilt_version_from_env
Resolve an overridden prebuilt version from ONNX_VERSION_ENV, else None.