Skip to content

Commit 586225e

Browse files
committed
create package for migraphx ep
1 parent 479551b commit 586225e

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

setup.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,8 @@ def parse_arg_remove_string(argv, arg_name_equal):
6666
elif parse_arg_remove_boolean(sys.argv, "--use_rocm"):
6767
is_rocm = True
6868
rocm_version = parse_arg_remove_string(sys.argv, "--rocm_version=")
69+
if parse_arg_remove_boolean(sys.argv, "--use_migraphx"):
70+
is_migraphx = True
6971
elif parse_arg_remove_boolean(sys.argv, "--use_migraphx"):
7072
is_migraphx = True
7173
elif parse_arg_remove_boolean(sys.argv, "--use_openvino"):
@@ -89,8 +91,10 @@ def parse_arg_remove_string(argv, arg_name_equal):
8991
elif parse_arg_remove_boolean(sys.argv, "--use_qnn"):
9092
package_name = "onnxruntime-qnn"
9193

92-
if is_rocm or is_migraphx:
94+
if is_rocm:
9395
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"
9498

9599
# PEP 513 defined manylinux1_x86_64 and manylinux1_i686
96100
# PEP 571 defined manylinux2010_x86_64 and manylinux2010_i686

tools/ci_build/build.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2224,6 +2224,8 @@ def build_python_wheel(
22242224
args.append("--use_rocm")
22252225
if rocm_version:
22262226
args.append(f"--rocm_version={rocm_version}")
2227+
if use_migraphx:
2228+
args.append("--use_migraphx")
22272229
elif use_migraphx:
22282230
args.append("--use_migraphx")
22292231
elif use_openvino:

0 commit comments

Comments
 (0)