DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-06-26 23:15:49 +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
1 changed files with 3 additions and 2 deletions
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