tilelang.transform.metal.mark_host_metal_context¶

Attributes¶

tvm_call_packed_lowered

類別¶

MarkHostMetalContextMutator

函式¶

MarkHostMetalContext()

Module Contents¶

tilelang.transform.metal.mark_host_metal_context.tvm_call_packed_lowered¶

class tilelang.transform.metal.mark_host_metal_context.MarkHostMetalContextMutator(*args, **kwargs)¶

Bases: tvm.tir.PyStmtExprMutator

is_in_compute_scope = False¶

visit_attr_stmt_(stmt)¶

visit_evaluate_(op)¶

參數:: op (tvm.tir.Evaluate)

tilelang.transform.metal.mark_host_metal_context.MarkHostMetalContext()¶