Skip to content

Commit

Permalink
Update OpenVINO build to use a common version across windows and linux (
Browse files Browse the repository at this point in the history
  • Loading branch information
deadeyegoodwin authored Jun 25, 2021
1 parent bdbb4bc commit 3acb0c3
Showing 1 changed file with 6 additions and 12 deletions.
18 changes: 6 additions & 12 deletions build.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,8 +70,7 @@
'21.05', # upstream container
'1.8.0', # ORT
'2021.2.200', # ORT OpenVINO
'2021.2.200', # Standalone OpenVINO (non-windows)
'2021.2') # Standalone OpenVINO (windows)
'2021.2') # Standalone OpenVINO
}

EXAMPLE_BACKENDS = ['identity', 'square', 'repeat']
Expand Down Expand Up @@ -417,20 +416,15 @@ def onnxruntime_cmake_args(images, library_paths):


def openvino_cmake_args():
if target_platform() == 'windows':
cargs = [
'-DTRITON_BUILD_OPENVINO_VERSION={}'.format(
TRITON_VERSION_MAP[FLAGS.version][5]),
]
cargs = [
'-DTRITON_BUILD_OPENVINO_VERSION={}'.format(
TRITON_VERSION_MAP[FLAGS.version][4]),
]

if target_platform() == 'windows':
if 'base' in images:
cargs.append('-DTRITON_BUILD_CONTAINER={}'.format(images['base']))
else:
cargs = [
'-DTRITON_BUILD_OPENVINO_VERSION={}'.format(
TRITON_VERSION_MAP[FLAGS.version][4]),
]

if 'base' in images:
cargs.append('-DTRITON_BUILD_CONTAINER={}'.format(images['base']))
else:
Expand Down

0 comments on commit 3acb0c3

Please sign in to comment.