Add MIGraphX execution provider support (#740)

Introduces support for the ONNX Runtime MIGraphX execution provider, including build detection, UI options, localization, and session creation logic. Updates documentation to reflect MIGraphX as the recommended AMD GPU backend, noting ROCM deprecation in ONNX Runtime 1.23.0.

Co-authored-by: Kaito Udagawa <[email protected]>

Author: Copilot

CMakeLists.txt

@@ -177,6 +177,17 @@ if(HAVE_ONNXRUNTIME_ROCM_EP)
   target_compile_definitions(${CMAKE_PROJECT_NAME} PRIVATE HAVE_ONNXRUNTIME_ROCM_EP)
 endif()
 
+check_library_exists(
+  onnxruntime::onnxruntime
+  OrtSessionOptionsAppendExecutionProvider_MIGraphX
+  onnxruntime_c_api.h
+  HAVE_ONNXRUNTIME_MIGRAPHX_EP
+)
+if(HAVE_ONNXRUNTIME_MIGRAPHX_EP)
+  message(STATUS "ONNX Runtime MIGraphX Execution Provider found")
+  target_compile_definitions(${CMAKE_PROJECT_NAME} PRIVATE HAVE_ONNXRUNTIME_MIGRAPHX_EP)
+endif()
+
 check_library_exists(
   onnxruntime::onnxruntime
   OrtSessionOptionsAppendExecutionProvider_Tensorrt

README.md

@@ -63,7 +63,7 @@ GPU support:
 
 - On Windows, we plan to support WinML acceleration.
 - On Mac we support CoreML for acceleration, which is efficient on Apple Silicon. **Note:** This plugin does not support cross-architecture translation (Rosetta2). Intel binaries on Apple Silicon or Apple Silicon binaries on Intel will crash.
-- On Linux CUDA and ROCM are supported if this plugin is built from source. Ensure your ONNX Runtime installation has CUDA or ROCM support.
+- On Linux CUDA, ROCM (deprecated in ONNX Runtime 1.23.0), and MIGraphX are supported if this plugin is built from source. Ensure your ONNX Runtime installation has CUDA, ROCM, or MIGraphX support. For AMD GPUs, MIGraphX is recommended as ROCM was removed from ONNX Runtime starting with version 1.23.0.
 - The goal of this plugin is to be available for everyone on every system, even if they don't own a GPU.
 
 Number of CPU threads is controllable through the UI settings. A 2-thread setting works best.

data/locale/en-US.ini

@@ -9,6 +9,7 @@ InferenceDevice="Inference device"
 CPU="CPU"
 GPUCUDA="GPU - CUDA"
 GPUROCM="GPU - ROCM"
+GPUMIGRAPHX="GPU - MIGraphX"
 TENSORRT="TensorRT"
 CoreML="CoreML"
 SegmentationModel="Segmentation model"

src/background-filter.cpp

@@ -170,6 +170,9 @@ obs_properties_t *background_filter_properties(void *data)
 #ifdef HAVE_ONNXRUNTIME_ROCM_EP
 	obs_property_list_add_string(p_use_gpu, obs_module_text("GPUROCM"), USEGPU_ROCM);
 #endif
+#ifdef HAVE_ONNXRUNTIME_MIGRAPHX_EP
+	obs_property_list_add_string(p_use_gpu, obs_module_text("GPUMIGRAPHX"), USEGPU_MIGRAPHX);
+#endif
 #ifdef HAVE_ONNXRUNTIME_TENSORRT_EP
 	obs_property_list_add_string(p_use_gpu, obs_module_text("TENSORRT"), USEGPU_TENSORRT);
 #endif

src/consts.h

@@ -17,6 +17,7 @@ const char *const MODEL_RMBG = "models/bria_rmbg_1_4_qint8.onnx";
 const char *const USEGPU_CPU = "cpu";
 const char *const USEGPU_CUDA = "cuda";
 const char *const USEGPU_ROCM = "rocm";
+const char *const USEGPU_MIGRAPHX = "migraphx";
 const char *const USEGPU_TENSORRT = "tensorrt";
 const char *const USEGPU_COREML = "coreml";
 

src/enhance-filter.cpp

@@ -63,6 +63,9 @@ obs_properties_t *enhance_filter_properties(void *data)
 #ifdef HAVE_ONNXRUNTIME_ROCM_EP
 	obs_property_list_add_string(p_use_gpu, obs_module_text("GPUROCM"), USEGPU_ROCM);
 #endif
+#ifdef HAVE_ONNXRUNTIME_MIGRAPHX_EP
+	obs_property_list_add_string(p_use_gpu, obs_module_text("GPUMIGRAPHX"), USEGPU_MIGRAPHX);
+#endif
 #ifdef HAVE_ONNXRUNTIME_TENSORRT_EP
 	obs_property_list_add_string(p_use_gpu, obs_module_text("TENSORRT"), USEGPU_TENSORRT);
 #endif

src/ort-utils/ort-session-utils.cpp

@@ -65,6 +65,11 @@ int createOrtSession(filter_data *tf)
 			Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_ROCM(sessionOptions, 0));
 		}
 #endif
+#ifdef HAVE_ONNXRUNTIME_MIGRAPHX_EP
+		if (tf->useGPU == USEGPU_MIGRAPHX) {
+			Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_MIGraphX(sessionOptions, 0));
+		}
+#endif
 #ifdef HAVE_ONNXRUNTIME_TENSORRT_EP
 		if (tf->useGPU == USEGPU_TENSORRT) {
 			Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_Tensorrt(sessionOptions, 0));