github-starred/TinyTorch
2
0
Fork 0
mirror of https://github.com/MLSysBook/TinyTorch.git synced 2026-04-27 20:07:31 -05:00
Code Issues 7 Packages Projects Releases Wiki Activity

Extend integration test mapping to cover all 20 modules

Browse Source 
Added explicit comments explaining which tests apply to each tier:
- Foundation (01-07): Core integration tests
- Architecture (08-13): CNN and NLP pipeline tests
- Performance (14-19): Module-specific tests only
- Capstone (20): Comprehensive validation
This commit is contained in:
Vijay Janapa Reddi
2025-12-02 23:07:04 -08:00
parent 47635d1550
commit 4aa444517b
1 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tito/commands/module/test.py
View File

@@ -229,7 +229,9 @@ class ModuleTestCommand(BaseCommand):
relevant_tests = []
# Map module numbers to relevant integration tests
# Each module inherits tests from earlier modules (progressive testing)
integration_test_map = {
# Foundation modules (01-07)
1: ["test_basic_integration.py"],
2: ["test_basic_integration.py"],
3: ["test_layers_integration.py"],
@@ -237,12 +239,27 @@ class ModuleTestCommand(BaseCommand):
5: ["test_gradient_flow.py"],
6: ["test_training_flow.py"],
7: ["test_training_flow.py"],
# Architecture modules (08-13)
8: ["test_dataloader_integration.py"],
9: ["test_cnn_integration.py"],
10: [], # Tokenization is self-contained
11: [], # Embeddings tested in NLP pipeline
10: [], # Tokenization: self-contained, no integration deps
11: [], # Embeddings: tested in NLP pipeline (module 12)
12: ["test_nlp_pipeline_flow.py"],
13: ["test_nlp_pipeline_flow.py"],
# Performance modules (14-19) - build on all previous
# These use the same integration tests to ensure optimizations
# don't break existing functionality
14: [], # Profiling: observational, no integration changes
15: [], # Quantization: tested in module-specific tests
16: [], # Compression: tested in module-specific tests
17: [], # Memoization: tested in module-specific tests
18: [], # Acceleration: tested in module-specific tests
19: [], # Benchmarking: tested in module-specific tests
# Capstone (20) - runs comprehensive validation
20: ["test_training_flow.py", "test_nlp_pipeline_flow.py", "test_cnn_integration.py"],
}
# Collect all relevant tests up to and including this module