dmlc--dgl
Minjie Wang
f1b19a6b67
* add cuda utils; change g.to; add g.device * split array.h into several headers * cuda index select * file * three cuda kernels * add cuda elementwise arith and several others * cuda CSRIsNonZero * fix lint * lint * lint * fix bug in changing ctx to property * address comments * remove unused codes * address comments
18 行
493 B
C
18 行
493 B
C
/*!
|
|
* Copyright (c) 2020 by Contributors
|
|
* \file dgl/array.h
|
|
* \brief Common array operations required by DGL.
|
|
*
|
|
* Note that this is not meant for a full support of array library such as ATen.
|
|
* Only a limited set of operators required by DGL are implemented.
|
|
*/
|
|
#ifndef DGL_ARRAY_H_
|
|
#define DGL_ARRAY_H_
|
|
#include "./aten/types.h"
|
|
#include "./aten/array_ops.h"
|
|
#include "./aten/macro.h"
|
|
#include "./aten/spmat.h"
|
|
#include "./aten/csr.h"
|
|
#include "./aten/coo.h"
|
|
#endif // DGL_ARRAY_H_
|