Simon Fan 54c5f474a7 Forward rank and world size info to Torchbench models when using dynamo runner (#108438) ... Adding support to pass rank and world_size to torchbench model, via its extra_args parameter: https://github.com/pytorch/benchmark/blob/main/torchbenchmark/util/model.py#L83C80-L83C90 This is used for models which distribute over multiple GPUs e.g. simple_gpt https://github.com/pytorch/benchmark/pull/1867 Also add an option to skip multiprocess only gpu models Testing via `python benchmarks/dynamo/torchbench.py -d cuda --output=benchmark_logs/performance.csv --inference --performance --timing --print-memory --multiprocess --only simple_gpt` Pull Request resolved: https://github.com/pytorch/pytorch/pull/108438 Approved by: https://github.com/Chillee		2023-09-14 21:01:20 +00:00
..
actions	Fix the use of head_branch in filter-test-configs action (#107753)	2023-08-23 03:14:36 +00:00
ci_commit_pins	Forward rank and world size info to Torchbench models when using dynamo runner (#108438)	2023-09-14 21:01:20 +00:00
ISSUE_TEMPLATE
requirements	Bump scipy from 1.9.0 to 1.10.1 in /.github/requirements (#104763)	2023-08-24 10:10:40 +09:00
scripts	Allow marking multiple unstable configs of the same job name (#109185)	2023-09-13 17:06:37 +00:00
templates	Try to use linux.arm64.2xlarge runners (#107672)	2023-09-06 22:06:57 +00:00
workflows	Use the correct channel token when uploading nightly triton conda (#109073)	2023-09-13 17:12:33 +00:00
actionlint.yaml
auto_request_review.yml	Remove jbschlosser from symbolic-shapes auto request list (#107482)	2023-08-18 20:51:19 +00:00
ISSUE_TEMPLATE.md
labeler.yml	Add some more files to ciflow/inductor (#105112)	2023-07-13 14:44:42 +00:00
merge_rules.yaml	Add test/onnx_caffe2 to ONNX Exporter merge rule (#109295)	2023-09-14 19:06:12 +00:00
PULL_REQUEST_TEMPLATE.md
pytorch-circleci-labels.yml
pytorch-probot.yml
regenerate.sh
requirements-gha-cache.txt