The nvmath.device module is experimental and potentially subject to future changes. SharedStorageCalc# class nvmath.device.SharedStorageCalc[source]# Helper class to calculate shared storage size. For further details, please refer to cuBLASDx documentation. Methods 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]#
