tiny-tpu
Minimal TPU implementation with 8x8 systolic array and PyTorch integration
Language: python
Author: GitHub Repos (@github-repos)
0 stars · 1 views
Viewing path: LICENSE
Files
- test_memory_controller.py (py)
- matmul_compiler.py (py)
- compiler (txt)
- __init__.py (py)
- simulator.py (py)
- tiny_transformer (txt)
- models (txt)
- mnist_mlp (txt)
- test (txt)
- helpers (txt)
- visualizer (txt)
- tiny_tpu (txt)
- simulator (txt)
- assembler (txt)
- __init__.py (py)
- model.py (py)
- inference.py (py)
- train.py (py)
- convert.py (py)
- __init__.py (py)
- model.py (py)
- convert.py (py)
- attention_head (txt)
- inference.py (py)
- __init__.py (py)
- inference.py (py)
- test_unified_buffer.py (py)
- test_tiling_controller.py (py)
- test_matmul.py (py)
- __init__.py (py)
- model.py (py)
- __init__.py (py)
- test_sequencer.py (py)
- test_accumulator_buffer.py (py)
- logger.py (py)
- test_pe.py (py)
- __init__.py (py)
- test_systolic.py (py)
- test_activation_fifo.py (py)
- test_softmax_unit.py (py)
- format.py (py)
- test_layernorm_unit.py (py)
- convert.py (py)
- __init__.py (py)
- Makefile (txt)
- images (txt)
- README.md (md)
- setup.py (py)
- test_activation_unit.py (py)
- pe_detail.png (image)
- LICENSE (txt)
- systolic_dataflow.png (image)
- architecture.png (image)
- CONTRIBUTING.md (md)
- pyproject.toml (toml)
- isa_encoding.png (image)
- requirements.txt (txt)
- tpu_architecture.png (image)
- demo_results.png (image)
- performance.png (image)
- memory_map.png (image)
- pytorch (txt)
- src (txt)
- lexer.py (py)
- extractor.py (py)
- compiler.py (py)
- attention_compiler.py (py)
- ATTRIBUTION.md (markdown)
- assembler.py (py)
- __init__.py (py)
- codegen.py (py)
- parser.py (py)
- quantizer.py (py)
- __init__.py (py)
- test_weight_fifo.py (py)
- test_decoder.py (py)
- __init__.py (py)
- __init__.py (py)
