-
Notifications
You must be signed in to change notification settings - Fork 6
Expand file tree
/
Copy pathpyfgc.h
More file actions
42 lines (36 loc) · 1.37 KB
/
pyfgc.h
File metadata and controls
42 lines (36 loc) · 1.37 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#pragma once
#include "pybind11/pybind11.h"
#include "pybind11/numpy.h"
#include "aesctr/wy.h"
#include "minicore/minicore.h"
#include "./pyhelpers.h"
void init_smw(py::module &);
void init_pycsparse(py::module &);
void init_cmp(py::module &);
void init_coreset(py::module &);
void init_centroid(py::module &);
void init_hashers(py::module &);
void init_omp_helpers(py::module &m);
void init_clustering(py::module &m);
void init_d2s(py::module &m);
void init_clustering_csr(py::module &m);
void init_clustering_dense(py::module &m);
void init_clustering_soft_csr(py::module &m);
void init_clustering_soft(py::module &m);
void init_pydense(py::module &m);
void init_arrcmp(py::module &m);
void init_clustering_softdense(py::module &m);
void init_counthist(py::module &m);
//void init_hvg(py::module &m);
// Direct CSR mode (no copying) saves memory, but takes a long time to compile
// define BUILD_CSR_CLUSTERING to 1 or modify this file to enable.
#ifndef BUILD_CSR_CLUSTERING
#define BUILD_CSR_CLUSTERING 1
#endif
using CSType = coresets::CoresetSampler<float, uint32_t>;
using FNA = py::array_t<float, py::array::c_style | py::array::forcecast>;
using DNA = py::array_t<double, py::array::c_style | py::array::forcecast>;
using INA = py::array_t<uint32_t, py::array::c_style | py::array::forcecast>;
using SMF = blz::SM<float>;
using SMD = blz::SM<double>;
namespace dist = minicore::distance;