@@ -80,8 +80,6 @@ class StatusTest : public testing::TestWithParam<CppStacktracesMode> {
80
80
namespace cpp_test {
81
81
82
82
// Prefix of the C++ stacktrace PyTorch adds to the error message.
83
- constexpr inline char kTorchCppStacktracePrefixDeprecated [] =
84
- " Exception raised from OkOrThrow at torch_xla/csrc/status.cpp:" ;
85
83
constexpr inline char kTorchCppStacktracePrefix [] =
86
84
" Exception raised from ThrowStatusError at torch_xla/csrc/status.cpp:" ;
87
85
@@ -102,50 +100,6 @@ inline std::string GetStatusPropagationTrace(const absl::Status& status) {
102
100
: " " ;
103
101
}
104
102
105
- TEST_P (StatusTest, OkOrThrowWithOkStatus) {
106
- absl::Status ok_status = absl::OkStatus ();
107
- EXPECT_NO_THROW (OkOrThrow (ok_status));
108
- }
109
-
110
- TEST_P (StatusTest, OkOrThrowWithErrorStatus) {
111
- try {
112
- absl::Status error_status = absl::InvalidArgumentError (kMessage );
113
- OkOrThrow (error_status);
114
- } catch (const c10::Error& error) {
115
- if (IsShowCppStacktracesMode ()) {
116
- EXPECT_THAT (std::string_view (error.what ()),
117
- ::testing::StartsWith (absl::StrCat(
118
- kMessage , " \n\n " , kTorchCppStacktracePrefixDeprecated )));
119
- } else {
120
- EXPECT_EQ (std::string_view (error.what_without_backtrace ()),
121
- std::string_view (kMessage ));
122
- }
123
- }
124
- }
125
-
126
- TEST_P (StatusTest, GetValueOrThrowWithOkStatusOr) {
127
- int value = 42 ;
128
- absl::StatusOr<int > status_or = value;
129
- int result = GetValueOrThrow (std::move (status_or));
130
- EXPECT_EQ (result, value);
131
- }
132
-
133
- TEST_P (StatusTest, GetValueOrThrowWithErrorStatusOr) {
134
- try {
135
- absl::StatusOr<int > error_status = absl::InvalidArgumentError (kMessage );
136
- int value = GetValueOrThrow (error_status);
137
- } catch (const c10::Error& error) {
138
- if (IsShowCppStacktracesMode ()) {
139
- EXPECT_THAT (std::string_view (error.what ()),
140
- ::testing::StartsWith (absl::StrCat(
141
- kMessage , " \n\n " , kTorchCppStacktracePrefixDeprecated )));
142
- } else {
143
- EXPECT_EQ (std::string_view (error.what_without_backtrace ()),
144
- std::string_view (kMessage ));
145
- }
146
- }
147
- }
148
-
149
103
TEST_P (StatusTest, MaybeWithLocationPropagatesErrorStatus) {
150
104
absl::Status error_status = absl::InvalidArgumentError (kMessage );
151
105
absl::Status result =
@@ -345,61 +299,6 @@ TEST_P(StatusTest, MacroErrorWithLocation) {
345
299
}
346
300
}
347
301
348
- TEST_P (StatusTest, OkOrThrowWithErrorPropagationWithNewMessage) {
349
- int32_t errline0 = __LINE__ + 2 ;
350
- auto innerfn = [&]() -> absl::Status {
351
- return XLA_ERROR_WITH_LOCATION (absl::InvalidArgumentError (kMessage ));
352
- };
353
-
354
- int32_t errline1 = __LINE__ + 2 ;
355
- auto midfn = [&]() -> absl::Status {
356
- XLA_RETURN_IF_ERROR (innerfn (), kNewMessage );
357
- return absl::OkStatus ();
358
- };
359
-
360
- int32_t errline2 = __LINE__ + 2 ;
361
- auto outerfn = [&]() -> absl::Status {
362
- XLA_RETURN_IF_ERROR (midfn ());
363
- return absl::OkStatus ();
364
- };
365
-
366
- try {
367
- OkOrThrow (outerfn ());
368
- } catch (const c10::Error& error) {
369
- if (IsShowCppStacktracesMode ()) {
370
- // Expected Error Message Prefix
371
- // =============================
372
- //
373
- // New test error kMessage
374
- //
375
- // Status Propagation Stacktrace:
376
- // From: ./test/cpp/test_status_common.h:329 (error: Test error
377
- // kMessage) From: ./test/cpp/test_status_common.h:335 (error: New
378
- // test error kMessage) From: ./test/cpp/test_status_common.h:342
379
- //
380
- // C++ Stacktrace:
381
- //
382
- std::ostringstream oss;
383
- oss << kNewMessage ;
384
- oss << " \n\n " ;
385
- oss << " Status Propagation Trace:" ;
386
- oss << kEntryPrefix << " From: operator() at " << __FILE__ << " :"
387
- << errline0 << " (error: " << kMessage << " )" ;
388
- oss << kEntryPrefix << " From: operator() at " << __FILE__ << " :"
389
- << errline1 << " (error: " << kNewMessage << " )" ;
390
- oss << kEntryPrefix << " From: operator() at " << __FILE__ << " :"
391
- << errline2;
392
- oss << " \n\n " ;
393
- oss << kTorchCppStacktracePrefixDeprecated ;
394
- EXPECT_THAT (std::string_view (error.what ()),
395
- ::testing::StartsWith (oss.str()));
396
- } else {
397
- EXPECT_EQ (std::string_view (error.what_without_backtrace ()),
398
- std::string_view (kNewMessage ));
399
- }
400
- }
401
- }
402
-
403
302
TEST_P (StatusTest, MacroThrowIfErrorWithErrorPropagationWithNewMessage) {
404
303
int32_t errline0 = __LINE__ + 2 ;
405
304
auto innerfn = [&]() -> absl::Status {
0 commit comments