tvm
Public Member Functions | List of all members
tvm::auto_scheduler::RPCRunner Class Reference

Managed reference to RPCRunnerNode. More...

#include <measure.h>

Inheritance diagram for tvm::auto_scheduler::RPCRunner:
Collaboration diagram for tvm::auto_scheduler::RPCRunner:

Public Member Functions

 RPCRunner (const String &key, const String &host, int port, int priority, int n_parallel, int timeout, int number, int repeat, int min_repeat_ms, double cooldown_interval, bool enable_cpu_cache_flush, int device)
 The constructor. See the corresponding class in python/tvm/auto_scheduler/measure.py for more detailed parameter explanation. More...
 
 TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS (RPCRunner, ProgramRunner, RPCRunnerNode)
 
- Public Member Functions inherited from tvm::auto_scheduler::ProgramRunner
 TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS (ProgramRunner, ObjectRef, ProgramRunnerNode)
 
- Public Member Functions inherited from tvm::runtime::ObjectRef
 ObjectRef ()=default
 default constructor More...
 
 ObjectRef (ObjectPtr< Object > data)
 Constructor from existing object ptr. More...
 
bool same_as (const ObjectRef &other) const
 Comparator. More...
 
bool operator== (const ObjectRef &other) const
 Comparator. More...
 
bool operator!= (const ObjectRef &other) const
 Comparator. More...
 
bool operator< (const ObjectRef &other) const
 Comparator. More...
 
bool defined () const
 
const Objectget () const
 
const Objectoperator-> () const
 
bool unique () const
 
int use_count () const
 
template<typename ObjectType , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>>
const ObjectType * as () const
 Try to downcast the internal Object to a raw pointer of a corresponding type. More...
 
template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>>
Optional< ObjectRefType > as () const
 Try to downcast the ObjectRef to a Optional<T> of the requested type. More...
 

Additional Inherited Members

- Public Types inherited from tvm::runtime::ObjectRef
using ContainerType = Object
 type indicate the container type. More...
 
- Static Public Attributes inherited from tvm::runtime::ObjectRef
static constexpr bool _type_is_nullable = true
 
- Protected Member Functions inherited from tvm::runtime::ObjectRef
Objectget_mutable () const
 
- Static Protected Member Functions inherited from tvm::runtime::ObjectRef
template<typename T >
static T DowncastNoCheck (ObjectRef ref)
 Internal helper function downcast a ref without check. More...
 
static void FFIClearAfterMove (ObjectRef *ref)
 Clear the object ref data field without DecRef after we successfully moved the field. More...
 
template<typename ObjectType >
static ObjectPtr< ObjectType > GetDataPtr (const ObjectRef &ref)
 Internal helper function get data_ as ObjectPtr of ObjectType. More...
 
- Protected Attributes inherited from tvm::runtime::ObjectRef
ObjectPtr< Objectdata_
 Internal pointer that backs the reference. More...
 

Detailed Description

Managed reference to RPCRunnerNode.

See also
RPCRunnerNode

Constructor & Destructor Documentation

◆ RPCRunner()

tvm::auto_scheduler::RPCRunner::RPCRunner ( const String key,
const String host,
int  port,
int  priority,
int  n_parallel,
int  timeout,
int  number,
int  repeat,
int  min_repeat_ms,
double  cooldown_interval,
bool  enable_cpu_cache_flush,
int  device 
)

The constructor. See the corresponding class in python/tvm/auto_scheduler/measure.py for more detailed parameter explanation.

Parameters
keyThe key of the device registered in the RPC tracker.
hostThe host address of the RPC Tracker.
portThe port of RPC Tracker.
priorityThe priority of this run request, larger is more prior.
n_parallelThe number of tasks run in parallel.
timeoutTimeout of a run.
numberThe number of times to run the generated code for taking average.
repeatThe number of times to repeat the measurement.
min_repeat_msThe minimum duration of one repeat in milliseconds.
cooldown_intervalThe cool down interval between two measurements.
enable_cpu_cache_flushWhether to flush cache on CPU between repeated measurements.
deviceWhich device to run on if multiple are available.

Member Function Documentation

◆ TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS()

tvm::auto_scheduler::RPCRunner::TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS ( RPCRunner  ,
ProgramRunner  ,
RPCRunnerNode   
)

The documentation for this class was generated from the following file: