24#ifdef BUILD_CUDA_MODULE
25void CUDAResetHeap(Tensor &heap);
50 std::atomic<int>
cpu = {0};
55 std::vector<int64_t> value_dsizes,
Definition HashBackendBuffer.h:46
int64_t GetKeyDsize() const
Return key's data size in bytes.
Definition HashBackendBuffer.cpp:77
Tensor key_buffer_
Definition HashBackendBuffer.h:102
void ResetHeap()
Reset the heap and heap top.
Definition HashBackendBuffer.cpp:60
std::vector< Tensor > value_buffers_
Definition HashBackendBuffer.h:103
Tensor heap_
Definition HashBackendBuffer.h:99
Device GetDevice() const
Return device of the buffer.
Definition HashBackendBuffer.cpp:73
Tensor GetIndexHeap() const
Return the index heap tensor.
Definition HashBackendBuffer.cpp:97
int64_t GetCapacity() const
Return capacity of the buffer.
Definition HashBackendBuffer.cpp:75
HeapTop heap_top_
Definition HashBackendBuffer.h:100
int GetHeapTopIndex() const
Return the current heap top.
Definition HashBackendBuffer.cpp:103
int64_t common_block_size_
Definition HashBackendBuffer.h:105
HeapTop & GetHeapTop()
Definition HashBackendBuffer.cpp:99
std::vector< int64_t > blocks_per_element_
Definition HashBackendBuffer.h:106
Tensor GetKeyBuffer() const
Return the key buffer tensor.
Definition HashBackendBuffer.cpp:110
std::vector< int64_t > GetValueDsizes() const
Return value's data sizes in bytes.
Definition HashBackendBuffer.cpp:81
Tensor GetValueBuffer(size_t i=0) const
Return the selected value buffer tensor at index i.
Definition HashBackendBuffer.cpp:116
int64_t GetCommonBlockSize() const
Get the common block size divisor of all values types.
Definition HashBackendBuffer.cpp:89
std::vector< Tensor > GetValueBuffers() const
Return the value buffer tensors.
Definition HashBackendBuffer.cpp:112
std::vector< int64_t > GetValueBlocksPerElement() const
Return value's data sizes in the unit of common block size divisor.
Definition HashBackendBuffer.cpp:93
void CPUResetHeap(Tensor &heap)
Definition CPUHashBackendBuffer.cpp:13
uint32_t buf_index_t
Definition HashBackendBuffer.h:44
Definition PinholeCameraIntrinsic.cpp:16
Definition HashBackendBuffer.h:48
std::atomic< int > cpu
Definition HashBackendBuffer.h:50
Tensor cuda
Definition HashBackendBuffer.h:49