From 22c163be2572719d1415d71906b7b9f7d2b4481f Mon Sep 17 00:00:00 2001 From: A-transformer Date: Thu, 27 Feb 2025 11:55:32 +0400 Subject: [PATCH] pytest Integration --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index d3cbbdf..9cde064 100644 --- a/README.md +++ b/README.md @@ -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