24 #ifndef TVM_RELAX_ATTRS_DISTRIBUTED_H_
25 #define TVM_RELAX_ATTRS_DISTRIBUTED_H_
41 refl::ObjectDef<DistributionAttrs>()
43 "The device mesh of a tensor's distribution plan")
45 "The placement of a tensor's distribution plan");
48 static constexpr
const char*
_type_key =
"relax.attrs.DistributionAttrs";
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:384
Base class of all attribute class.
Definition: attrs.h:103
Managed reference to a DeviceMesh.
Definition: global_info.h:64
Managed reference to a Placement.
Definition: struct_info.h:107
Struct info for DTensor (Distributed Tensor)
Definition: repr_printer.h:91
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
Data structure for distributed inference.
Attributes for redistribute and annotate_sharding operator.
Definition: distributed.h:35
distributed::DeviceMesh device_mesh
Definition: distributed.h:36
distributed::Placement placement
Definition: distributed.h:37
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(DistributionAttrs, BaseAttrsNode)
static constexpr const char * _type_key
Definition: distributed.h:48
static void RegisterReflection()
Definition: distributed.h:39