077b553fda
[StreamExecutor] Allow HostExecutor users to control the stack sizes of threads used for HostStream via. Also include non_portable_tags in the keys used when creating an Executor. There seems to be no good reason that it is omitted. Will fix https://github.com/google/jax/issues/432 when included in a jaxlib release. PiperOrigin-RevId: 309472318 Change-Id: Ia2535616047390d6bf6f2da82a666a321dcc9f5d
64 lines
2.2 KiB
C++
64 lines
2.2 KiB
C++
/* Copyright 2015 The TensorFlow Authors. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
==============================================================================*/
|
|
|
|
// IWYU pragma: private, include "third_party/tensorflow/stream_executor/stream_executor.h"
|
|
|
|
#ifndef TENSORFLOW_STREAM_EXECUTOR_LIB_STATUS_H_
|
|
#define TENSORFLOW_STREAM_EXECUTOR_LIB_STATUS_H_
|
|
|
|
#include "absl/strings/string_view.h"
|
|
#include "tensorflow/core/lib/core/status.h"
|
|
#include "tensorflow/stream_executor/lib/error.h" // IWYU pragma: export
|
|
#include "tensorflow/stream_executor/platform/logging.h"
|
|
|
|
namespace stream_executor {
|
|
namespace port {
|
|
|
|
using Status = tensorflow::Status;
|
|
|
|
#define SE_CHECK_OK(val) TF_CHECK_OK(val)
|
|
#define SE_ASSERT_OK(val) \
|
|
ASSERT_EQ(::stream_executor::port::Status::OK(), (val))
|
|
|
|
// Define some canonical error helpers.
|
|
inline Status UnimplementedError(absl::string_view message) {
|
|
return Status(error::UNIMPLEMENTED, message);
|
|
}
|
|
inline Status InvalidArgumentError(absl::string_view message) {
|
|
return Status(error::INVALID_ARGUMENT, message);
|
|
}
|
|
inline Status InternalError(absl::string_view message) {
|
|
return Status(error::INTERNAL, message);
|
|
}
|
|
inline Status FailedPreconditionError(absl::string_view message) {
|
|
return Status(error::FAILED_PRECONDITION, message);
|
|
}
|
|
|
|
} // namespace port
|
|
} // namespace stream_executor
|
|
|
|
namespace perftools {
|
|
namespace gputools {
|
|
|
|
// Temporarily pull stream_executor into perftools::gputools while we migrate
|
|
// code to the new namespace. TODO(b/77980417): Remove this once we've
|
|
// completed the migration.
|
|
using namespace stream_executor; // NOLINT[build/namespaces]
|
|
|
|
} // namespace gputools
|
|
} // namespace perftools
|
|
|
|
#endif // TENSORFLOW_STREAM_EXECUTOR_LIB_STATUS_H_
|