github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

mlperf cron install tensorflow try 2 (#13067)

Browse Source
This commit is contained in:
chenyu
2025-11-02 17:11:01 -05:00
committed by GitHub
parent c58cf91850
commit 4c22f089fc
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.github/workflows/mlperf.yml vendored
View File

@@ -17,6 +17,10 @@ jobs:
steps:
- name: Checkout Code
uses: actions/checkout@v4
- name: Setup Environment
uses: ./.github/actions/setup-tinygrad
with:
pydeps: 'tensorflow'
- name: Cleanup running AM processes
run: python extra/amdpci/am_smi.py --pids --kill
- name: Symlink datasets
@@ -26,6 +30,5 @@ jobs:
- name: Run bert
run: |
rm "~/.cache/tinygrad/cache_mlperf.db" || true
pip install tensorflow
BENCHMARK_LOG=mlpert_train_bert LOGMLPERF=0 CACHEDB="~/.cache/tinygrad/cache_mlperf.db" examples/mlperf/training_submission_v6.0/tinycorp/benchmarks/bert/implementations/tinybox_red/run_and_time.sh
rm "~/.cache/tinygrad/cache_mlperf.db"