DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-04-05 04:59:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

pytest Integration

Browse Source
This commit is contained in:
A-transformer 2025-02-27 11:55:32 +04:00 committed by GitHub
parent 60cce9a6e3
commit 22c163be25
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@ -76,10 +76,11 @@ git clone --recursive git@github.com:deepseek-ai/DeepGEMM.git
python setup.py develop
# Test JIT compilation
python tests/test_jit.py
pip install pytest
pytest tests/test_jit.py -v -s
# Test all GEMM implements (normal, contiguous-grouped and masked-grouped)
python tests/test_core.py
pytest tests/test_core.py -v -s
```
### Installation