- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for RefreshStatus (0.33 sec)
-
tensorflow/compiler/jit/xla_device.h
TF_LOCKS_EXCLUDED(mu_); bool AllowsSyncOnCompletion() const override TF_LOCKS_EXCLUDED(mu_); // Installs an error handling callback when RefreshStatus sees !status.ok(). void SetHandleDeviceErrorCallback(std::function<Status()> callback); Status RefreshStatus() override TF_LOCKS_EXCLUDED(mu_); private: absl::StatusOr<xla::LocalClient*> GetOrCreateClient() const;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tensor.cc
streams_defined_on_ = {stream}; } Status XlaTensor::RefreshStatusOfStreams() { mutex_lock lock(mu_); Status status; for (se::Stream* stream : streams_defined_on_) { status.Update(stream->RefreshStatus()); } return status; } // The pointer tag, OR-ed into the XlaTensor's address to distinguish it from // device-side tensors, which are either CPU or GPU memory pointers. This works
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
} }; StreamExecutor* executor = GetExecutor(0); TF_ASSERT_OK_AND_ASSIGN(auto stream, executor->CreateStream()); TF_ASSERT_OK(stream->RefreshStatus()); status_ok = false; auto updated_status = stream->RefreshStatus(); ASSERT_FALSE(stream->ok()); ASSERT_EQ(updated_status.message(), "Test error"); } TEST_F(StreamExecutorTest, CreateEvent) { static bool event_created = false;
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/c/experimental/stream_executor/stream_executor_internal.h
return s; } void Destroy() { if (stream_handle_ != nullptr) { stream_executor_->destroy_stream(device_, stream_handle_); stream_handle_ = nullptr; } } absl::Status RefreshStatus() override { tensorflow::TF_StatusPtr c_status(TF_NewStatus()); stream_executor_->get_stream_status(device_, stream_handle_, c_status.get());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
return local_device_error_callback(); } return absl::OkStatus(); } Status XlaDevice::RefreshStatus() { std::shared_ptr<se::Stream> stream; { mutex_lock lock(mu_); stream = stream_; } if (!stream) { return absl::OkStatus(); } Status status = stream->RefreshStatus(); if (!status.ok()) { // Ignore errors from HandleDeviceError, since by definition the status is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
// on some platforms. Releasing it in another stream to avoid that. if (!device_allows_sync_on_completion && !device_to_host_stream->RefreshStatus().ok()) { auto status_or_new_stream = client_->mutable_backend()->BorrowStream( stream_->parent()->device_ordinal()); if (status_or_new_stream.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0)