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

Function AllocateCudaHostBuffer

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

Source from the content-addressed store, hash-verified

462// ----------------------------------------------------------------------
463
464Result<std::shared_ptr<CudaHostBuffer>> AllocateCudaHostBuffer(int device_number,
465 const int64_t size) {
466 ARROW_ASSIGN_OR_RAISE(auto manager, CudaDeviceManager::Instance());
467 return manager->AllocateHost(device_number, size);
468}
469
470Result<uintptr_t> GetDeviceAddress(const uint8_t* cpu_data,
471 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