github-starred/TinyTorch
2
0
Fork 0
mirror of https://github.com/MLSysBook/TinyTorch.git synced 2026-04-29 06:57:30 -05:00
Code Issues 7 Packages Projects Releases Wiki Activity

Update test files with progressive integration and checkpoint improvements

Browse Source
This commit is contained in:
Vijay Janapa Reddi
2025-12-04 11:08:17 -08:00
parent 6590194aea
commit 4264699b5f
54 changed files with 624 additions and 618 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinytorch/optimization/acceleration.py generated
View File

@@ -21,6 +21,10 @@ __all__ = ['vectorized_matmul', 'fused_gelu', 'tiled_matmul']
#| default_exp optimization.acceleration
#| export
# %% ../../modules/18_acceleration/18_acceleration.ipynb 5
# Import from TinyTorch package (previous modules must be completed and exported)
from tinytorch.core.tensor import Tensor
# %% ../../modules/18_acceleration/18_acceleration.ipynb 7
def vectorized_matmul(a: Tensor, b: Tensor) -> Tensor:
"""