Searched refs:StreamExecutorMemoryAllocator (Results 1 – 15 of 15) sorted by relevance
27 StreamExecutorMemoryAllocator::StreamExecutorMemoryAllocator( in StreamExecutorMemoryAllocator() function in xla::StreamExecutorMemoryAllocator33 StatusOr<OwningDeviceMemory> StreamExecutorMemoryAllocator::Allocate( in Allocate()50 Status StreamExecutorMemoryAllocator::Deallocate(int device_ordinal, in Deallocate()62 StatusOr<se::StreamExecutor*> StreamExecutorMemoryAllocator::GetStreamExecutor( in GetStreamExecutor()80 bool StreamExecutorMemoryAllocator::AllowsAsynchronousDeallocation() const { in AllowsAsynchronousDeallocation()
79 class StreamExecutorMemoryAllocator : public DeviceMemoryAllocator {81 StreamExecutorMemoryAllocator(
182 std::unique_ptr<StreamExecutorMemoryAllocator> memory_allocator_;
37 xla::StreamExecutorMemoryAllocator allocator(platform, executors); in TEST()
159 memory_allocator_ = absl::make_unique<StreamExecutorMemoryAllocator>( in Backend()
42 class TestAllocator : public StreamExecutorMemoryAllocator {45 : StreamExecutorMemoryAllocator( in TestAllocator()
48 return StreamExecutorMemoryAllocator::Allocate(device_ordinal, size, in Allocate()59 return StreamExecutorMemoryAllocator::Deallocate(device_ordinal, mem); in Deallocate()
739 StreamExecutorMemoryAllocator allocator(platform, executors); in BM_DynamicSlice()
903 StreamExecutorMemoryAllocator allocator(platform, executors); in BM_LocalClientOverhead()
833 StreamExecutorMemoryAllocator allocator(platform, executors); in BM_ParallelFusion()
1268 StreamExecutorMemoryAllocator allocator(platform, executors); in BM_WhileLoop()
73 StreamExecutorMemoryAllocator allocator_;
179 absl::optional<StreamExecutorMemoryAllocator> se_allocator; in RunOnComputation()
166 optional<StreamExecutorMemoryAllocator> se_allocator; in PickBestAlgorithm()
274 StreamExecutorMemoryAllocator allocator( in ReplayComputation()