Commit 8e12a57b authored by rusty1s's avatar rusty1s
Browse files

fix ubuntu build

parent 54f2d481
#include "ego_sample_cpu.h"
#include <ATen/Parallel.h>
#include <process.h>
#include "utils.h"
#ifdef _WIN32
#include <process.h>
#endif
inline torch::Tensor vec2tensor(std::vector<int64_t> vec) {
return torch::from_blob(vec.data(), {(int64_t)vec.size()}, at::kLong).clone();
}
......
#include "hgt_sample_cpu.h"
#include <process.h>
#include "utils.h"
#ifdef _WIN32
#include <process.h>
#endif
#define MAX_NEIGHBORS 50
using namespace std;
......
#include "neighbor_sample_cpu.h"
#include <process.h>
#include "utils.h"
#ifdef _WIN32
#include <process.h>
#endif
using namespace std;
namespace {
......
#include "sample_cpu.h"
#include <process.h>
#include "utils.h"
#ifdef _WIN32
#include <process.h>
#endif
// Returns `rowptr`, `col`, `n_id`, `e_id`
std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor>
sample_adj_cpu(torch::Tensor rowptr, torch::Tensor col, torch::Tensor idx,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment