- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for unified_memory_deallocate (0.2 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor.cc
} void* UnifiedMemoryAllocate(uint64 size) override { CHECK(stream_executor_->unified_memory_allocate); return stream_executor_->unified_memory_allocate(&device_, size); } void UnifiedMemoryDeallocate(void* mem) override { CHECK(stream_executor_->unified_memory_deallocate); stream_executor_->unified_memory_deallocate(&device_, mem); } absl::optional<AllocatorStats> GetAllocatorStats() override {
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_test.cc
static bool allocate_called = false; static bool deallocate_called = false; se_.unified_memory_allocate = [](const SP_Device* const device, uint64_t size) { allocate_called = true; return malloc(size); }; se_.unified_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)