warp.tile

class warp.tile(

dtype,

shape,

storage='register',

layout='rowmajor',

strides=None,

owner=True,

)

A Warp tile object.

Parameters:

• dtype (Any)

• shape (tuple[int, ...] | list[int])

• storage (str)

• layout (str)

• strides (tuple[int, ...] | None)

• owner (builtins.bool)

dtype

The data type of the tile

Type:

DType

shape

Dimensions of the tile

Type:

Shape

storage

‘register’ or ‘shared’ memory storage

Type:

str

layout

‘rowmajor’ or ‘colmajor’ layout

Type:

str

strides

Number of tile elements between successive tile entries in each dimension

Type:

tuple[int]

size

Total number of tile elements

Type:

int

owner

Whether this tile aliases or owns its data

Type:

bool

__init__(

dtype,

shape,

storage='register',

layout='rowmajor',

strides=None,

owner=True,

)

Parameters:

• dtype (Any)

• shape (tuple[int, ...] | list[int])

• storage (str)

• layout (str)

• strides (tuple[int, ...] | None)

• owner (bool)

Methods

__init__(dtype, shape[, storage, layout, ...])
align(bytes)
cinit([requires_grad])
ctype()
round_up(bytes)
size_in_bytes()

Attributes

alignment

alignment = 16

ctype()

cinit(requires_grad=False)

size_in_bytes()

static round_up(bytes)

align(bytes)