mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
5c5eece6d8
Although FreeBSD is not officially supported, this PR fixes some errors on FreeBSD. Pull Request resolved: https://github.com/pytorch/pytorch/pull/105897 Approved by: https://github.com/kit1980
104 lines
3.0 KiB
C++
104 lines
3.0 KiB
C++
#include <c10/util/Exception.h>
|
|
#include <gtest/gtest.h>
|
|
#include <stdexcept>
|
|
|
|
using c10::Error;
|
|
|
|
namespace {
|
|
|
|
template <class Functor>
|
|
inline void expectThrowsEq(Functor&& functor, const char* expectedMessage) {
|
|
try {
|
|
std::forward<Functor>(functor)();
|
|
} catch (const Error& e) {
|
|
EXPECT_STREQ(e.what_without_backtrace(), expectedMessage);
|
|
return;
|
|
}
|
|
ADD_FAILURE() << "Expected to throw exception with message \""
|
|
<< expectedMessage << "\" but didn't throw";
|
|
}
|
|
} // namespace
|
|
|
|
TEST(ExceptionTest, TORCH_INTERNAL_ASSERT_DEBUG_ONLY) {
|
|
#ifdef NDEBUG
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-goto,hicpp-avoid-goto)
|
|
ASSERT_NO_THROW(TORCH_INTERNAL_ASSERT_DEBUG_ONLY(false));
|
|
// Does nothing - `throw ...` should not be evaluated
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-goto,hicpp-avoid-goto)
|
|
ASSERT_NO_THROW(TORCH_INTERNAL_ASSERT_DEBUG_ONLY(
|
|
(throw std::runtime_error("I'm throwing..."), true)));
|
|
#else
|
|
ASSERT_THROW(TORCH_INTERNAL_ASSERT_DEBUG_ONLY(false), c10::Error);
|
|
ASSERT_NO_THROW(TORCH_INTERNAL_ASSERT_DEBUG_ONLY(true));
|
|
#endif
|
|
}
|
|
|
|
// On these platforms there's no assert
|
|
#if !defined(__ANDROID__) && !defined(__APPLE__) && !defined(__FreeBSD__) && \
|
|
!(defined(USE_ROCM) && ROCM_VERSION < 40100)
|
|
TEST(ExceptionTest, CUDA_KERNEL_ASSERT) {
|
|
// This function always throws even in NDEBUG mode
|
|
ASSERT_DEATH_IF_SUPPORTED({ CUDA_KERNEL_ASSERT(false); }, "Assert");
|
|
}
|
|
#endif
|
|
|
|
TEST(WarningTest, JustPrintWarning) {
|
|
TORCH_WARN("I'm a warning");
|
|
}
|
|
|
|
TEST(ExceptionTest, ErrorFormatting) {
|
|
expectThrowsEq(
|
|
[]() { TORCH_CHECK(false, "This is invalid"); }, "This is invalid");
|
|
|
|
expectThrowsEq(
|
|
[]() {
|
|
try {
|
|
TORCH_CHECK(false, "This is invalid");
|
|
} catch (Error& e) {
|
|
TORCH_RETHROW(e, "While checking X");
|
|
}
|
|
},
|
|
"This is invalid (While checking X)");
|
|
|
|
expectThrowsEq(
|
|
[]() {
|
|
try {
|
|
try {
|
|
TORCH_CHECK(false, "This is invalid");
|
|
} catch (Error& e) {
|
|
TORCH_RETHROW(e, "While checking X");
|
|
}
|
|
} catch (Error& e) {
|
|
TORCH_RETHROW(e, "While checking Y");
|
|
}
|
|
},
|
|
R"msg(This is invalid
|
|
While checking X
|
|
While checking Y)msg");
|
|
}
|
|
|
|
static int assertionArgumentCounter = 0;
|
|
static int getAssertionArgument() {
|
|
return ++assertionArgumentCounter;
|
|
}
|
|
|
|
static void failCheck() {
|
|
TORCH_CHECK(false, "message ", getAssertionArgument());
|
|
}
|
|
|
|
static void failInternalAssert() {
|
|
TORCH_INTERNAL_ASSERT(false, "message ", getAssertionArgument());
|
|
}
|
|
|
|
TEST(ExceptionTest, DontCallArgumentFunctionsTwiceOnFailure) {
|
|
assertionArgumentCounter = 0;
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-goto,hicpp-avoid-goto)
|
|
EXPECT_ANY_THROW(failCheck());
|
|
EXPECT_EQ(assertionArgumentCounter, 1) << "TORCH_CHECK called argument twice";
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-goto,hicpp-avoid-goto)
|
|
EXPECT_ANY_THROW(failInternalAssert());
|
|
EXPECT_EQ(assertionArgumentCounter, 2)
|
|
<< "TORCH_INTERNAL_ASSERT called argument twice";
|
|
}
|