tilelang.ir

Classes

Fill	Base class of all IR Nodes.
AtomicAdd	Base class of all IR Nodes.
Copy	Base class of all IR Nodes.
Conv2DIm2ColOp	Base class of all IR Nodes.
GemmWarpPolicy	Base class of all IR Nodes.
Gemm	Base class of all IR Nodes.
GemmSP	Base class of all IR Nodes.
FinalizeReducerOp	Base class of all IR Nodes.
ParallelOp	Base class of all IR Nodes.
ReduceOp	Base class of all IR Nodes.
CumSumOp	Base class of all IR Nodes.
RegionOp	Base class of all IR Nodes.
ReduceType	Base class of all IR Nodes.

Module Contents

class tilelang.ir.Fill

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.AtomicAdd

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.Copy

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.Conv2DIm2ColOp

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.GemmWarpPolicy

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

policy_type: int

m_warp: int

n_warp: int

compute_warp_partition(M, N, block_size, target, is_wgmma)

Parameters:

• M (int)

• N (int)

• block_size (int)

• target (tvm.target.Target)

• is_wgmma (bool)

class tilelang.ir.Gemm

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.GemmSP

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.FinalizeReducerOp

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.ParallelOp

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.ReduceOp

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.CumSumOp

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.RegionOp

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.

class tilelang.ir.ReduceType

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

Base class of all IR Nodes.