tvm
|
Managed reference to HardwareParamsNode. More...
#include <search_task.h>
Public Member Functions | |
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 > | |
const ObjectType * | as () const |
Try to downcast the internal Object to a raw pointer of a corresponding type. More... | |
Additional Inherited Members | |
![]() | |
using | ContainerType = Object |
type indicate the container type. More... | |
![]() | |
static constexpr bool | _type_is_nullable = true |
![]() | |
Object * | get_mutable () const |
![]() | |
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... | |
![]() | |
ObjectPtr< Object > | data_ |
Internal pointer that backs the reference. More... | |
Managed reference to HardwareParamsNode.
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.
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::auto_scheduler::HardwareParams::TVM_DEFINE_OBJECT_REF_COW_METHOD | ( | HardwareParamsNode | ) |
tvm::auto_scheduler::HardwareParams::TVM_DEFINE_OBJECT_REF_METHODS | ( | HardwareParams | , |
ObjectRef | , | ||
HardwareParamsNode | |||
) |