forked from alibaba/BladeDISC
-
Notifications
You must be signed in to change notification settings - Fork 1
116 lines (108 loc) · 4.39 KB
/
torchbench_cpu.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
name: TorchBenchCPU
# Add cronjob later
# Use workflow dispatch to manual trigger the job for now
on:
schedule:
# * is a special character in YAML so you have to quote this string
- cron: '30 15 * * *' # 11:30 pm UTC+8:00
- cron: '30 15 * * 5' # 11:30 pm UTC+8:00 every Friday
workflow_dispatch:
jobs:
TorchBenchIntelPartial-Pre:
if: github.event.schedule != '30 15 * * 5' # daily or dispatch
uses: ./.github/workflows/reusable.yml
with:
name: torch-offcial-benchmark
base_image: bladedisc/bladedisc:latest-runtime-torch-pre-cpu
device: g7-bench
dockerfile: docker/cronjobs/Dockerfile.torch.bench
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_cpu.sh X86-intel partial pre
push_command: ""
remote_dev_docker: ""
secrets: inherit
TorchBenchIntelFull-Pre:
if: github.event.schedule == '30 15 * * 5'
uses: ./.github/workflows/reusable.yml
with:
name: torch-offcial-benchmark
base_image: bladedisc/bladedisc:latest-runtime-torch-pre-cpu
device: g7-bench
timeout_minutes: 10080 # 7 days
dockerfile: docker/cronjobs/Dockerfile.torch.bench
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_cpu.sh X86-intel full pre
push_command: ""
remote_dev_docker: ""
secrets: inherit
TorchBenchYitianPartial-200:
if: github.event.schedule != '30 15 * * 5' # daily or dispatch
uses: ./.github/workflows/reusable.yml
with:
name: torch-offcial-benchmark
base_image: bladedisc/bladedisc:latest-runtime-torch2.0.0-cpu-yitian
device: yitian-bench-sh2
proxy_config: source ~/.cache/proxy_config
dockerfile: docker/cronjobs/Dockerfile.torch.bench.aarch64
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_cpu.sh AArch64-yitian partial 200
push_command: ""
remote_dev_docker: ""
secrets: inherit
TorchBenchYitianAMPPartial-200:
if: github.event.schedule != '30 15 * * 5' # daily or dispatch
uses: ./.github/workflows/reusable.yml
with:
name: torch-offcial-benchmark
base_image: bladedisc/bladedisc:latest-runtime-torch2.0.0-cpu-yitian
device: yitian-bench-sh2
proxy_config: source ~/.cache/proxy_config
dockerfile: docker/cronjobs/Dockerfile.torch.bench.aarch64
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_cpu.sh AArch64-yitian-amp partial 200
push_command: ""
remote_dev_docker: ""
secrets: inherit
TorchBenchYitianFull-200:
if: github.event.schedule == '30 15 * * 5'
uses: ./.github/workflows/reusable.yml
with:
name: torch-offcial-benchmark
base_image: bladedisc/bladedisc:latest-runtime-torch2.0.0-cpu-yitian
device: yitian-bench
timeout_minutes: 14400 # 10 days
proxy_config: source ~/.cache/proxy_config
dockerfile: docker/cronjobs/Dockerfile.torch.bench.aarch64
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_cpu.sh AArch64-yitian full 200
push_command: ""
remote_dev_docker: ""
secrets: inherit
TorchBenchG6RFull-200:
if: github.event.schedule == '30 15 * * 5'
uses: ./.github/workflows/reusable.yml
with:
name: torch-offcial-benchmark
base_image: bladedisc/bladedisc:latest-runtime-torch2.0.0-cpu-aarch64
device: g6r-bench
timeout_minutes: 14400 # 10 days
dockerfile: docker/cronjobs/Dockerfile.torch.bench.aarch64
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_cpu.sh AArch64-g6r full 200
push_command: ""
remote_dev_docker: ""
secrets: inherit
TorchBenchAMDFull-Pre:
if: github.event.schedule == '30 15 * * 5'
uses: ./.github/workflows/reusable.yml
with:
name: torch-offcial-benchmark
base_image: bladedisc/bladedisc:latest-runtime-torch-pre-cpu
device: g7a-bench
timeout_minutes: 14400 # 10 days
dockerfile: docker/cronjobs/Dockerfile.torch.bench
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_cpu.sh X86-amd full pre
push_command: ""
remote_dev_docker: ""
secrets: inherit