KnnSearchOpKernelSYCL.cpp File Reference

(d32b4fc (Mon Sep 14 15:46:25 2026 -0700))

#include <c10/xpu/XPUStream.h>
#include <sycl/sycl.hpp>
#include "open3d/core/SYCLContext.h"
#include "open3d/core/nns/KnnIndex.h"
#include "open3d/core/nns/NeighborSearchCommon.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/misc/TorchOpen3DBridge.h"
#include "torch/script.h"

Macros
#define	INSTANTIATE(T, TIndex)

Functions
template<class T , class TIndex >
void	KnnSearchSYCL (const torch::Tensor &points, const torch::Tensor &queries, const int64_t k, const torch::Tensor &points_row_splits, const torch::Tensor &queries_row_splits, torch::Tensor &neighbors_index, torch::Tensor &neighbors_row_splits, torch::Tensor &neighbors_distance)

Macro Definition Documentation

INSTANTIATE

#define INSTANTIATE	(		T,
			TIndex
	)

Value:

    template void KnnSearchSYCL<T, TIndex>(                            \
            const torch::Tensor& points, const torch::Tensor& queries, \
            const int64_t k, const torch::Tensor& points_row_splits,   \
            const torch::Tensor& queries_row_splits,                   \
            torch::Tensor& neighbors_index,                            \
            torch::Tensor& neighbors_row_splits,                       \
            torch::Tensor& neighbors_distance);

Function Documentation

KnnSearchSYCL()

template<class T , class TIndex >

void KnnSearchSYCL	(	const torch::Tensor &	points,
		const torch::Tensor &	queries,
		const int64_t	k,
		const torch::Tensor &	points_row_splits,
		const torch::Tensor &	queries_row_splits,
		torch::Tensor &	neighbors_index,
		torch::Tensor &	neighbors_row_splits,
		torch::Tensor &	neighbors_distance
	)