Add support for tf32 and set precision to bf16-mixed if available #63
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: code checks | |
| on: | |
| push: | |
| branches: | |
| - main | |
| paths: | |
| - .pre-commit-config.yaml | |
| - .github/workflows/code_checks.yml | |
| - '**.py' | |
| - poetry.lock | |
| - pyproject.toml | |
| - '**.ipynb' | |
| pull_request: | |
| branches: | |
| - main | |
| paths: | |
| - .pre-commit-config.yaml | |
| - .github/workflows/code_checks.yml | |
| - '**.py' | |
| - poetry.lock | |
| - pyproject.toml | |
| - '**.ipynb' | |
| jobs: | |
| run-code-check: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - uses: actions/[email protected] | |
| - name: Install and configure Poetry | |
| uses: snok/install-poetry@v1 | |
| with: | |
| virtualenvs-create: true | |
| virtualenvs-in-project: true | |
| - uses: actions/[email protected] | |
| with: | |
| python-version: '3.9' | |
| cache: 'poetry' | |
| - name: Install dependencies and check code | |
| run: | | |
| poetry env use '3.9' | |
| source .venv/bin/activate | |
| poetry install --with test --all-extras | |
| pre-commit run --all-files | |
| - name: pip-audit (gh-action-pip-audit) | |
| uses: pypa/[email protected] | |
| with: | |
| virtual-environment: .venv/ | |
| ignore-vulns: GHSA-cqh9-jfqr-h9jj |