ContractionPlanPreference#

class nvmath.tensor.ContractionPlanPreference(contraction)[source]#

An interface to configure nvmath.tensor.BinaryContraction.plan() and nvmath.tensor.TernaryContraction.plan(). The current configuration can also be queried.

Methods

__init__(contraction)[source]#: ctor for internal use only.

Attributes

algo#: Query the algo. See the cuTensor documentation for more information.

autotune_mode#: Query the autotune mode. See the cuTensor documentation for more information.

cache_mode#: Query the cache mode. See the cuTensor documentation for more information.

incremental_count#: Query the incremental count. See the cuTensor documentation for more information.

jit#: Query the jit compilation mode. See the cuTensor documentation for more information.

kernel_rank#: Query the kernel rank. See the cuTensor documentation for more information.