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 "
49 static constexpr
const char*
_type_key =
"relax.attrs.AllReduceAttrs";
60 refl::ObjectDef<AllGatherAttrs>()
62 "The number of workers, also the number of parts the given buffer should be "
65 "Whether the allgather operation performs in group or globally or in group as "
69 static constexpr
const char*
_type_key =
"relax.attrs.AllGatherAttrs";
80 refl::ObjectDef<ScatterCollectiveAttrs>()
82 "The number of workers, also the number of parts the given buffer should be "
85 "The axis of the tensor to be scattered. The tensor will be chunked along "
89 static constexpr
const char*
_type_key =
"relax.attrs.ScatterCollectiveAttrs";
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:384
Base class of all attribute class.
Definition: attrs.h:103
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:54
static void RegisterReflection()
Definition: ccl.h:58
static constexpr const char * _type_key
Definition: ccl.h:69
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(AllGatherAttrs, BaseAttrsNode)
int num_workers
Definition: ccl.h:55
bool in_group
Definition: ccl.h:56
Attributes used in allreduce operators.
Definition: ccl.h:34
bool in_group
Definition: ccl.h:36
static void RegisterReflection()
Definition: ccl.h:38
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(AllReduceAttrs, BaseAttrsNode)
String op_type
Definition: ccl.h:35
static constexpr const char * _type_key
Definition: ccl.h:49
Attributes used in scatter operators.
Definition: ccl.h:74
static void RegisterReflection()
Definition: ccl.h:78
int num_workers
Definition: ccl.h:75
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(ScatterCollectiveAttrs, BaseAttrsNode)
int axis
Definition: ccl.h:76
static constexpr const char * _type_key
Definition: ccl.h:89