tilelang.carver.roller.rasterization module#

Rasteration Plan For L2 Cache Locality

class tilelang.carver.roller.rasterization.NoRasterization#

Bases: Rasterization

get_code() List[str]#
class tilelang.carver.roller.rasterization.Rasterization#

Bases: object

get_code() List[str]#
property panel_width#
panel_width_ = None#
class tilelang.carver.roller.rasterization.Rasterization2DColumn(panel_width=4)#

Bases: Rasterization

Rasterization by Column, each column line width is panel_width

_

| | |
| | |

|_| |_|

get_code(panel_width: Optional[int] = None) List[str]#
get_device_function() str#
class tilelang.carver.roller.rasterization.Rasterization2DRow(panel_width=4)#

Bases: Rasterization

Rasterization by Row, each Row line width is panel_width

_________|

|_________ __________|

get_code() List[str]#