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 committed Apr 10, 2024
1 parent 2387a2d commit 5371b1c
Show file tree
Hide file tree
Showing 6 changed files with 52 additions and 233 deletions.
51 changes: 51 additions & 0 deletions .github/workflows/benchmarks.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
name: Benchmark

on:
push:
branches:
- main
pull_request:

concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
cancel-in-progress: true

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

steps:
- uses: asottile/workflows/.github/actions/fast-checkout@v1.6.0

- 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: pytest benchmarks/ -vvv --codspeed
70 changes: 0 additions & 70 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.

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

This file was deleted.

30 changes: 0 additions & 30 deletions .github/workflows/scheduled_test_pypi_package.yml

This file was deleted.

1 change: 1 addition & 0 deletions requirements/requirements-benchmarks.txt
Original file line number Diff line number Diff line change
@@ -1 +1,2 @@
pytest-benchmark
pytest-codspeed

0 comments on commit 5371b1c

Please sign in to comment.