tilelang.ir¶

Classes¶

`Fill`
`AtomicAdd`
`Copy`
`Conv2DIm2ColOp`
`GemmWarpPolicy`
`GemmSPWarpPolicy`
`Gemm`
`GemmSP`
`FinalizeReducerOp`
`ParallelOp`
`ReduceOp`
`CumSumOp`
`RegionOp`
`ReduceType`

Module Contents¶

class tilelang.ir.Fill¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.AtomicAdd¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.Copy¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.Conv2DIm2ColOp¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.GemmWarpPolicy¶

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

policy_type: int¶

m_warp: int¶

n_warp: int¶

compute_warp_partition(M, N, block_size, target, gemm_inst)¶

参数:

M (int)
N (int)
block_size (int)
target (tvm.target.Target)
gemm_inst (tilelang.tileop.gemm.inst.GemmInst)

class tilelang.ir.GemmSPWarpPolicy¶

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

policy_type: int¶

m_warp: int¶

n_warp: int¶

compute_warp_partition(M, N, block_size, target, gemm_inst, bits)¶

参数:

M (int)
N (int)
block_size (int)
target (tvm.target.Target)
gemm_inst (tilelang.tileop.gemm.inst.GemmInst)
bits (int)

class tilelang.ir.Gemm¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.GemmSP¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.FinalizeReducerOp¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.ParallelOp¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.ReduceOp¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.CumSumOp¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.RegionOp¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

class tilelang.ir.ReduceType¶: Bases: tvm.ir.base.Node, tvm.runtime.Scriptable