File tree Expand file tree Collapse file tree 2 files changed +7
-1
lines changed Expand file tree Collapse file tree 2 files changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -66,6 +66,8 @@ def parse_arg_remove_string(argv, arg_name_equal):
66
66
elif parse_arg_remove_boolean (sys .argv , "--use_rocm" ):
67
67
is_rocm = True
68
68
rocm_version = parse_arg_remove_string (sys .argv , "--rocm_version=" )
69
+ if parse_arg_remove_boolean (sys .argv , "--use_migraphx" ):
70
+ is_migraphx = True
69
71
elif parse_arg_remove_boolean (sys .argv , "--use_migraphx" ):
70
72
is_migraphx = True
71
73
elif parse_arg_remove_boolean (sys .argv , "--use_openvino" ):
@@ -89,8 +91,10 @@ def parse_arg_remove_string(argv, arg_name_equal):
89
91
elif parse_arg_remove_boolean (sys .argv , "--use_qnn" ):
90
92
package_name = "onnxruntime-qnn"
91
93
92
- if is_rocm or is_migraphx :
94
+ if is_rocm :
93
95
package_name = "onnxruntime-rocm" if not nightly_build else "ort-rocm-nightly"
96
+ elif is_migraphx :
97
+ package_name = "onnxruntime-migraphx" if not nightly_build else "ort-migraphx-nightly"
94
98
95
99
# PEP 513 defined manylinux1_x86_64 and manylinux1_i686
96
100
# PEP 571 defined manylinux2010_x86_64 and manylinux2010_i686
Original file line number Diff line number Diff line change @@ -2224,6 +2224,8 @@ def build_python_wheel(
2224
2224
args .append ("--use_rocm" )
2225
2225
if rocm_version :
2226
2226
args .append (f"--rocm_version={ rocm_version } " )
2227
+ if use_migraphx :
2228
+ args .append ("--use_migraphx" )
2227
2229
elif use_migraphx :
2228
2230
args .append ("--use_migraphx" )
2229
2231
elif use_openvino :
You can’t perform that action at this time.
0 commit comments