Skip to content

Unify correctness and benchmark testing with pytest#8

Merged
voltjia merged 8 commits intomasterfrom
unify-tests
Apr 13, 2026
Merged

Unify correctness and benchmark testing with pytest#8
voltjia merged 8 commits intomasterfrom
unify-tests

Conversation

@voltjia
Copy link
Copy Markdown
Collaborator

@voltjia voltjia commented Apr 13, 2026

Summary

  • Add bench module with assert_match for correctness checking and benchmark for performance benchmarking via triton.testing.
  • Split modules.py into a modules/ package with separate files for each module.
  • Move max_pool2d kernel into ops/ninetoothed/ for consistency with other operators.
  • Add pytest infrastructure with correctness tests (test_ops.py) and benchmark tests (test_benchmarks.py).
  • Remove the 13 standalone operator scripts, since both correctness and benchmarking are now covered by pytest.

Usage

pytest # Correctness tests only.
pytest -m benchmark # Benchmark tests only.
pytest -m benchmark -k TestMM # Single operator benchmark.

@voltjia voltjia merged commit e2309e3 into master Apr 13, 2026
2 checks passed
@voltjia voltjia deleted the unify-tests branch April 13, 2026 08:33
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant