mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
abb2204f6a
Makes TORCH_CHECK_* run unconditionally, leaving only TORCH_DCHECK_* special-cased to be optimized out in release builds. Fixes a bug in #82032, relating to this comment https://github.com/pytorch/pytorch/pull/82032#issuecomment-1203726409 Pull Request resolved: https://github.com/pytorch/pytorch/pull/83216 Approved by: https://github.com/ezyang, https://github.com/datumbox
110 lines
3.7 KiB
C++
110 lines
3.7 KiB
C++
#ifndef C10_UTIL_LOGGING_IS_GOOGLE_GLOG_H_
|
|
#define C10_UTIL_LOGGING_IS_GOOGLE_GLOG_H_
|
|
|
|
#include <map>
|
|
#include <set>
|
|
#include <vector>
|
|
|
|
#include <iomanip> // because some of the caffe2 code uses e.g. std::setw
|
|
// Using google glog. For glog 0.3.2 versions, stl_logging.h needs to be before
|
|
// logging.h to actually use stl_logging. Because template magic.
|
|
// In addition, we do not do stl logging in .cu files because nvcc does not like
|
|
// it. Some mobile platforms do not like stl_logging, so we add an
|
|
// overload in that case as well.
|
|
|
|
#ifdef __CUDACC__
|
|
#include <cuda.h>
|
|
#endif
|
|
|
|
#if !defined(__CUDACC__) && !defined(C10_USE_MINIMAL_GLOG)
|
|
#include <glog/stl_logging.h>
|
|
|
|
// Old versions of glog don't declare this using declaration, so help
|
|
// them out. Fortunately, C++ won't complain if you declare the same
|
|
// using declaration multiple times.
|
|
namespace std {
|
|
using ::operator<<;
|
|
}
|
|
|
|
#else // !defined(__CUDACC__) && !defined(C10_USE_MINIMAL_GLOG)
|
|
|
|
// In the cudacc compiler scenario, we will simply ignore the container
|
|
// printout feature. Basically we need to register a fake overload for
|
|
// vector/string - here, we just ignore the entries in the logs.
|
|
|
|
namespace std {
|
|
#define INSTANTIATE_FOR_CONTAINER(container) \
|
|
template <class... Types> \
|
|
ostream& operator<<(ostream& out, const container<Types...>&) { \
|
|
return out; \
|
|
}
|
|
|
|
INSTANTIATE_FOR_CONTAINER(vector)
|
|
INSTANTIATE_FOR_CONTAINER(map)
|
|
INSTANTIATE_FOR_CONTAINER(set)
|
|
#undef INSTANTIATE_FOR_CONTAINER
|
|
} // namespace std
|
|
|
|
#endif
|
|
|
|
#include <glog/logging.h>
|
|
|
|
// Additional macros on top of glog
|
|
#define TORCH_CHECK_EQ(val1, val2) CHECK_EQ(val1, val2)
|
|
#define TORCH_CHECK_NE(val1, val2) CHECK_NE(val1, val2)
|
|
#define TORCH_CHECK_LE(val1, val2) CHECK_LE(val1, val2)
|
|
#define TORCH_CHECK_LT(val1, val2) CHECK_LT(val1, val2)
|
|
#define TORCH_CHECK_GE(val1, val2) CHECK_GE(val1, val2)
|
|
#define TORCH_CHECK_GT(val1, val2) CHECK_GT(val1, val2)
|
|
|
|
#ifndef NDEBUG
|
|
#define TORCH_DCHECK_EQ(val1, val2) DCHECK_EQ(val1, val2)
|
|
#define TORCH_DCHECK_NE(val1, val2) DCHECK_NE(val1, val2)
|
|
#define TORCH_DCHECK_LE(val1, val2) DCHECK_LE(val1, val2)
|
|
#define TORCH_DCHECK_LT(val1, val2) DCHECK_LT(val1, val2)
|
|
#define TORCH_DCHECK_GE(val1, val2) DCHECK_GE(val1, val2)
|
|
#define TORCH_DCHECK_GT(val1, val2) DCHECK_GT(val1, val2)
|
|
#else // !NDEBUG
|
|
// These versions generate no code in optimized mode.
|
|
#define TORCH_DCHECK_EQ(val1, val2) \
|
|
while (false) \
|
|
DCHECK_EQ(val1, val2)
|
|
#define TORCH_DCHECK_NE(val1, val2) \
|
|
while (false) \
|
|
DCHECK_NE(val1, val2)
|
|
#define TORCH_DCHECK_LE(val1, val2) \
|
|
while (false) \
|
|
DCHECK_LE(val1, val2)
|
|
#define TORCH_DCHECK_LT(val1, val2) \
|
|
while (false) \
|
|
DCHECK_LT(val1, val2)
|
|
#define TORCH_DCHECK_GE(val1, val2) \
|
|
while (false) \
|
|
DCHECK_GE(val1, val2)
|
|
#define TORCH_DCHECK_GT(val1, val2) \
|
|
while (false) \
|
|
DCHECK_GT(val1, val2)
|
|
#endif // NDEBUG
|
|
|
|
// Check that a pointer is not null.
|
|
#define TORCH_CHECK_NOTNULL(val) CHECK_NOTNULL(val)
|
|
|
|
#ifndef NDEBUG
|
|
// Debug only version of TORCH_CHECK_NOTNULL
|
|
#define TORCH_DCHECK_NOTNULL(val) DCHECK_NOTNULL(val)
|
|
#else // !NDEBUG
|
|
// Optimized version - generates no code.
|
|
#define TORCH_DCHECK_NOTNULL(val) \
|
|
while (false) \
|
|
DCHECK_NOTNULL(val)
|
|
#endif // NDEBUG
|
|
|
|
// Log with source location information override (to be used in generic
|
|
// warning/error handlers implemented as functions, not macros)
|
|
//
|
|
// Note, we don't respect GOOGLE_STRIP_LOG here for simplicity
|
|
#define LOG_AT_FILE_LINE(n, file, line) \
|
|
::google::LogMessage(file, line, ::google::GLOG_##n).stream()
|
|
|
|
#endif // C10_UTIL_LOGGING_IS_GOOGLE_GLOG_H_
|