Skip to content

Adding cycle count and benchmarks . #285

Adding cycle count and benchmarks .

Adding cycle count and benchmarks . #285

Workflow file for this run

name: pytest notebooks
on:
pull_request:
push:
branches: [main]
jobs:
pytest-container:
runs-on: ubuntu-latest
container:
image: graphcore/pytorch:3.2.0-ubuntu-20.04
steps:
- uses: actions/checkout@v3
- name : Install package dependencies
run: |
apt update -y
apt install git -y
- name: Install requirements
run: |
pip install -U pip
pip install -e ".[test,ipu]"
- name: Log installed environment
run: |
python3 -m pip freeze
- name: Test nanoDFT demo notebook
env:
JAX_IPU_USE_MODEL: 1
JAX_IPU_MODEL_NUM_TILES: 46
run: |
pytest --nbmake --nbmake-timeout=3000 notebooks/nanoDFT-demo.ipynb