chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
/**
|
||||
* Copyright (c) 2022 by Contributors
|
||||
* @file sparse/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 <dgl/aten/coo.h>
|
||||
#include <dgl/aten/csr.h>
|
||||
#include <dgl/kernel.h>
|
||||
#include <dgl/runtime/dlpack_convert.h>
|
||||
#include <dmlc/logging.h>
|
||||
|
||||
#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_
|
||||
Reference in New Issue
Block a user