The nvmath.device module is experimental and potentially subject to future changes.

make_fragment_like#

nvmath.device.make_fragment_like(
tensor: OpaqueTensor,
dtype,
) → OpaqueTensor[source]#

make_fragment_like is a helper function for creating register fragments with the same layout as input tensor, but different dtype.

Parameters:

  • tensor (OpaqueTensor) – The input tensor to be used as a template for the fragment.

  • dtype – The data type of the fragment to be created.

Refer to the cuBLASDx documentation for more details on how to use this function: api/other_tensors.html#imported-tensor-utilities

On this page