Partition#
-
class nvmath.
device. Partition(*args)[source]# Partition of a global memory tensor into a partitioned tensor. This is used for accessing the C matrix when working with register fragments.
Note
Do not create directly, use
nvmath..device. Partitioner. partition_like_C() Refer to the cuBLASDx documentation for more details on how to use this class: https://docs.nvidia.com/cuda/cublasdx/api/other_tensors.html#partitioner-register-tensor-other-label
Methods