tilelang.engine.host_mapped

Parameter contracts for device-accessible mapped host memory.

Attributes

HOST_MAPPED_PARAMS_ATTR

Functions

get_host_mapped_params(func)	Read and validate parameter indices, including on cache restoration.
validate_host_mapped_params(func, target, ...[, out_idx])	Reject unsupported execution and allocation paths before compilation.

Module Contents

tilelang.engine.host_mapped.HOST_MAPPED_PARAMS_ATTR = 'tl.host_mapped_params'

tilelang.engine.host_mapped.get_host_mapped_params(func)

Read and validate parameter indices, including on cache restoration.

Parameters:

func (tvm.tirx.PrimFunc)

Return type:

tuple[int, Ellipsis]

tilelang.engine.host_mapped.validate_host_mapped_params(func, target, execution_backend, out_idx=None)

Reject unsupported execution and allocation paths before compilation.

Return type:

tuple[int, Ellipsis]