nvmath.bindings.cublasLt.ReductionScheme

enum nvmath.bindings.cublasLt.ReductionScheme(value)[source]

See cublasLtReductionScheme_t.

Member Type:

int

Valid values are as follows:

NONE = <ReductionScheme.NONE: 0>
INPLACE = <ReductionScheme.INPLACE: 1>
COMPUTE_TYPE = <ReductionScheme.COMPUTE_TYPE: 2>
OUTPUT_TYPE = <ReductionScheme.OUTPUT_TYPE: 4>
MASK = <ReductionScheme.MASK: 7>