run on update_benchmark too (#1460)

* run on update_benchmark too

* amd inference test

* name it better

* add 10 CIFAR training steps
This commit is contained in:
George Hotz
2023-08-06 08:58:37 -07:00
committed by GitHub
parent 3d41674b42
commit 2ab282bfec
2 changed files with 28 additions and 7 deletions

View File

@@ -4,18 +4,36 @@ on:
push:
branches:
- master
- update_benchmark
jobs:
testmacbenchmark:
name: Mac Benchmark
runs-on: self-hosted
runs-on: [self-hosted, macOS]
steps:
- name: Checkout Code
uses: actions/checkout@v3
- name: Run metal ops test
- name: Run model inference benchmark
run: python3 test/external/external_model_benchmark.py
- name: Run 10 CIFAR training steps
run: STEPS=10 python3 examples/hlb_cifar10.py
- uses: actions/upload-artifact@v3
with:
name: ONNX Inference Speed
name: ONNX Inference Speed (Mac)
path: onnx_inference_speed.csv
testamdbenchmark:
name: AMD Benchmark
runs-on: [self-hosted, Linux]
steps:
- name: Checkout Code
uses: actions/checkout@v3
- name: Run model inference benchmark
run: python3 test/external/external_model_benchmark.py
- name: Run 10 CIFAR training steps
run: STEPS=10 python3 examples/hlb_cifar10.py
- uses: actions/upload-artifact@v3
with:
name: ONNX Inference Speed (AMD)
path: onnx_inference_speed.csv