tilelang.tileop.gemm.gemm_base¶
Classes¶
Module Contents¶
- class tilelang.tileop.gemm.gemm_base.GemmBase¶
Bases:
object- gemm_node: tvm.ir.base.Node¶
- abstract infer_layout(target, thread_nums)¶
- Parameters:
target (tvm.target.Target)
thread_nums (int)
- abstract lower(target, thread_nums, thread_var)¶
- Parameters:
target (tvm.target.Target)
thread_nums (int)
thread_var (tvm.tir.Var)
- is_gemm_ss()¶
- Return type:
bool
- is_gemm_sr()¶
- Return type:
bool
- is_gemm_rs()¶
- Return type:
bool
- is_gemm_rr()¶
- Return type:
bool
- property M: int¶
- Return type:
int
- property N: int¶
- Return type:
int
- property K: int¶
- Return type:
int
- property trans_A: bool¶
- Return type:
bool
- property trans_B: bool¶
- Return type:
bool
- property in_dtype: str¶
- Return type:
str
- property accum_dtype: str¶
- Return type:
str
- property chunk: int¶
- Return type:
int
- property A: tvm.tir.Buffer¶
- Return type:
tvm.tir.Buffer
- property B: tvm.tir.Buffer¶
- Return type:
tvm.tir.Buffer
- property C: tvm.tir.Buffer¶
- Return type:
tvm.tir.Buffer
- property APtr: tvm.tir.PrimExpr¶
- Return type:
tvm.tir.PrimExpr
- property BPtr: tvm.tir.PrimExpr¶
- Return type:
tvm.tir.PrimExpr
- property CPtr: tvm.tir.PrimExpr¶
- Return type:
tvm.tir.PrimExpr
- property stride_A: int¶
- Return type:
int
- property stride_B: int¶
- Return type:
int
- property offset_A: int¶
- Return type:
int
- property offset_B: int¶
- Return type:
int
- property clear_accum: bool¶
- Return type:
bool
- property k_pack: int¶
- Return type:
int
- property wg_wait: int¶
- Return type:
int
- property policy: tilelang.ir.GemmWarpPolicy¶
- Return type: