Helper class to calculate shared storage size.
For further details, please refer to cuBLASDx documentation.
Methods
-
__init__()
-
add(alignment: int, matrix_size_bytes: int) → None[source]
-
add(
- alignment: int,
- elem_size: int,
- num_elements: int,
) → None
-
add(
- alignment: int,
- elem_size: int,
- layout: Layout,
) → None
-
get()[source]
