9 #ifndef NEIGHBORS_BUFFER_KD_TREE_UTIL_H_
10 #define NEIGHBORS_BUFFER_KD_TREE_UTIL_H_
18 #include <sys/resource.h>
#define UINT_TYPE
Definition: global.h:12
void partition_array_via_pivot(void *array, INT_TYPE count, INT_TYPE axis, INT_TYPE size_per_elt, FLOAT_TYPE pivot_value)
Definition: util.c:393
void read_patterns(const char *ifile, FLOAT_TYPE **patterns, FLOAT_TYPE **labels, INT_TYPE *num, INT_TYPE *dim)
double get_raw_train_mem_device_bytes(TREE_RECORD *tree_record, TREE_PARAMETERS *params)
Definition: util.c:276
FLOAT_TYPE squared(FLOAT_TYPE a)
Definition: util.c:155
void cb_free(circular_buffer *cb)
Definition: util.c:224
double get_test_tmp_mem_device_bytes(TREE_RECORD *tree_record, TREE_PARAMETERS *params)
Definition: util.c:321
void cb_read_batch(circular_buffer *cb, INT_TYPE *items_array, INT_TYPE num_elts_to_remove)
Definition: util.c:216
void dist_insert_batch(FLOAT_TYPE *train_patt, INT_TYPE train_idx, FLOAT_TYPE *test_patterns, INT_TYPE ntest_patterns, FLOAT_TYPE *d_min, INT_TYPE *idx_min, INT_TYPE dim, UINT_TYPE K)
Definition: util.c:83
void copy_element(void *dest, const void *src, INT_TYPE size_elt)
Definition: util.c:470
double get_test_max_buffer_device_bytes(TREE_RECORD *tree_record, TREE_PARAMETERS *params)
Definition: util.c:377
void set_default_parameters(BRUTE_PARAMETERS *params)
Definition: util.c:17
void cb_add_elt(circular_buffer *cb, INT_TYPE *item)
Definition: util.c:233
double get_train_max_buffer_device_bytes(TREE_RECORD *tree_record, TREE_PARAMETERS *params)
Definition: util.c:362
INT_TYPE cb_get_number_items(circular_buffer *cb)
Definition: util.c:209
void cb_read_batch_fast(circular_buffer *cb, INT_TYPE *items_array, INT_TYPE num_elts_to_remove)
void swap_elements(void *p1, void *p2, int size_elt)
Definition: util.c:451
INT_TYPE cb_is_empty(circular_buffer *cb)
Definition: util.c:230
void cb_init(circular_buffer *cb, INT_TYPE size)
Definition: util.c:203
void cb_write(circular_buffer *cb, INT_TYPE *item)
Definition: util.c:236
double get_train_mem_with_chunks_device_bytes(TREE_RECORD *tree_record, TREE_PARAMETERS *params)
Definition: util.c:298
#define INT_TYPE
Definition: global.h:11
void kd_tree_insert(FLOAT_TYPE pattern_dist, INT_TYPE pattern_idx, FLOAT_TYPE *nearest_dist, INT_TYPE *nearest_idx, UINT_TYPE K)
Definition: util.c:170
FLOAT_TYPE kd_tree_dist(FLOAT_TYPE *a, FLOAT_TYPE *b, INT_TYPE dim)
Definition: util.c:137
char * readline(FILE *input)
double get_total_mem_device_bytes(TREE_RECORD *tree_record, TREE_PARAMETERS *params)
Definition: util.c:345
INT_TYPE cb_is_full(circular_buffer *cb)
Definition: util.c:227
circular_buffer * cb_double_size(circular_buffer *cb)
Definition: util.c:246
void cb_read(circular_buffer *cb, INT_TYPE *item)
Definition: util.c:242
#define FLOAT_TYPE
Definition: float.h:17
void check_parameters(BRUTE_PARAMETERS *params)
Definition: util.c:30