Skip to content

Commit cd2c275

Browse files
authored
Merge branch 'dev' into dev
2 parents 22b9a87 + 888ae5c commit cd2c275

3 files changed

Lines changed: 7 additions & 10 deletions

File tree

automation/script/remote_run.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -101,11 +101,7 @@ def remote_run(self_module, i):
101101
# Even if we're running from Windows locally, the remote commands execute
102102
# on the remote server
103103
run_cmds.append(
104-
'if [ "$(uname)" = "Darwin" ]; then '
105-
f'curl -sSL https://raw.githubusercontent.com/mlcommons/mlcflow/refs/heads/dev/docs/install/mlcflow_macos.sh | bash -s -- --yes --venv-dir {remote_mlc_python_venv}; '
106-
'else '
107-
f'curl -sSL https://raw.githubusercontent.com/mlcommons/mlcflow/refs/heads/dev/docs/install/mlcflow_linux.sh | bash -s -- --yes --venv-dir {remote_mlc_python_venv}; '
108-
'fi')
104+
f'curl -sSL https://raw.githubusercontent.com/mlcommons/mlcflow/refs/heads/dev/docs/install/mlcflow_unix_installer.sh | bash -s -- --yes --venv-dir {remote_mlc_python_venv}')
109105
run_cmds.append(f". {remote_mlc_python_venv}/bin/activate")
110106
if i.get('remote_pull_mlc_repos', False):
111107
run_cmds.append("mlc pull repo")

script/get-mlperf-multi-node-system-info/customize.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,8 +98,9 @@ def preprocess(i):
9898

9999
# set remote run tags
100100
rr_tags = "get,mlperf,single-node,system-info"
101-
if env.get('MLC_ACCELERATOR_BACKEND', '') == 'cuda':
102-
rr_tags += ",_cuda"
101+
backend = env.get('MLC_ACCELERATOR_BACKEND', '')
102+
if backend in ('cuda', 'rocm', 'xpu'):
103+
rr_tags += f",_{backend}"
103104
ssh_ids = [
104105
s.strip() for s in env['MLC_MULTINODE_SYSTEM_SSH_IDS'].split(',') if s.strip()]
105106

script/get-mlperf-single-node-system-info/meta.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,21 +26,21 @@ deps:
2626
- tags: detect,cpu
2727
- tags: get,cuda-devices
2828
enable_if_env:
29-
MLC_GPU_PLATFORM:
29+
MLC_ACCELERATOR_BACKEND:
3030
- cuda
3131
skip_if_env:
3232
MLC_HOST_OS_TYPE:
3333
- windows
3434
- tags: get,rocm-devices
3535
enable_if_env:
36-
MLC_GPU_PLATFORM:
36+
MLC_ACCELERATOR_BACKEND:
3737
- rocm
3838
skip_if_env:
3939
MLC_HOST_OS_TYPE:
4040
- windows
4141
- tags: get,xpu-devices
4242
enable_if_env:
43-
MLC_GPU_PLATFORM:
43+
MLC_ACCELERATOR_BACKEND:
4444
- xpu
4545
skip_if_env:
4646
MLC_HOST_OS_TYPE:

0 commit comments

Comments
 (0)