tilelang.transform.metal.mark_host_metal_context

Attributes

tvm_call_packed_lowered

Classes

MarkHostMetalContextMutator

Functions

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)

Parameters:

op (tvm.tir.Evaluate)

tilelang.transform.metal.mark_host_metal_context.MarkHostMetalContext()