tilelang.jit.abi¶
Shared helpers for preparing TVM-FFI callable ABIs.
Functions¶
|
Resolve output indices and expose them to TVM-FFI lowering. |
Module Contents¶
- tilelang.jit.abi.prepare_tvm_ffi_callee_allocated_outputs(func, out_idx)¶
Resolve output indices and expose them to TVM-FFI lowering.
- Parameters:
func (tvm.tirx.PrimFunc)
out_idx (list[int] | int | None)
- Return type:
tuple[tvm.tirx.PrimFunc, list[int] | None]