Managed reference to HardwareParamsNode.
More...
#include <search_task.h>
|
| HardwareParams (int num_cores, int vector_unit_bytes, int cache_line_bytes, int max_shared_memory_per_block, int max_local_memory_per_block, int max_threads_per_block, int max_vthread_extent, int warp_size) |
| The constructor. More...
|
|
| TVM_DEFINE_OBJECT_REF_METHODS (HardwareParams, ObjectRef, HardwareParamsNode) |
|
| TVM_DEFINE_OBJECT_REF_COW_METHOD (HardwareParamsNode) |
|
| 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...
|
|
◆ HardwareParams()
tvm::auto_scheduler::HardwareParams::HardwareParams |
( |
int |
num_cores, |
|
|
int |
vector_unit_bytes, |
|
|
int |
cache_line_bytes, |
|
|
int |
max_shared_memory_per_block, |
|
|
int |
max_local_memory_per_block, |
|
|
int |
max_threads_per_block, |
|
|
int |
max_vthread_extent, |
|
|
int |
warp_size |
|
) |
| |
The constructor.
- Parameters
-
num_cores | The number of cores. |
vector_unit_bytes | The width of vector units in bytes. |
cache_line_bytes | The size of cache line in bytes. |
max_shared_memory_per_block | The max amount of shared memory per block for GPU. |
max_local_memory_per_block | The max amount of local memory per block for GPU. |
max_threads_per_block | The max number of threads per block for GPU. |
max_vthread_extent | The max extent of vthread for GPU. |
warp_size | The warp size for GPU |
◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()
tvm::auto_scheduler::HardwareParams::TVM_DEFINE_OBJECT_REF_COW_METHOD |
( |
HardwareParamsNode |
| ) |
|
◆ TVM_DEFINE_OBJECT_REF_METHODS()
The documentation for this class was generated from the following file: