radius_cpu.h 316 Bytes
Newer Older
Alexander Liao's avatar
Alexander Liao committed
1
2
3
4
5
6
7
8
#pragma once

#include <torch/extension.h>
#include "utils/neighbors.h"
#include "utils/neighbors.cpp"
#include <iostream>
#include "compat.h"

9
10
11
torch::Tensor radius_cpu(torch::Tensor query, torch::Tensor support,
                         torch::Tensor ptr_x, torch::Tensor ptr_y, 
			 			 float radius, int max_num);