Enum TensorRtCollectiveOperation

Namespace

JYPPX.TensorRtSharp

Assembly

JYPPX.TensorRtSharp.dll

Selects the collective communication primitive for a TensorRT 11 DistCollective layer. 选择 TensorRT 11 DistCollective 层的集合通信原语。

public enum TensorRtCollectiveOperation

Fields

AllGather = 1

Represents the AllGather value of TensorRtCollectiveOperation. 表示 TensorRtCollectiveOperation 的 AllGather 取值。

AllReduce = 0

Represents the AllReduce value of TensorRtCollectiveOperation. 表示 TensorRtCollectiveOperation 的 AllReduce 取值。

AllToAll = 5

Represents the AllToAll value of TensorRtCollectiveOperation. 表示 TensorRtCollectiveOperation 的 AllToAll 取值。

Broadcast = 2

Represents the Broadcast value of TensorRtCollectiveOperation. 表示 TensorRtCollectiveOperation 的 Broadcast 取值。

Gather = 6

Represents the Gather value of TensorRtCollectiveOperation. 表示 TensorRtCollectiveOperation 的 Gather 取值。

Reduce = 3

Represents the Reduce value of TensorRtCollectiveOperation. 表示 TensorRtCollectiveOperation 的 Reduce 取值。

ReduceScatter = 4

Represents the ReduceScatter value of TensorRtCollectiveOperation. 表示 TensorRtCollectiveOperation 的 ReduceScatter 取值。

Scatter = 7

Represents the Scatter value of TensorRtCollectiveOperation. 表示 TensorRtCollectiveOperation 的 Scatter 取值。