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

PlanPreferenceAttribute

enum nvmath.bindings.cutensor.PlanPreferenceAttribute(value)

See cutensorPlanPreferenceAttribute_t.

Member Type:

int

Valid values are as follows:

AUTOTUNE_MODE = <PlanPreferenceAttribute.AUTOTUNE_MODE: 0>

CACHE_MODE = <PlanPreferenceAttribute.CACHE_MODE: 1>

INCREMENTAL_COUNT = <PlanPreferenceAttribute.INCREMENTAL_COUNT: 2>

ALGO = <PlanPreferenceAttribute.ALGO: 3>

KERNEL_RANK = <PlanPreferenceAttribute.KERNEL_RANK: 4>

JIT = <PlanPreferenceAttribute.JIT: 5>

GPU_ARCH = <PlanPreferenceAttribute.GPU_ARCH: 6>