update gh actions (#3033)

* update checkout actions

* update upload artifact

* update setup python

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
This commit is contained in:
jxdv 2024-01-10 01:52:22 +00:00 committed by GitHub
parent 3f80c1a098
commit ef3aa6d7fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 26 additions and 26 deletions

View File

@ -18,7 +18,7 @@ jobs:
PYTHONPATH: .
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Symlink models and datasets
run: |
mkdir -p weights
@ -48,7 +48,7 @@ jobs:
run: STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar.txt
- name: Run 10 CIFAR training steps w winograd
run: WINO=1 STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar_wino.txt
- uses: actions/upload-artifact@v3
- uses: actions/upload-artifact@v4
with:
name: Speed (Mac)
path: |
@ -74,7 +74,7 @@ jobs:
PYTHONPATH: .
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Run model inference benchmark
run: CUDA=1 python3 test/external/external_model_benchmark.py
- name: Test speed vs torch
@ -87,7 +87,7 @@ jobs:
run: CUDA=1 JIT=1 HALF=1 python3 examples/gpt2.py --count 10 --temperature 0 --timing
- name: Run GPT2 w HALF/BEAM
run: CUDA=1 JIT=1 HALF=1 BEAM=4 CACHELEVEL=0 python3 examples/gpt2.py --count 10 --temperature 0 --timing | tee gpt2_half_beam.txt
- uses: actions/upload-artifact@v3
- uses: actions/upload-artifact@v4
with:
name: Speed (NVIDIA)
path: |
@ -108,7 +108,7 @@ jobs:
PYTHONPATH: .
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Show off tinybox
run: /opt/rocm/bin/rocm-bandwidth-test
- name: Symlink models and datasets
@ -142,7 +142,7 @@ jobs:
# run: WINO=1 STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar_wino.txt
# - name: Run 10 CIFAR training steps w WINO/HALF/HIP
# run: HALF=1 WINO=1 STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar_wino_half_hip.txt
- uses: actions/upload-artifact@v3
- uses: actions/upload-artifact@v4
with:
name: Speed (AMD)
path: |

View File

@ -12,7 +12,7 @@ jobs:
deploy:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2
- uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v2
with:

View File

@ -57,7 +57,7 @@ jobs:
with:
path: base
- name: Set up Python 3.10
uses: actions/setup-python@v4
uses: actions/setup-python@v5
with:
python-version: '3.10'
- name: Count Line Diff

View File

@ -19,9 +19,9 @@ jobs:
# TODO: run the pre-commit hook to replace a lot of this
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Set up Python 3.8
uses: actions/setup-python@v4
uses: actions/setup-python@v5
with:
python-version: 3.8
- name: Cache python packages
@ -73,9 +73,9 @@ jobs:
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Set up Python 3.8
uses: actions/setup-python@v4
uses: actions/setup-python@v5
with:
python-version: 3.8
- name: Cache python packages
@ -106,9 +106,9 @@ jobs:
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Set up Python 3.11
uses: actions/setup-python@v4
uses: actions/setup-python@v5
with:
python-version: 3.11
- name: Cache python packages
@ -139,7 +139,7 @@ jobs:
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Install OpenCL
run: |
echo 'Acquire::http::Pipeline-Depth "5";' | sudo tee -a /etc/apt/apt.conf.d/99parallel
@ -150,7 +150,7 @@ jobs:
intel-oneapi-runtime-dpcpp-sycl-opencl-cpu=2023.2.1-16 intel-oneapi-runtime-tbb-common=2021.10.0-49541 \
intel-oneapi-runtime-tbb=2021.10.0-49541 intel-oneapi-runtime-opencl=2023.2.1-16
- name: Set up Python 3.11
uses: actions/setup-python@v4
uses: actions/setup-python@v5
with:
python-version: 3.11
- name: Cache python packages
@ -220,9 +220,9 @@ jobs:
# timeout-minutes: 20
# steps:
# - name: Checkout Code
# uses: actions/checkout@v3
# uses: actions/checkout@v4
# - name: Set up Python 3.11
# uses: actions/setup-python@v4
# uses: actions/setup-python@v5
# with:
# python-version: 3.11
# - name: Cache python packages
@ -262,9 +262,9 @@ jobs:
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Set up Python 3.11
uses: actions/setup-python@v4
uses: actions/setup-python@v5
with:
python-version: 3.11
- name: Cache python packages
@ -299,9 +299,9 @@ jobs:
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Set up Python 3.11
uses: actions/setup-python@v4
uses: actions/setup-python@v5
with:
python-version: 3.11
- name: Cache python packages
@ -379,9 +379,9 @@ jobs:
steps:
- name: Checkout Code
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Set up Python 3.11
uses: actions/setup-python@v4
uses: actions/setup-python@v5
with:
python-version: 3.11
- name: Cache python packages
@ -459,9 +459,9 @@ jobs:
# timeout-minutes: 20
# steps:
# - name: Checkout Code
# uses: actions/checkout@v3
# uses: actions/checkout@v4
# - name: Set up Python 3.11
# uses: actions/setup-python@v4
# uses: actions/setup-python@v5
# with:
# python-version: 3.11
# - name: Cache python packages