From 2d4b193abab28bbda24da38bf307b20c09adcd7d Mon Sep 17 00:00:00 2001 From: Vijay Janapa Reddi Date: Sun, 20 Jul 2025 12:19:49 -0400 Subject: [PATCH] Replace manual test calls with automatic discovery --- modules/source/12_compression/compression_dev.py | 12 +----------- modules/source/13_kernels/kernels_dev.py | 12 +----------- 2 files changed, 2 insertions(+), 22 deletions(-) diff --git a/modules/source/12_compression/compression_dev.py b/modules/source/12_compression/compression_dev.py index 080f3994..0df8d083 100644 --- a/modules/source/12_compression/compression_dev.py +++ b/modules/source/12_compression/compression_dev.py @@ -1817,18 +1817,8 @@ Time to test your implementation! This section uses TinyTorch's standardized tes # ============================================================================= if __name__ == "__main__": - # Unit tests - test_compression_metrics() - test_magnitude_pruning() - test_quantization() - test_distillation() - test_structured_pruning() - test_comprehensive_comparison() - # Integration tests - test_compression_integration() - test_comprehensive_compression_integration() - from tito.tools.testing import run_module_tests_auto + # Automatically discover and run all tests in this module success = run_module_tests_auto("Compression") diff --git a/modules/source/13_kernels/kernels_dev.py b/modules/source/13_kernels/kernels_dev.py index 26ae965f..0c713c47 100644 --- a/modules/source/13_kernels/kernels_dev.py +++ b/modules/source/13_kernels/kernels_dev.py @@ -1404,18 +1404,8 @@ Time to test your implementation! This section uses TinyTorch's standardized tes # ============================================================================= if __name__ == "__main__": - # Unit tests - test_matmul_baseline() - test_vectorized_operations() - test_cache_friendly_matmul() - test_parallel_processing() - test_simple_kernel_timing() - test_compressed_kernels() - final_performance_test() - # Integration test - test_module_kernel_sequential_model() - from tito.tools.testing import run_module_tests_auto + # Automatically discover and run all tests in this module success = run_module_tests_auto("Kernels")