File tree Expand file tree Collapse file tree 1 file changed +17
-0
lines changed Expand file tree Collapse file tree 1 file changed +17
-0
lines changed Original file line number Diff line number Diff line change @@ -352,10 +352,16 @@ class LogFatal {
352352#ifdef _MSC_VER
353353#pragma disagnostic push
354354#pragma warning(disable : 4722)
355+ #elif defined(__CUDACC__)
356+ #pragma nv_exec_check_disable
357+ #pragma diag_suppress 1305
355358#endif
356359 [[noreturn]] ~LogFatal () TVM_THROW_EXCEPTION { GetEntry ().Finalize (); }
357360#ifdef _MSC_VER
358361#pragma disagnostic pop
362+ #elif defined(__CUDACC__)
363+ #pragma nv_exec_check_enable
364+ #pragma diag_default 1305
359365#endif
360366 std::ostringstream& stream () { return GetEntry ().stream_ ; }
361367
@@ -560,15 +566,26 @@ std::unique_ptr<std::string> LogCheckFormat(const X& x, const Y& y) {
560566 return LogCheck##name<int , int >(x, y); \
561567 }
562568
569+ #if defined(__GNUC__) || defined(__clang__) // GCC and Clang
563570#pragma GCC diagnostic push
564571#pragma GCC diagnostic ignored "-Wsign-compare"
572+ #elif defined(_MSC_VER) // MSVC
573+ #pragma warning(push)
574+ #pragma warning(disable : 4389) // '==' : signed/unsigned mismatch
575+ #endif
576+
565577TVM_CHECK_FUNC (_LT, <)
566578TVM_CHECK_FUNC (_GT, >)
567579TVM_CHECK_FUNC (_LE, <=)
568580TVM_CHECK_FUNC (_GE, >=)
569581TVM_CHECK_FUNC (_EQ, ==)
570582TVM_CHECK_FUNC (_NE, !=)
583+
584+ #if defined(__GNUC__) || defined(__clang__) // GCC and Clang
571585#pragma GCC diagnostic pop
586+ #elif defined(_MSC_VER) // MSVC
587+ #pragma warning(pop)
588+ #endif
572589
573590} // namespace detail
574591
You can’t perform that action at this time.
0 commit comments