DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-04-07 10:04:04 +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 python setup.py develop
# Test JIT compilation # 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) # Test all GEMM implements (normal, contiguous-grouped and masked-grouped)
python tests/test_core.py pytest tests/test_core.py -v -s
``` ```
### Installation ### Installation