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¶
abstract get_code()¶
Return type:

List[str]

property panel_width¶
class tilelang.carver.roller.rasterization.NoRasterization¶

Bases: Rasterization

__repr__()¶
Return type:

str

get_code()¶
Return type:

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__()¶
Return type:

str

abstract get_code()¶
Return type:

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__()¶
Return type:

str

get_device_function()¶
Return type:

str

get_code(panel_width=None)¶
Parameters:

panel_width (int)

Return type:

List[str]