From 5479ffebb0699b8f3206738dcfd3d066f7e2a3b5 Mon Sep 17 00:00:00 2001
From: A-transformer <cl5743590921@gmail.com>
Date: Thu, 27 Feb 2025 11:44:52 +0400
Subject: [PATCH] pytest Integration

---
 README.md | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 14435f3..c371925 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