Skip to content

Removed pycuda version fix #423

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
May 18, 2025
Merged

Removed pycuda version fix #423

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
cd91cdf
Removed pycuda version fix
arjunsuresh May 18, 2025
e6beae0
Upgrade numpy for nvmitten
arjunsuresh May 18, 2025
a2759a8
Fix nvidia-mitten install
arjunsuresh May 18, 2025
bf45610
detect sudo error
arjunsuresh May 18, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion script/app-mlperf-inference-nvidia/meta.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -274,7 +274,6 @@ deps:
- tags: get,generic-python-lib,_package.pycuda
names:
- pycuda
version: "2022.2.2"

- tags: get,generic-python-lib,_package.nvmitten
update_tags_from_env_with_prefix:
Expand Down
2 changes: 1 addition & 1 deletion script/detect-sudo/customize.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -174,7 +174,7 @@ def prompt_sudo(logger):
if not prompt_retry(): # If the user chooses not to retry or times out
return -1
except subprocess.CalledProcessError as e:
logger.error(f"Command failed: {e.output.decode('utf-8')}")
logger.error(f"Command failed: {e.output}")
reset_terminal() # Reset terminal in case of failure
return -1
except Exception as e:
Expand Down
6 changes: 0 additions & 6 deletions script/get-nvidia-mitten/customize.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,6 @@ def preprocess(i):

os_info = i['os_info']
env = i['env']
script_path = i['artifact'].path

if env.get('MLC_MLPERF_INFERENCE_VERSION', '') == "5.0":
extra_run_cmd = 'patch -p1 < {}'.format(os.path.join(
script_path, 'patch', 'numpy-mitten-v5.0.patch'))
env['EXTRA_RUN_CMD'] = extra_run_cmd

return {'return': 0}

Expand Down
14 changes: 13 additions & 1 deletion script/get-nvidia-mitten/meta.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ automation_alias: script
automation_uid: 5b4e0237da074764
cache: true
category: MLPerf benchmark support
default_version: master
default_version: main
deps:
- tags: detect,os
- names:
Expand All @@ -18,6 +18,7 @@ deps:
extra_cache_tags: nvidia,mitten,src
force_env_keys:
- MLC_GIT_CHECKOUT
- MLC_GIT_PATCH_FILEPATHS
names:
- nvidia-mitten-git-src
tags: get,git,_repo.https://github.com/NVIDIA/mitten
Expand All @@ -32,3 +33,14 @@ tags:
- mitten
- nvidia-mitten
uid: 1c045f2902374de9
variations:
patched:
group: numpy-patch
default: true
adr:
nvidia-mitten-git-src:
tags: _patch
env:
MLC_GIT_PATCH_FILEPATHS: <<<MLC_TMP_CURRENT_SCRIPT_PATH>>>/patch/numpy-mitten-v5.0.patch
default:
group: numpy-patch
5 changes: 2 additions & 3 deletions script/get-nvidia-mitten/run.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
#!/bin/bash
cd ${MLC_NVIDIA_MITTEN_SRC}
echo "EXTRA_RUN_CMD = ${EXTRA_RUN_CMD}"
eval "${EXTRA_RUN_CMD}"
test $? -eq 0 || exit $?
${MLC_PYTHON_BIN_WITH_PATH} -m pip install .
PIP_EXTRA=`python3 -c "import importlib.metadata; print(' --break-system-packages ' if int(importlib.metadata.version('pip').split('.')[0]) >= 23 else '')"`
${MLC_PYTHON_BIN_WITH_PATH} -m pip install . ${PIP_EXTRA}
test $? -eq 0 || exit $?
Loading