mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-22 13:28:06 -05:00
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:
24
.github/workflows/benchmark.yml
vendored
24
.github/workflows/benchmark.yml
vendored
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user