Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Enhance check_nan_inf implementation for CPU. #48591

Merged
merged 22 commits into from
Dec 12, 2022
Merged

Enhance check_nan_inf implementation for CPU. #48591

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
eca7bfa
Enable to print device info.
Xreki Nov 30, 2022
33c7a18
Enhance the nan and inf checking for cpu.
Xreki Nov 30, 2022
126ffc9
Implement a common print function.
Xreki Dec 1, 2022
bea7cce
Unify the check of complex numbers.
Xreki Dec 1, 2022
124a424
Rewrite the omp method.
Xreki Dec 1, 2022
aeabf0f
Merge branch 'develop' into amp/opt_cpu_check_naninf
Xreki Dec 1, 2022
2177c1c
Polish codes.
Xreki Dec 1, 2022
07f1e53
Merge branch 'develop' into amp/opt_cpu_check_naninf
Xreki Dec 1, 2022
f3a7187
Change the log and fix unittest.
Xreki Dec 1, 2022
26bbdff
Fix log for complex.
Xreki Dec 1, 2022
713298c
Count and print the number of nan and inf.
Xreki Dec 2, 2022
95ff902
Merge branch 'develop' into amp/opt_cpu_check_naninf
Xreki Dec 2, 2022
1c66e70
Fix the printing format.
Xreki Dec 2, 2022
719cf72
Merge branch 'develop' into amp/opt_cpu_check_naninf
Xreki Dec 2, 2022
e4855f2
Polish codes.
Xreki Dec 2, 2022
3f96dae
Merge branch 'develop' into amp/opt_cpu_check_naninf
Xreki Dec 8, 2022
384aa87
Change the print content.
Xreki Dec 9, 2022
1ddb845
Add unittest.
Xreki Dec 9, 2022
daaa36a
Merge branch 'develop' into amp/opt_cpu_check_naninf
Xreki Dec 9, 2022
40db842
Add unittest
Xreki Dec 11, 2022
fbb3cd4
Remove float16 unittest for cpu.
Xreki Dec 11, 2022
3623040
Merge branch 'develop' into amp/opt_cpu_check_naninf
Xreki Dec 11, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
46 changes: 22 additions & 24 deletions paddle/fluid/eager/tests/task_tests/nan_inf_utils_test.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,32 +30,30 @@ PD_DECLARE_KERNEL(strings_empty, CPU, ALL_LAYOUT);

namespace egr {

#define CHECK_NAN_INF(tensors) \
{ \
bool caught_exception = false; \
try { \
CheckTensorHasNanOrInf("nan_inf_test", tensors); \
} catch (paddle::platform::EnforceNotMet & error) { \
caught_exception = true; \
std::string ex_msg = error.what(); \
EXPECT_TRUE(ex_msg.find("There are `nan` or `inf` in tensor") != \
std::string::npos); \
} \
EXPECT_TRUE(caught_exception); \
#define CHECK_NAN_INF(tensors) \
{ \
bool caught_exception = false; \
try { \
CheckTensorHasNanOrInf("nan_inf_test", tensors); \
} catch (paddle::platform::EnforceNotMet & error) { \
caught_exception = true; \
std::string ex_msg = error.what(); \
EXPECT_TRUE(ex_msg.find("There are NAN or INF") != std::string::npos); \
} \
EXPECT_TRUE(caught_exception); \
}

#define CHECK_NO_NAN_INF(tensors) \
{ \
bool caught_exception = false; \
try { \
CheckTensorHasNanOrInf("nan_inf_test", tensors); \
} catch (paddle::platform::EnforceNotMet & error) { \
caught_exception = true; \
std::string ex_msg = error.what(); \
EXPECT_TRUE(ex_msg.find("There are `nan` or `inf` in tensor") != \
std::string::npos); \
} \
EXPECT_FALSE(caught_exception); \
#define CHECK_NO_NAN_INF(tensors) \
{ \
bool caught_exception = false; \
try { \
CheckTensorHasNanOrInf("nan_inf_test", tensors); \
} catch (paddle::platform::EnforceNotMet & error) { \
caught_exception = true; \
std::string ex_msg = error.what(); \
EXPECT_TRUE(ex_msg.find("There are NAN or INF") != std::string::npos); \
} \
EXPECT_FALSE(caught_exception); \
}

TEST(NanInfUtils, Functions) {
Expand Down
Loading