Skip to content
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.

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

Merge from cTuning #874

Merged
merged 14 commits into from
Jul 24, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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
6 changes: 6 additions & 0 deletions cm-mlops/script/app-mlperf-inference/_cm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -290,6 +290,9 @@ variations:
- all
CM_MLPERF_ACCURACY_RESULTS_DIR:
- 'on'
skip_if_env:
CM_MLPERF_IMPLEMENTATION:
- nvidia-original
names:
- mlperf-accuracy-script
- openimages-accuracy-script
Expand Down Expand Up @@ -327,6 +330,9 @@ variations:
- all
CM_MLPERF_ACCURACY_RESULTS_DIR:
- 'on'
skip_if_env:
CM_MLPERF_IMPLEMENTATION:
- nvidia-original
names:
- mlperf-accuracy-script
- 3d-unet-accuracy-script
Expand Down
1 change: 1 addition & 0 deletions cm-mlops/script/app-mlperf-inference/customize.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ def postprocess(i):

env = i['env']
inp = i['input']
env['CMD'] = ''
state = i['state']

if env.get('CM_MLPERF_USER_CONF', '') == '':
Expand Down
8 changes: 8 additions & 0 deletions cm-mlops/script/app-mlperf-inference/run.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
#!/bin/bash

cmd="${CMD}"
if [[ -n ${cmd} ]]; then
echo "$cmd"
eval "$cmd"
test $? -eq 0 || exit $?
fi
Original file line number Diff line number Diff line change
Expand Up @@ -260,7 +260,7 @@ def preprocess(i):

print("Output Dir: '" + OUTPUT_DIR + "'")
print(user_conf)
if env.get('CM_MLPERF_POWER','') == "yes" and os.path.exists(env['CM_MLPERF_POWER_LOG_DIR']):
if env.get('CM_MLPERF_POWER','') == "yes" and os.path.exists(env.get('CM_MLPERF_POWER_LOG_DIR', '')):
shutil.rmtree(env['CM_MLPERF_POWER_LOG_DIR'])
else:
print("Run files exist, skipping run...\n")
Expand Down
1 change: 1 addition & 0 deletions cm-mlops/script/get-git-repo/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
"depth": "CM_GIT_DEPTH",
"folder": "CM_GIT_CHECKOUT_FOLDER",
"patch": "CM_GIT_PATCH",
"update": "CM_GIT_REPO_PULL",
"submodules": "CM_GIT_RECURSE_SUBMODULES"
},
"default_variation": "default",
Expand Down
6 changes: 3 additions & 3 deletions cm-mlops/script/install-cuda-prebuilt/customize.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,10 @@ def preprocess(i):

env['WGET_URL']="https://developer.download.nvidia.com/compute/cuda/"+env['CM_VERSION']+"/local_installers/"+env['CM_CUDA_LINUX_FILENAME']

extra_options = env.get('CUDA_ADDITIONAL_INSTALL_OPTIONS', '')
if env.get('CM_CUDA_INSTALL_DRIVER','') == "yes":
env['CUDA_ADDITIONAL_INSTALL_OPTIONS'] = " --driver"
else:
env['CUDA_ADDITIONAL_INSTALL_OPTIONS'] = ""
extra_options += " --driver"
env['CUDA_ADDITIONAL_INSTALL_OPTIONS'] = extra_options

env['CM_CUDA_INSTALLED_PATH'] = os.path.join(os.getcwd(), 'install')
env['CM_NVCC_BIN_WITH_PATH'] = os.path.join(os.getcwd(), 'install', 'bin', nvcc_bin)
Expand Down
4 changes: 3 additions & 1 deletion cm-mlops/script/install-cuda-prebuilt/run.sh
Original file line number Diff line number Diff line change
Expand Up @@ -4,5 +4,7 @@ CUR=${PWD}

INSTALL_DIR=${CUR}/install

${CM_SUDO} bash ${CM_CUDA_RUN_FILE_PATH} --toolkitpath=${INSTALL_DIR} --defaultroot=${INSTALL_DIR} --toolkit ${CUDA_ADDITIONAL_INSTALL_OPTIONS} --silent
cmd="${CM_SUDO} bash ${CM_CUDA_RUN_FILE_PATH} --toolkitpath=${INSTALL_DIR} --defaultroot=${INSTALL_DIR} --toolkit ${CUDA_ADDITIONAL_INSTALL_OPTIONS} --silent "
echo "${cmd}"
eval "${cmd}"

45 changes: 36 additions & 9 deletions cm-mlops/script/reproduce-mlperf-inference-nvidia/_cm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,9 @@ default_env:
CM_SKIP_PREPROCESS_DATASET: 'no'
CM_SKIP_MODEL_DOWNLOAD: 'no'
CM_MLPERF_SUT_NAME_IMPLEMENTATION_PREFIX: nvidia_original
CM_CALL_RUNNER: 'yes'
CM_MLPERF_SKIP_RUN: 'no'
env:
CM_CALL_MLPERF_RUNNER: 'no'

# Map script inputs to environment variables
input_mapping:
Expand All @@ -40,7 +42,7 @@ input_mapping:
mlperf_conf: CM_MLPERF_CONF
mode: CM_MLPERF_LOADGEN_MODE
output_dir: CM_MLPERF_OUTPUT_DIR
performance_sample_count: CM_MLPERF_LOADGEN_PERFORMANCE_SAMPLE_COUNT
performance_sample_count: CM_MLPERF_PERFORMANCE_SAMPLE_COUNT
scenario: CM_MLPERF_LOADGEN_SCENARIO
user_conf: CM_MLPERF_USER_CONF
devices: CM_MLPERF_NVIDIA_HARNESS_DEVICES
Expand Down Expand Up @@ -250,14 +252,14 @@ post_deps:

- names:
- runner
- mlperf-runner
skip_if_env:
CM_MLPERF_SKIP_RUN:
- 'yes'
- yes
CM_MLPERF_NVIDIA_HARNESS_RUN_MODE:
- run_harness
tags: benchmark-mlperf
enable_if_env:
CM_CALL_RUNNER:
CM_CALL_MLPERF_RUNNER:
- yes


Expand Down Expand Up @@ -293,6 +295,7 @@ variations:
group: model
env:
CM_MODEL: retinanet
CM_ML_MODEL_STARTING_WEIGHTS_FILENAME: "https://zenodo.org/record/6617981/files/resnext50_32x4d_fpn.pth"
deps:
- tags: get,generic-python-lib,_Pillow
- tags: get,generic-python-lib,_torch
Expand All @@ -310,36 +313,43 @@ variations:
- bert_
env:
CM_MODEL: bert-99
CM_NOT_ML_MODEL_STARTING_WEIGHTS_FILENAME: "https://zenodo.org/record/3750364/files/bert_large_v1_1_fake_quant.onnx"

bert-99.9:
group: model
base:
- bert_
env:
CM_MODEL: bert-99.9
CM_NOT_ML_MODEL_STARTING_WEIGHTS_FILENAME: "https://zenodo.org/record/3733910/files/model.onnx"

3d-unet_:
deps:
- tags: get,generic-python-lib,_transformers
- tags: get,generic-python-lib,_pandas
version_max: "1.5.3"

3d-unet-99:
group: model
base:
- 3d-unet_
env:
CM_MODEL: 3d-unet-99
CM_ML_MODEL_STARTING_WEIGHTS_FILENAME: "https://zenodo.org/record/5597155/files/3dunet_kits19_128x128x128.onnx"

3d-unet-99.9:
group: model
base:
- 3d-unet_
env:
CM_MODEL: 3d-unet-99.9
CM_ML_MODEL_STARTING_WEIGHTS_FILENAME: "https://zenodo.org/record/5597155/files/3dunet_kits19_128x128x128.onnx"

rnnt:
group: model
env:
CM_MODEL: rnnt
CM_ML_MODEL_STARTING_WEIGHTS_FILENAME: "https://zenodo.org/record/3662521/files/DistributedDataParallel_1576581068.9962234-epoch-100.pt"
deps:
- tags: get,generic-python-lib,_toml
- tags: get,generic-python-lib,_torchvision
Expand Down Expand Up @@ -479,9 +489,9 @@ variations:
- dlrm-99.9
- tags: reproduce,mlperf,inference,nvidia,harness,_calibrate
inherit_variation_tags: true
skip_if_env:
enable_if_env:
CM_MODEL:
- rnnt
- retinanet
force_cache: true
skip_inherit_variation_groups:
- run-mode
Expand Down Expand Up @@ -557,6 +567,7 @@ variations:
env:
CM_MLPERF_NVIDIA_HARNESS_RUN_MODE: run_harness
MLPERF_NVIDIA_RUN_COMMAND: run_harness
CM_CALL_MLPERF_RUNNER: 'yes'
new_env_keys:
- CM_MLPERF_*
- CM_DATASET_*
Expand Down Expand Up @@ -897,12 +908,20 @@ variations:
env:
CM_NVIDIA_CUSTOM_GPU: "yes"

l4,resnet50:
default_env:
CM_MLPERF_LOADGEN_OFFLINE_TARGET_QPS: 10500
CM_MLPERF_LOADGEN_SERVER_TARGET_QPS: 9000
CM_MLPERF_LOADGEN_SINGLESTREAM_TARGET_LATENCY: 0.35
CM_MLPERF_LOADGEN_MULTISTREAM_TARGET_LATENCY: 1

l4,resnet50,offline,run_harness:
add_deps_recursive:
build-engine:
tags: _batch_size.16
tags: _batch_size.32

env:
CM_MODEL_BATCH_SIZE: "16"
CM_MODEL_BATCH_SIZE: "32"
CM_MLPERF_NVIDIA_HARNESS_GPU_COPY_STREAMS: "2"
CM_MLPERF_NVIDIA_HARNESS_GPU_INFERENCE_STREAMS: "1"
CM_MLPERF_NVIDIA_HARNESS_USE_GRAPHS: 'True'
Expand Down Expand Up @@ -952,10 +971,18 @@ variations:
env:
CM_NVIDIA_CUSTOM_GPU: "yes"

t4,resnet50:
default_env:
CM_MLPERF_LOADGEN_OFFLINE_TARGET_QPS: 4900
CM_MLPERF_LOADGEN_SERVER_TARGET_QPS: 4000
CM_MLPERF_LOADGEN_SINGLESTREAM_TARGET_LATENCY: 0.6
CM_MLPERF_LOADGEN_MULTISTREAM_TARGET_LATENCY: 2

t4,resnet50,offline,run_harness:
add_deps_recursive:
build-engine:
tags: _batch_size.256

env:
CM_MODEL_BATCH_SIZE: "256"
CM_MLPERF_NVIDIA_HARNESS_GPU_COPY_STREAMS: "4"
Expand Down
19 changes: 13 additions & 6 deletions cm-mlops/script/reproduce-mlperf-inference-nvidia/customize.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,6 @@ def preprocess(i):
if not os.path.exists(model_path):
cmds.append(f"make download_model BENCHMARKS='{model_name}'")
else:
env['CM_CALL_RUNNER'] = "no"
return {'return':0}

elif make_command == "preprocess_data":
Expand Down Expand Up @@ -200,20 +199,28 @@ def preprocess(i):
multistream_target_latency_ns = int(float(multistream_target_latency) * 1000000)
run_config += f" --multi_stream_expected_latency_ns={multistream_target_latency_ns}"

high_accuracy = "99.9" in env['CM_MODEL']

use_triton = env.get('CM_MLPERF_NVIDIA_HARNESS_USE_TRITON')
if use_triton:
run_config += f" --use_triton --config_ver=triton"
if high_accuracy:
run_config += f" --use_triton --config_ver=high_accuracy_triton"
else:
run_config += f" --use_triton --config_ver=triton"
else:
if high_accuracy:
run_config += f" --config_ver=high_accuracy"

user_conf_path = env.get('CM_MLPERF_USER_CONF')
if user_conf_path:
if user_conf_path and env['CM_MLPERF_NVIDIA_HARNESS_RUN_MODE'] == "run_harness":
run_config += f" --user_conf_path={user_conf_path}"

mlperf_conf_path = env.get('CM_MLPERF_INFERENCE_CONF_PATH')
if mlperf_conf_path:
if mlperf_conf_path and env['CM_MLPERF_NVIDIA_HARNESS_RUN_MODE'] == "run_harness":
run_config += f" --mlperf_conf_path={mlperf_conf_path}"

power_setting = env.get('CM_MLPERF_NVIDIA_HARNESS_POWER_SETTING')
if power_setting:
if power_setting and env['CM_MLPERF_NVIDIA_HARNESS_RUN_MODE'] == "run_harness":
run_config += f" --power_setting={power_setting}"

gpu_copy_streams = env.get('CM_MLPERF_NVIDIA_HARNESS_GPU_COPY_STREAMS')
Expand All @@ -240,7 +247,7 @@ def preprocess(i):
if input_format:
run_config += f" --input_format={input_format}"

performance_sample_count = env.get('CM_MLPERF_LOADGEN_PERFORMANCE_SAMPLE_COUNT')
performance_sample_count = env.get('CM_MLPERF_PERFORMANCE_SAMPLE_COUNT')
if performance_sample_count:
run_config += f" --performance_sample_count={performance_sample_count}"

Expand Down
8 changes: 8 additions & 0 deletions cm-mlops/script/reproduce-mlperf-inference-nvidia/run.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
#!/bin/bash
if [[ ${CM_CALL_MLPERF_RUNNER} == "no" ]]; then
cd ${CM_RUN_DIR}
cmd=${CM_RUN_CMD}
echo "${cmd}"
eval "${cmd}"
test $? -eq 0 || exit $?
fi
Loading