-
Notifications
You must be signed in to change notification settings - Fork 164
57 lines (53 loc) · 2.03 KB
/
torchbench.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
name: TorchBench
# 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:
TorchBenchCudaTiny:
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-cu118
device: gpu-bench
dockerfile: docker/cronjobs/Dockerfile.torch.bench
extra_run_args: --cap-add=SYS_ADMIN --gpus all
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_gpu.sh cuda tiny
push_command: ""
remote_dev_docker: ""
secrets: inherit
TorchBenchCudaPartial:
if: github.event.schedule == '30 15 * * *'
uses: ./.github/workflows/reusable.yml
with:
name: torch-offcial-benchmark
base_image: bladedisc/bladedisc:latest-runtime-torch-pre-cu118
device: gpu-bench
dockerfile: docker/cronjobs/Dockerfile.torch.bench
extra_run_args: --cap-add=SYS_ADMIN --gpus all
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_gpu.sh cuda partial
push_command: ""
remote_dev_docker: ""
secrets: inherit
TorchBenchCudaFull:
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-cu118
device: gpu-bench
timeout_minutes: 4320 # 72h
dockerfile: docker/cronjobs/Dockerfile.torch.bench
extra_run_args: --cap-add=SYS_ADMIN --gpus all
extra_envs: -e RELATED_DIFF_PERCENT=5
exec_command: bash ./pytorch_blade/benchmark/TorchBench/torch_bench_gpu.sh cuda full
push_command: ""
remote_dev_docker: ""
secrets: inherit