feat: Add Non-intrusive Tensor Dumping for Model Inference (#10566)

2025-10-31 12:04:48 +08:00
parent 34c286b811
commit ab95d35fcb
6 changed files with 272 additions and 15 deletions
@@ -14,6 +14,7 @@ class TestFile:
 # NOTE: please sort the test cases alphabetically by the test file name
 suites = {
    "per-commit-1-gpu": [
+        TestFile("debug_utils/test_tensor_dump_forward_hook.py", 15),
        TestFile("function_call/test_json_schema_constraint.py", 30),
        TestFile("hicache/test_hicache.py", 116),
        TestFile("hicache/test_hicache_eagle.py", 150),