Latest development documentation · Updated 2026-09-08. A documentation snapshot for package 1.0.0.dev20260909 is not available.

On this page

tensorplay.library.wrap_triton

tensorplay.library.wrap_triton(kernel: Any) → TritonKernelWrapper[source]

Mark a Triton kernel as launchable from within a triton_op.

Accepts a triton.runtime.jit.JITFunction (the @triton.jit result) or any grid-indexable launcher; idempotent on wrappers.

Search documentation