FilesExpand file tree

 main

/

maxsim.h

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

41 lines (33 loc) · 1.14 KB

 main

/

maxsim.h

Copy path

Top

File metadata and controls

• Code
• Blame

41 lines (33 loc) · 1.14 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

/**

* @brief MaxSim late-interaction declarations for NumKong Python bindings.

* @file python/maxsim.h

* @author Ash Vardanian

* @date March 9, 2026

*

* Declares the MaxSimPackedMatrix type and API functions for MaxSim

* (ColBERT late-interaction scoring) used by the Python module.

*/

#ifndef NK_PYTHON_MAXSIM_H

#define NK_PYTHON_MAXSIM_H

#include "numkong.h"

#ifdef __cplusplus

extern "C" {

#endif

/** @brief Pre-packed matrix for MaxSim late-interaction scoring. */

typedef struct MaxSimPackedMatrix {

PyObject_HEAD nk_dtype_t dtype;

nk_size_t vector_count;

nk_size_t depth;

char start[];

} MaxSimPackedMatrix;

extern PyTypeObject MaxSimPackedMatrixType;

PyObject *api_maxsim_pack(PyObject *self, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames);

PyObject *api_maxsim_packed(PyObject *self, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames);

PyObject *api_maxsim(PyObject *self, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames);

extern char const doc_maxsim_pack[];

extern char const doc_maxsim_packed[];

extern char const doc_maxsim[];

#ifdef __cplusplus

}

#endif

#endif // NK_PYTHON_MAXSIM_H