Comet/gpu_optimize_storm_drift.h at master · gpufit/Comet · GitHub

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#ifndef GPU_STORM_DRIFT_OPTIMIZE_INCLUDED
#define GPU_STORM_DRIFT_OPTIMIZE_INCLUDED

#include <vector>
#include <stdlib.h>

#ifdef __cplusplus
extern "C" {
#endif

    int gpu_opt_storm_drift_initialize_2d_portable(int argc, void *argv[]);
    int gpu_opt_storm_drift_compute_2d_portable(int argc, void *argv[]);
    int gpu_opt_storm_drift_free_2d_portable(int argc, void *argv[]);
    int gpu_opt_storm_drift_initialize_3d_portable(int argc, void *argv[]);
    int gpu_opt_storm_drift_compute_3d_portable(int argc, void *argv[]);
    int gpu_opt_storm_drift_free_3d_portable(int argc, void *argv[]);

#ifdef __cplusplus
}
#endif

int gpu_opt_storm_drift_compute_2d(
    int n_coordinates,
    int n_timepoints,
    size_t n_coordinate_pairs,
    float gaussian_scale,
    float * drift_trajectory,
    float * output_cost_function,
    int flag_calculate_derivatives,
    float * output_derivatives);

int gpu_opt_storm_drift_initialize_2d(
    int n_coordinates,
    int n_time_points,
    float * coordinates_x,
    float * coordinates_y,
    int * coordinates_time,
    size_t n_coordinate_pairs,
    int * pair_indices_i,
    int * pair_indices_j);

int gpu_opt_storm_drift_free_2d();

int gpu_opt_storm_drift_compute_3d(
    int n_coordinates,
    int n_timepoints,
    size_t n_coordinate_pairs,
    float gaussian_scale,
    float * drift_trajectory,
    float * output_cost_function,
    int flag_calculate_derivatives,
    float * output_derivatives);

int gpu_opt_storm_drift_initialize_3d(
    int n_coordinates,
    int n_time_points,
    float * coordinates_x,
    float * coordinates_y,
    float * coordinates_z,
    int * coordinates_time,
    size_t n_coordinate_pairs,
    int * pair_indices_i,
    int * pair_indices_j);

int gpu_opt_storm_drift_free_3d();


#endif // !GPU_STORM_DRIFT_OPTIMIZE_INCLUDED