Merge branch 'master' of github.com:tinygrad/tinygrad into cifar_fp16

This commit is contained in:
Yixiang Gao
2024-01-03 10:11:46 -08:00
16 changed files with 376 additions and 131 deletions

View File

@@ -96,8 +96,8 @@ jobs:
gpt2_half_beam.txt
testamdbenchmark:
name: AMD Benchmark
runs-on: [self-hosted, Linux, ROCM]
name: tinybox Benchmark
runs-on: [self-hosted, Linux, tinybox]
defaults:
run:
shell: bash -o pipefail {0}
@@ -107,9 +107,10 @@ jobs:
steps:
- name: Checkout Code
uses: actions/checkout@v3
- name: Show off tinybox
run: /opt/rocm/bin/rocm-bandwidth-test
- name: Symlink models and datasets
run: |
ln -s ~/tinygrad/weights/sd-v1-4.ckpt weights/sd-v1-4.ckpt
ln -s ~/tinygrad/weights/bpe_simple_vocab_16e6.txt.gz weights/bpe_simple_vocab_16e6.txt.gz
ln -s ~/tinygrad/weights/LLaMA weights/LLaMA
ln -s ~/tinygrad/extra/datasets/cifar-10-python.tar.gz extra/datasets/cifar-10-python.tar.gz

View File

@@ -1,7 +1,7 @@
name: Unit Tests
env:
# increment this when downloads substantially change to avoid the internet
DOWNLOAD_CACHE_VERSION: '3'
DOWNLOAD_CACHE_VERSION: '4'
on:
push: