dmlc--dgl
Hongzhi (Steve), Chen
bcd3768426
* replace * blabla * balbla * blabla Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
26 行
596 B
C++
26 行
596 B
C++
/**
|
|
* Copyright (c) 2020 by Contributors
|
|
* @file featgraph/include/featgraph.h
|
|
* @brief FeatGraph kernel headers.
|
|
*/
|
|
#ifndef FEATGRAPH_H_
|
|
#define FEATGRAPH_H_
|
|
|
|
#include <dlpack/dlpack.h>
|
|
|
|
namespace dgl {
|
|
namespace featgraph {
|
|
|
|
/* @brief Load Featgraph module from given path. */
|
|
void LoadFeatGraphModule(const std::string& path);
|
|
|
|
/* @brief Call Featgraph's SDDMM kernel. */
|
|
void SDDMMTreeReduction(
|
|
DLManagedTensor* row, DLManagedTensor* col, DLManagedTensor* lhs,
|
|
DLManagedTensor* rhs, DLManagedTensor* out);
|
|
|
|
} // namespace featgraph
|
|
} // namespace dgl
|
|
|
|
#endif // FEATGRAPH_H_
|