Skip to content

Commit

Permalink
feat: add CodSpeed to the project
Browse files Browse the repository at this point in the history
  • Loading branch information
adriencaccia authored and CodSpeed Install Bot committed Aug 28, 2024
1 parent 343357d commit d645a84
Show file tree
Hide file tree
Showing 7 changed files with 61 additions and 345 deletions.
54 changes: 54 additions & 0 deletions .github/workflows/benchmarks.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,54 @@
name: Benchmark

on:
push:
branches:
- main
pull_request:
workflow_dispatch:

jobs:
benchmark_cpu:
name: CPU Pytest benchmark
runs-on: ubuntu-latest

steps:
- uses: asottile/workflows/.github/actions/[email protected]

- name: Setup environment
uses: actions/setup-python@v4
with:
python-version: 3.11

- name: Install pytorch
shell: bash
run: pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/cpu

- name: Install Kornia dev
shell: bash
run: pip install .[dev,x]

- name: Check torch version
shell: bash
run: pip show torch | grep 2.2.0 || false

- name: Print dependencies and kornia version
shell: bash
run: |
python -c "import torch;print('Pytorch version: ', torch.__version__)"
python -c "import kornia;print('Kornia version: ', kornia.__version__)"
- name: Install benchmark requirements
run: pip install -r requirements/requirements-benchmarks.txt

- name: Run benchmarks
uses: CodSpeedHQ/action@v2
with:
token: ${{ secrets.CODSPEED_TOKEN }}
# run only 4 benchmarks because they take a long time to run for the time being
run: |
pytest -vvv --codspeed -n auto \
benchmarks/augmentation/2d_geometric_test.py::test_aug_2d_thin_plate_spline \
benchmarks/augmentation/2d_geometric_test.py::test_aug_2d_elastic_transform \
benchmarks/augmentation/2d_intensity_test.py::test_aug_2d_linear_corner_illumination \
benchmarks/augmentation/2d_intensity_test.py::test_aug_2d_solarize
128 changes: 0 additions & 128 deletions .github/workflows/pr_test_cpu.yml

This file was deleted.

68 changes: 0 additions & 68 deletions .github/workflows/pypi-release.yml

This file was deleted.

117 changes: 0 additions & 117 deletions .github/workflows/scheduled_test_cpu.yml

This file was deleted.

Loading

0 comments on commit d645a84

Please sign in to comment.