vivyd.core.TaichiContext

class TaichiContext(**ti_params)

Bases: object

A context manager for temporarily enabling Taichi acceleration within a block of code.

Parameters:

**ti_params – Additional parameters to pass to ti.init() when enabling Taichi. See use_taichi() for more details on the available parameters.

Example

with TaichiContext():
    # Code in this block will use Taichi acceleration if available
    ...