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")