24 #ifndef TVM_RELAX_ATTRS_CCL_H_
25 #define TVM_RELAX_ATTRS_CCL_H_
27 #include <tvm/ffi/reflection/registry.h>
40 refl::ObjectDef<AllReduceAttrs>()
42 "The type of reduction operation to be applied to the input data. Now only sum is "
45 "Whether the reduction operation performs in group or globally or in group as "
58 refl::ObjectDef<AllGatherAttrs>()
60 "The number of workers, also the number of parts the given buffer should be "
63 "Whether the allgather operation performs in group or globally or in group as "
76 refl::ObjectDef<ScatterCollectiveAttrs>()
78 "The number of workers, also the number of parts the given buffer should be "
81 "The axis of the tensor to be scattered. The tensor will be chunked along "
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:385
Base class of all attribute class.
Definition: attrs.h:102
Definition: repr_printer.h:91
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
Attributes used in allgather operators.
Definition: ccl.h:52
static void RegisterReflection()
Definition: ccl.h:56
int num_workers
Definition: ccl.h:53
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.AllGatherAttrs", AllGatherAttrs, BaseAttrsNode)
bool in_group
Definition: ccl.h:54
Attributes used in allreduce operators.
Definition: ccl.h:34
bool in_group
Definition: ccl.h:36
static void RegisterReflection()
Definition: ccl.h:38
ffi::String op_type
Definition: ccl.h:35
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.AllReduceAttrs", AllReduceAttrs, BaseAttrsNode)
Attributes used in scatter operators.
Definition: ccl.h:70
static void RegisterReflection()
Definition: ccl.h:74
int num_workers
Definition: ccl.h:71
int axis
Definition: ccl.h:72
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.ScatterCollectiveAttrs", ScatterCollectiveAttrs, BaseAttrsNode)