25 #ifndef TVM_RELAX_DISTRIBUTED_GLOBAL_INFO_H_
26 #define TVM_RELAX_DISTRIBUTED_GLOBAL_INFO_H_
32 namespace distributed {
50 refl::ObjectDef<DeviceMeshNode>()
56 static constexpr
const char*
_type_key =
"relax.distributed.DeviceMesh";
GlobalInfo are globally static object that are referred by the IR itself. Base node for all global in...
Definition: global_info.h:43
Managed reference to GlobalInfoNode.
Definition: global_info.h:56
Range container
Definition: expr.h:698
Definition: global_info.h:37
static constexpr const char * _type_key
Definition: global_info.h:56
ffi::Shape shape
logical shape of the mesh
Definition: global_info.h:40
TVM_DECLARE_FINAL_OBJECT_INFO(DeviceMeshNode, GlobalInfoNode)
Array< Integer > device_ids
device ids in the mesh
Definition: global_info.h:43
static void RegisterReflection()
Definition: global_info.h:48
Optional< Range > device_range
Optionally use range to represent device_ids.
Definition: global_info.h:46
Managed reference to a DeviceMesh.
Definition: global_info.h:64
TVM_DEFINE_OBJECT_REF_METHODS(DeviceMesh, GlobalInfo, DeviceMeshNode)
DeviceMesh(ffi::Shape shape, Range device_range)
DeviceMesh(ffi::Shape shape, Array< Integer > device_ids)
IRModule that holds the functions and type definitions.
Definition: repr_printer.h:91
Tensor shape(const Tensor &src, DataType dtype, const std::string name="T_shape", const std::string tag=kInjective)
Get the shape of input tensor.
Definition: transform.h:1945
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37