33 std::vector<int64_t> value_dsizes_host =
35 std::vector<int64_t> value_blocks_per_element_host =
37 n_values_ = value_blocks_per_element_host.size();
45 value_dsizes_host.data(),
48 value_blocks_per_element_host.data(),
58 std::vector<uint8_t *> value_ptrs(
n_values_);
60 value_ptrs[i] = value_buffers[i].GetDataPtr<uint8_t>();
61 cudaMemset(value_ptrs[i], 0,
capacity_ * value_dsizes_host[i]);
63 values_ =
static_cast<uint8_t **
>(
85 heap_[index - 1] = ptr;
#define OPEN3D_CUDA_CHECK(err)
Definition: CUDAUtils.h:47
Definition: CUDAHashBackendBufferAccessor.h:24
int64_t * value_blocks_per_element_
Definition: CUDAHashBackendBufferAccessor.h:108
__host__ void Setup(HashBackendBuffer &hashmap_buffer)
Definition: CUDAHashBackendBufferAccessor.h:26
int64_t * value_dsizes_
Definition: CUDAHashBackendBufferAccessor.h:107
uint8_t ** values_
Definition: CUDAHashBackendBufferAccessor.h:103
__device__ buf_index_t DeviceAllocate()
Definition: CUDAHashBackendBufferAccessor.h:79
int * heap_top_
Definition: CUDAHashBackendBufferAccessor.h:97
__device__ void DeviceFree(buf_index_t ptr)
Definition: CUDAHashBackendBufferAccessor.h:83
int64_t capacity_
Definition: CUDAHashBackendBufferAccessor.h:110
__host__ void Shutdown(const Device &device)
Definition: CUDAHashBackendBufferAccessor.h:73
uint8_t * keys_
Definition: CUDAHashBackendBufferAccessor.h:99
size_t n_values_
Definition: CUDAHashBackendBufferAccessor.h:102
__device__ void * GetKeyPtr(buf_index_t ptr)
Definition: CUDAHashBackendBufferAccessor.h:88
__device__ void * GetValuePtr(buf_index_t ptr, int value_idx=0)
Definition: CUDAHashBackendBufferAccessor.h:91
buf_index_t * heap_
Definition: CUDAHashBackendBufferAccessor.h:96
int64_t key_dsize_
Definition: CUDAHashBackendBufferAccessor.h:100
int64_t common_block_size_
Definition: CUDAHashBackendBufferAccessor.h:105
Definition: HashBackendBuffer.h:46
int64_t GetKeyDsize() const
Return key's data size in bytes.
Definition: HashBackendBuffer.cpp:77
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 & GetHeapTop()
Definition: HashBackendBuffer.cpp:99
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
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
static void MemcpyFromHost(void *dst_ptr, const Device &dst_device, const void *host_ptr, size_t num_bytes)
Same as Memcpy, but with host (CPU:0) as default src_device.
Definition: MemoryManager.cpp:77
static void * Malloc(size_t byte_size, const Device &device)
Definition: MemoryManager.cpp:22
static void Free(void *ptr, const Device &device)
Frees previously allocated memory at address ptr on device device.
Definition: MemoryManager.cpp:28
T * GetDataPtr()
Definition: Tensor.h:1143
void Synchronize()
Definition: CUDAUtils.cpp:58
uint32_t buf_index_t
Definition: HashBackendBuffer.h:44
Definition: PinholeCameraIntrinsic.cpp:16
Tensor cuda
Definition: HashBackendBuffer.h:49