1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
#include <gtest/gtest.h>
#include <torch/headeronly/util/Exception.h>
#include <torch/headeronly/util/shim_utils.h>
namespace torch {
namespace aot_inductor {
TEST(TestExceptions, TestStdTorchCheck) {
EXPECT_NO_THROW(STD_TORCH_CHECK(true, "dummy true message"));
EXPECT_NO_THROW(STD_TORCH_CHECK(true, "dummy ", "true ", "message"));
EXPECT_THROW(
STD_TORCH_CHECK(false, "dummy false message"), std::runtime_error);
EXPECT_THROW(
STD_TORCH_CHECK(false, "dummy ", "false ", "message"),
std::runtime_error);
}
TEST(TestExceptions, TestTorchErrorCodeCheck) {
EXPECT_NO_THROW(TORCH_ERROR_CODE_CHECK(0));
EXPECT_THROW(TORCH_ERROR_CODE_CHECK(1), std::runtime_error);
}
} // namespace aot_inductor
} // namespace torch
|
