API Reference

These pages are generated from the pyptx/ package and module docstrings.

• pyptx: Top-level public API for :mod:pyptx.
• pyptx.cache: Disk cache for compiled PTX → cubin.
• pyptx.jax_support: JAX runtime integration for :func:pyptx.kernel.
• pyptx.kernel: Kernel tracing, specialization, and runtime dispatch.
• pyptx.ptx: PTX instruction namespace.
• pyptx.reg: Register allocation and register-level DSL sugar.
• pyptx.smem: Shared-memory allocation, addressing, and barrier objects.
• pyptx.specs: Tensor boundary specifications for :func:pyptx.kernel.
• pyptx.torch_support: PyTorch runtime integration for :func:pyptx.kernel.
• pyptx.types: PTX scalar type descriptors.
• pyptx.wgmma_layout: Canonical GMMA shared-memory layouts for wgmma.mma_async.