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

ComputeDesc#

class nvmath.bindings.cutensor.ComputeDesc[source]#

See cutensorComputeDescriptor_t.

Methods

classmethod COMPUTE_16BF(cls)[source]#

classmethod COMPUTE_16F(cls)[source]#

classmethod COMPUTE_32F(cls)[source]#

classmethod COMPUTE_3XTF32(cls)[source]#

classmethod COMPUTE_4X16F(cls)[source]#

classmethod COMPUTE_64F(cls)[source]#

classmethod COMPUTE_8XINT8(cls)[source]#

classmethod COMPUTE_9X16BF(cls)[source]#

classmethod COMPUTE_TF32(cls)[source]#