Skip to content

Latest commit

 

History

History
 
 

sharding

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
requirements.txt
requirements.txt
 
 
simple_sharding.py
simple_sharding.py
 
 
test_simple_sharding.py
test_simple_sharding.py
 
 

README.md

Graphcore

Sharding examples

File structure

  • simple_sharding.py Minimal example of IPU sharding
  • README.md This file.

How to use this demo

  1. Prepare the TensorFlow environment.

    Install the Poplar SDK following the Getting Started guide for your IPU system. Make sure to run the enable.sh script for Poplar and activate a Python virtualenv with the tensorflow-1 wheel from the Poplar SDK installed.

  2. Run the script.

     python simple_sharding.py