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.