- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for host_memory_allocate (0.4 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor_test_util.cc
void PopulateDefaultStreamExecutor(SP_StreamExecutor* se) { *se = {SP_STREAMEXECUTOR_STRUCT_SIZE}; se->allocate = Allocate; se->deallocate = Deallocate; se->host_memory_allocate = HostMemoryAllocate; se->host_memory_deallocate = HostMemoryDeallocate; se->get_allocator_stats = GetAllocatorStats; se->device_memory_usage = DeviceMemoryUsage; se->create_stream = CreateStream;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 21 20:09:00 UTC 2023 - 8.7K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, deallocate); TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, get_allocator_stats); TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, host_memory_allocate); TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, host_memory_deallocate); if (platform.supports_unified_memory) { TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, unified_memory_allocate);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.h
// Memory allocated in this manner is required for use in asynchronous memcpy // operations, such as `memcpy_dtoh`. void* (*host_memory_allocate)(const SP_Device* device, uint64_t size); // Deallocates a region of host memory allocated by `host_memory_allocate`. void (*host_memory_deallocate)(const SP_Device* device, void* mem); // Allocates unified memory space of the given size, if supported. Unified
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 24 08:40:35 UTC 2022 - 21.6K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
executor->Deallocate(&mem); } TEST_F(StreamExecutorTest, HostMemoryAllocate) { static bool allocate_called = false; static bool deallocate_called = false; se_.host_memory_allocate = [](const SP_Device* const device, uint64_t size) { allocate_called = true; return malloc(size); }; se_.host_memory_deallocate = [](const SP_Device* const device, void* mem) { free(mem);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K bytes - Viewed (0)