Skip to content

Latest commit

 

History

History
20 lines (14 loc) · 640 Bytes

README.md

File metadata and controls

20 lines (14 loc) · 640 Bytes

learn_pallas

small scripts I use to teach myself about pallas

Pipelining

For a good introduction into pipelining see this link For the python programs related to pipelining see the pipelining folder. The order the python programs should be read is

  1. naive_add_kernel.py
  2. add_with_blocks.py
  3. add_with_blocks_performance_boost.py
  4. reduce_sum_one_dim_grid.py
  5. reduce_sum_three_dim_grid.py

This guide is not completed.

TRC

Thanks to Googles TRC team for providing the ressources to use modern TPUs.