Skip to content

Latest commit

 

History

History
32 lines (20 loc) · 534 Bytes

py_api.rst

File metadata and controls

32 lines (20 loc) · 534 Bytes

Python API Reference

Python API is used to control & access SPU, for example, to do data infeed/outfeed, to compile an XLA program to PPHlo, or to fire a PPHlo on an SPU runtime.

Runtime Setup

.. autoclass:: spu.Runtime
    :members:

Runtime IO

.. autoclass:: spu.Io
    :members:

Compiler

.. autofunction:: spu.compile

Simulation

.. autoclass:: spu.utils.simulation.Simulator
   :members:
   :undoc-members:


.. autofunction:: spu.utils.simulation.sim_jax