Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 44 additions & 0 deletions pytest.ini
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
# Pytest configuration for DeepGEMM
# https://docs.pytest.org/en/stable/reference/customize.html

[pytest]
# Test discovery
testpaths = tests
python_files = test_*.py
python_classes = Test*
python_functions = test_*

# Output options
addopts =
-v
--tb=short
--strict-markers
-ra

# Markers for categorizing tests
markers =
slow: marks tests as slow (deselect with '-m "not slow"')
gpu: marks tests that require GPU
sm90: marks tests that require SM90 (Hopper) GPU
sm100: marks tests that require SM100 (Blackwell) GPU
fp8: marks FP8 GEMM tests
bf16: marks BF16 GEMM tests
attention: marks attention kernel tests
einsum: marks einsum tests

# Timeout for tests (requires pytest-timeout)
# timeout = 300

# Minimum pytest version
minversion = 7.0

# Ignore certain warnings
filterwarnings =
ignore::DeprecationWarning
ignore::PendingDeprecationWarning

# Console output
console_output_style = progress

# Cache directory
cache_dir = .pytest_cache