Managed reference to LocalRunnerNode. More...
#include <measure.h>
Inheritance diagram for tvm::auto_scheduler::LocalRunner:
Collaboration diagram for tvm::auto_scheduler::LocalRunner:
Public Member Functions | |
| LocalRunner (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 (LocalRunner, ProgramRunner, LocalRunnerNode) | |
 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 Object * | get () const |
| const Object * | operator-> () 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 | |
| Object * | get_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< Object > | data_ |
| Internal pointer that backs the reference. More... | |
Detailed Description
Managed reference to LocalRunnerNode.
- See also
- LocalRunnerNode
Constructor & Destructor Documentation
◆ LocalRunner()
| tvm::auto_scheduler::LocalRunner::LocalRunner | ( | 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
-
timeout The timeout limit (in second) for each run. This is used in a wrapper of the multiprocessing.Process.join(). number The number of times to run the generated code for taking average. repeat The number of times to repeat the measurement. min_repeat_ms The minimum duration of one repeat in milliseconds. cooldown_interval The cool down interval between two measurements. enable_cpu_cache_flush Whether to flush cache on CPU between repeated measurements. device Which device to run on if multiple are available.
Member Function Documentation
◆ TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS()
| tvm::auto_scheduler::LocalRunner::TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS | ( | LocalRunner | , |
| ProgramRunner | , | ||
| LocalRunnerNode | |||
| ) |
The documentation for this class was generated from the following file:
- include/tvm/auto_scheduler/measure.h
