- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for SynchronizeAllActivity (0.26 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor_test_util.cc
uint64_t size, TF_Status* const status) {} void BlockHostForEvent(const SP_Device* const device, SP_Event event, TF_Status* const status) {} void SynchronizeAllActivity(const SP_Device* const device, TF_Status* const status) {} TF_Bool HostCallback(const SP_Device* const device, SP_Stream stream, SE_StatusCallbackFn const callback_fn,
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/compiler/jit/xla_device_compiler_client.cc
} void XlaDeviceCompilerClient::WaitForProgramsToFinish() { if (client_ == nullptr) return; for (auto* executor : client_->backend().stream_executors()) { bool ok = executor->SynchronizeAllActivity(); if (!ok) { LOG(ERROR) << "Error synchronizing activity while waiting for all " "programs to complete"; } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
ASSERT_FALSE(block_host_until_done_called); TF_ASSERT_OK(stream->BlockHostUntilDone()); ASSERT_TRUE(block_host_until_done_called); } TEST_F(StreamExecutorTest, SynchronizeAllActivity) { static bool synchronize_all_called = false; se_.synchronize_all_activity = [](const SP_Device* const device, TF_Status* const status) { TF_SetStatus(status, TF_OK, "");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
tf_stats.largest_free_block_bytes = se_stats->largest_free_block_bytes; return tf_stats; } bool XlaDeviceAllocator::ClearStats() { if (!stream_executor_->SynchronizeAllActivity()) { return false; } return stream_executor_->ClearAllocatorStats(); } XlaDeviceContext::XlaDeviceContext( std::shared_ptr<se::Stream> compute_stream,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
stats.bytes_reservable_limit = c_stats.bytes_reservable_limit; } stats.largest_free_block_bytes = c_stats.largest_free_block_bytes; return stats; } bool SynchronizeAllActivity() override { OwnedTFStatus c_status(TF_NewStatus()); stream_executor_->synchronize_all_activity(&device_, c_status.get()); if (TF_GetCode(c_status.get()) != TF_OK) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0)