mirror of
https://github.com/deepseek-ai/DeepGEMM.git
synced 2026-11-05 00:47:26 +00:00
[PR #26] [CLOSED] Pytest #210
Labels
No labels
pull-request
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
deepseek-ai/DeepGEMM#210
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
📋 Pull Request Information
Original PR: https://github.com/deepseek-ai/DeepGEMM/pull/26
Author: @A-transformer
Created: 2/27/2025
Status: ❌ Closed
Base:
main← Head:pytest📝 Commits (4)
a813073pytest Integration60cce9apytest Integration22c163bpytest Integrationb4d5f53pytest Integration📊 Changes
3 files changed (+113 additions, -74 deletions)
View changed files
📝
README.md(+3 -2)📝
tests/test_core.py(+63 -59)📝
tests/test_jit.py(+47 -13)📄 Description
Decorate the test function with @pytest.mark.skipif(...) so the test is skipped if CUDA is unavailable. Move all testing logic into a function named test_jit() so it’s automatically discoverable by pytest.
CUTLASS Configuration
Ensure that the CUTLASS_ROOT environment variable is set. For example:
export CUTLASS_ROOT=/root/DeepGEMM/third-party/cutlass
export CPATH="$CUTLASS_ROOT/include:$CPATH"
export CPLUS_INCLUDE_PATH="$CUTLASS_ROOT/include:$CPLUS_INCLUDE_PATH"
Running the Tests:
pip install pytest
pytest tests/test_jit.py -v -s
pytest tests/test_core.py -v -s
nvidia-smi

🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.