Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for host_memory_deallocate (0.16 sec)

  1. 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;
      se->destroy_stream = DestroyStream;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 21 20:09:00 UTC 2023
    - 8.7K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. tensorflow/c/experimental/stream_executor/stream_executor_test.cc

    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);
        deallocate_called = true;
      };
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 19:54:04 UTC 2024
    - 26.5K bytes
    - Viewed (0)
Back to top