mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-04-30 23:31:32 -05:00
4ea5a4e024
- Add tinytorch.utils.profiler following PyTorch's utils pattern - Includes SimpleProfiler class for educational performance measurement - Provides timing, memory usage, and system metrics - Follows PyTorch's torch.utils.* organizational pattern - Module 11: Kernels uses profiler for performance demonstrations Features: - Wall time and CPU time measurement - Memory usage tracking (peak, delta, percentages) - Array information (shape, size, dtype) - CPU and system metrics - Clean educational interface for ML performance learning Import pattern: from tinytorch.utils.profiler import SimpleProfiler
9 lines
179 B
Python
9 lines
179 B
Python
"""
|
|
TinyTorch Utils Package
|
|
|
|
Shared utilities for TinyTorch modules.
|
|
"""
|
|
|
|
from .profiler import SimpleProfiler, profile_function
|
|
|
|
__all__ = ['SimpleProfiler', 'profile_function'] |