TensorPlay
Reference guides
Copy
View MarkdownDownload .md

tensorplay.compiler

TensorPlay ships an in-process graph compiler used by tensorplay.stax and the CUDA graphs workflow. This module is TensorPlay-specific and has no direct upstream torch counterpart (it is conceptually closest to torch.compile internals plus torch.cuda.graphs).

The reference below is a static listing: the underlying extension modules are under active development, so this page intentionally does not use autodoc.

Graph IR

  • Graph, Node, GraphModule — graph representation and module wrapper

  • PassBase, DecomposePass, ConstFold, DeadCodeElimination, NormalizeOperators — compiler passes

  • NodePathTracer — attribution helper

CUDA graphs

  • CudaGraphManager, CudaGraphError — capture/replay management

  • Guard, GuardChain — shape/guard caching for replay validation

Caching and errors

  • CodeCache — compiled artifact cache

  • AOTError, GraphCaptureError — error types

tensorplay.audio

tensorplay.audio is the audio I/O and processing toolkit of TensorPlay, mirroring the torchaudio package layout. The listing below is a static overview; backend availability depends on installed optional dependencies, so

tensorplay.cuda

tensorplay.cuda.StreamContext

Ask DeepWiki