ML::Sparse Namespace Reference

Functions
void	brute_force_knn (raft::handle_t &handle, const int *idx_indptr, const int *idx_indices, const float *idx_data, size_t idx_nnz, int n_idx_rows, int n_idx_cols, const int *query_indptr, const int *query_indices, const float *query_data, size_t query_nnz, int n_query_rows, int n_query_cols, int *output_indices, float *output_dists, int k, size_t batch_size_index=DEFAULT_BATCH_SIZE, size_t batch_size_query=DEFAULT_BATCH_SIZE, cuvs::distance::DistanceType metric=cuvs::distance::DistanceType::L2Expanded, float metricArg=0)

Variables
constexpr int	DEFAULT_BATCH_SIZE = 1 << 16

Function Documentation

brute_force_knn()

void ML::Sparse::brute_force_knn	(	raft::handle_t &	handle,
		const int *	idx_indptr,
		const int *	idx_indices,
		const float *	idx_data,
		size_t	idx_nnz,
		int	n_idx_rows,
		int	n_idx_cols,
		const int *	query_indptr,
		const int *	query_indices,
		const float *	query_data,
		size_t	query_nnz,
		int	n_query_rows,
		int	n_query_cols,
		int *	output_indices,
		float *	output_dists,
		int	k,
		size_t	batch_size_index = DEFAULT_BATCH_SIZE,
		size_t	batch_size_query = DEFAULT_BATCH_SIZE,
		cuvs::distance::DistanceType	metric = cuvs::distance::DistanceType::L2Expanded,
		float	metricArg = 0
	)

Variable Documentation

DEFAULT_BATCH_SIZE

constexpr int ML::Sparse::DEFAULT_BATCH_SIZE = 1 << 16

constexpr