featgraph.h

/**
 *  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_