/** * Copyright (c) 2022 by Contributors * @file dgl_headers.h * @brief DGL headers used in the sparse library. This is a workaround to * avoid the macro naming conflict between dmlc/logging.h and torch logger. This * file includes all the DGL headers used in the sparse library and * undefines logging macros defined in dmlc/logging.h. There are two rules to * use this file. (1) All DGL headers used in the sparse library should be and * only be registered in this file. (2) When including Pytorch headers, this * file should be included in advance. */ #ifndef SPARSE_DGL_HEADERS_H_ #define SPARSE_DGL_HEADERS_H_ #include #include #include #include #undef CHECK #undef CHECK_OP #undef CHECK_EQ #undef CHECK_NE #undef CHECK_LE #undef CHECK_LT #undef CHECK_GE #undef CHECK_GT #undef CHECK_NOTNULL #undef DCHECK #undef DCHECK_EQ #undef DCHECK_NE #undef DCHECK_LE #undef DCHECK_LT #undef DCHECK_GE #undef DCHECK_GT #undef DCHECK_NOTNULL #undef VLOG #undef LOG #undef DLOG #undef LOG_IF #endif // SPARSE_DGL_HEADERS_H_