tilelang.carver.roller.rasterization

Rasteration Plan For L2 Cache Locality

Classes

Rasterization
NoRasterization
Rasterization2DRow	Rasterization by Row, each Row line width is panel_width
Rasterization2DColumn	Rasterization by Column, each column line width is panel_width

Module Contents

class tilelang.carver.roller.rasterization.Rasterization

panel_width_ = None

abstractmethod get_code()

返回类型:

list[str]

property panel_width

class tilelang.carver.roller.rasterization.NoRasterization

Bases: Rasterization

__repr__()

返回类型:

str

get_code()

返回类型:

list[str]

class tilelang.carver.roller.rasterization.Rasterization2DRow(panel_width=4)

Bases: Rasterization

Rasterization by Row, each Row line width is panel_width

_________|

|_________ __________|

panel_width_ = 4

__repr__()

返回类型:

str

abstractmethod get_code()

返回类型:

list[str]

class tilelang.carver.roller.rasterization.Rasterization2DColumn(panel_width=4)

Bases: Rasterization

Rasterization by Column, each column line width is panel_width

_

| | |

| | |

|_| |_|

panel_width_ = 4

__repr__()

返回类型:

str

get_device_function()

返回类型:

str

get_code(panel_width=None)

参数:

panel_width (int)

返回类型:

list[str]