Searched refs:event_pool (Results 1 – 12 of 12) sorted by relevance
| /external/rust/android-crates-io/crates/vulkano/src/sync/ |
| D | event.rs | 111 let handle = device.event_pool().lock().pop(); in from_pool()
226 self.device.event_pool().lock().push(raw_event); in drop()
349 fn event_pool() { in event_pool() function
352 assert_eq!(device.event_pool().lock().len(), 0); in event_pool()
355 assert_eq!(device.event_pool().lock().len(), 0); in event_pool()
359 assert_eq!(device.event_pool().lock().len(), 1); in event_pool()
361 assert_eq!(device.event_pool().lock().len(), 0); in event_pool()
|
| /external/rust/android-crates-io/crates/vulkano/src/device/ |
| D | mod.rs | 164 event_pool: Mutex<Vec<ash::vk::Event>>, field
419 event_pool: Mutex::new(Vec::new()), in new()
509 pub(crate) fn event_pool(&self) -> &Mutex<Vec<ash::vk::Event>> { in event_pool() method
510 &self.event_pool in event_pool()
631 for &raw_event in self.event_pool.lock().iter() { in drop()
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | CachingHostAllocator.cpp | 128 static auto* event_pool = new EventPool(); in create_event_internal() local
129 return event_pool->get(idx); in create_event_internal()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | BUILD | 39 name = "event_pool",
40 srcs = ["event_pool.cc"],
41 hdrs = ["event_pool.h"],
85 ":event_pool",
139 ":event_pool",
296 ":event_pool",
|
| D | local_device_state.h | 105 EventPool& event_pool() { return event_pool_; } in event_pool() function
|
| D | pjrt_stream_executor_client.cc | 413 local_device->event_pool().ThenAllocateAndRecordEvent( in AllocateDestinationBuffer()
453 local_device->event_pool().ThenAllocateAndRecordEvent( in AllocateDestinationBuffer()
495 local_device->event_pool().ThenAllocateAndRecordEvent(copy_stream); in AddDestinationBufferSynchronization()
612 local_device->event_pool().AllocateEvent(stream->parent()); in logical_on_device_shape()
1336 local_device->event_pool().AllocateEvent(stream->parent()); in ToLiteral()
1346 local_device->event_pool().ThenRecordEvent(stream, event_or.ValueOrDie()); in ToLiteral()
1731 local_device->event_pool().ThenAllocateAndRecordEvent(stream); in MakeTupleHelper()
2140 device_state->event_pool().ThenAllocateAndRecordEvent(stream); in ExecuteHelper()
|
| /external/pytorch/aten/src/ATen/mps/ |
| D | MPSEvent.mm | 253 …static std::shared_ptr<MPSEventPool> event_pool = std::make_shared<MPSEventPool>(getDefaultMPSStre…
254 return event_pool;
|
| /external/pytorch/torch/csrc/distributed/c10d/ |
| D | ProcessGroupUCC.hpp | 48 std::queue<std::unique_ptr<at::cuda::CUDAEvent>> event_pool; member
|
| D | ProcessGroupUCC.cpp | 213 ep->event_pool.push(std::move(fence)); in ~WorkUCC()
753 if (ep.event_pool.empty()) { in getPooledEvent()
756 ev = std::move(ep.event_pool.front()); in getPooledEvent()
757 ep.event_pool.pop(); in getPooledEvent()
|
| /external/pytorch/c10/cuda/ |
| D | CUDACachingAllocator.cpp | 2956 static auto* event_pool = new EventPool(); in create_event_internal() local
2957 return event_pool->get(idx); in create_event_internal()
|
| /external/bcc/libbpf-tools/powerpc/ |
| D | vmlinux_600.h | 53098 struct event_pool { struct
53112 struct event_pool pool; argument
|
| D | vmlinux.h | 53098 struct event_pool { struct
53112 struct event_pool pool; argument
|