MatmulPlanPreferences#

class nvmath.linalg.advanced.MatmulPlanPreferences(
reduction_scheme_mask: ReductionScheme | None = ReductionScheme.MASK,
max_waves_count: float | None = 0.0,
numerical_impl_mask: MatmulNumericalImplFlags | None = MatmulNumericalImplFlags.ALL,
limit: int = 8,
)[source]#

一个数据类，用于为 Matmul.plan() 方法和包装器函数 matmul() 提供选项。

reduction_scheme_mask (类型为

linalg.advanced.MatmulReductionScheme) : linalg.advanced.MatmulReductionScheme 中的枚举器，与按位运算符 | 结合使用。默认值为所有归约方案。

max_waves_count#

最大波计数。 选择大于 0 的值将排除设备利用率大于指定值的算法。 默认值为 0。

类型:

float

numerical_impl_mask (类型为

linalg.advanced.MatmulNumericalImplFlags) : linalg.advanced.MatmulNumericalImplFlags 中的枚举器，与按位运算符 | 结合使用。默认值是所有数值实现标志选项。

limit#

要考虑的算法数量。 如果未指定，将选择合适的默认值。

类型:

int

另请参阅

Matmul.plan(), matmul()