MCPcopy Create free account
hub / github.com/apache/arrow / AllocateCudaHostBuffer

Function AllocateCudaHostBuffer

cpp/src/arrow/gpu/cuda_memory.cc:474–478  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

472// ----------------------------------------------------------------------
473
474Result<std::shared_ptr<CudaHostBuffer>> AllocateCudaHostBuffer(int device_number,
475 const int64_t size) {
476 ARROW_ASSIGN_OR_RAISE(auto manager, CudaDeviceManager::Instance());
477 return manager->AllocateHost(device_number, size);
478}
479
480Result<uintptr_t> GetDeviceAddress(const uint8_t* cpu_data,
481 const std::shared_ptr<CudaContext>& ctx) {

Callers 2

SetBufferSizeMethod · 0.85
TEST_FFunction · 0.85

Calls 1

AllocateHostMethod · 0.45

Tested by 1

TEST_FFunction · 0.68