mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-03-12 03:23:51 -05:00
afd1cd442d
- Fix 14_profiling: Replace Tensor with Linear model in test_module, fix profile_forward_pass calls - Fix 15_quantization: Increase error tolerance for INT8 quantization test, add export marker for QuantizedLinear - Fix 19_benchmarking: Return Tensor objects from RealisticModel.parameters(), handle memoryview in pred_array.flatten() - Fix 20_capstone: Make imports optional (MixedPrecisionTrainer, QuantizedLinear, compression functions) - Fix 20_competition: Create Flatten class since it doesn't exist in spatial module - Fix 16_compression: Add export markers for magnitude_prune and structured_prune All modules now pass their inline tests.