25 #ifndef TVM_RELAX_DISTRIBUTED_GLOBAL_INFO_H_
26 #define TVM_RELAX_DISTRIBUTED_GLOBAL_INFO_H_
32 namespace distributed {
50 refl::ObjectDef<DeviceMeshNode>()
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:54
Range container
Definition: expr.h:689
Definition: global_info.h:37
ffi::Shape shape
logical shape of the mesh
Definition: global_info.h:40
static void RegisterReflection()
Definition: global_info.h:48
ffi::Optional< Range > device_range
Optionally use range to represent device_ids.
Definition: global_info.h:46
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.distributed.DeviceMesh", DeviceMeshNode, GlobalInfoNode)
ffi::Array< Integer > device_ids
device ids in the mesh
Definition: global_info.h:43
Managed reference to a DeviceMesh.
Definition: global_info.h:62
DeviceMesh(ffi::Shape shape, Range device_range)
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(DeviceMesh, GlobalInfo, DeviceMeshNode)
DeviceMesh(ffi::Shape shape, ffi::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:1960
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37