tilelang.metal.transform.mark_host_metal_context

Attributes

tvm_call_packed_lowered

Classes

MarkHostMetalContextMutator

Functions

MarkHostMetalContext()

Module Contents

tilelang.metal.transform.mark_host_metal_context.tvm_call_packed_lowered

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

Bases: tvm.tirx.PyStmtExprMutator

is_in_compute_scope = False

visit_attr_stmt_(stmt)

visit_evaluate_(op)

Parameters:

op (tvm.tirx.Evaluate)

tilelang.metal.transform.mark_host_metal_context.MarkHostMetalContext()