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");
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:385
Base class of all attribute class.
Definition: attrs.h:102
Managed reference to a DeviceMesh.
Definition: global_info.h:62
Managed reference to a Placement.
Definition: struct_info.h:105
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
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.DistributionAttrs", DistributionAttrs, BaseAttrsNode)
distributed::DeviceMesh device_mesh
Definition: distributed.h:36
distributed::Placement placement
Definition: distributed.h:37
static void RegisterReflection()
Definition: distributed.h:39