diff options
Diffstat (limited to 'tensorflow/stream_executor/lib/status.h')
-rw-r--r-- | tensorflow/stream_executor/lib/status.h | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/lib/status.h b/tensorflow/stream_executor/lib/status.h new file mode 100644 index 0000000000..b3ad13b0ae --- /dev/null +++ b/tensorflow/stream_executor/lib/status.h @@ -0,0 +1,23 @@ +#ifndef TENSORFLOW_STREAM_EXECUTOR_LIB_STATUS_H_ +#define TENSORFLOW_STREAM_EXECUTOR_LIB_STATUS_H_ + +#include "tensorflow/core/public/status.h" +#include "tensorflow/stream_executor/lib/error.h" +#include "tensorflow/stream_executor/platform/logging.h" + +namespace perftools { +namespace gputools { +namespace port { + +using tensorflow::Status; + +#define SE_CHECK_OK(val) \ + CHECK_EQ(::perftools::gputools::port::Status::OK(), (val)) +#define SE_ASSERT_OK(val) \ + ASSERT_EQ(::perftools::gputools::port::Status::OK(), (val)) + +} // namespace port +} // namespace gputools +} // namespace perftools + +#endif // TENSORFLOW_STREAM_EXECUTOR_LIB_STATUS_H_ |