Commit b4d88911 authored by Andrey Filippov's avatar Andrey Filippov
Browse files

Updated GPU code

parent e156dbea
Loading
Loading
Loading
Loading
+381 −24

File changed.

Preview size limit exceeded, changes collapsed.

+19 −1
Original line number Original line Diff line number Diff line
@@ -87,6 +87,24 @@ extern "C" __global__ void correlate2D(
		int               corr_radius,        // radius of the output correlation (7 for 15x15)
		int               corr_radius,        // radius of the output correlation (7 for 15x15)
		float           * gpu_corrs);          // correlation output data
		float           * gpu_corrs);          // correlation output data


extern "C" __global__ void correlate2D_inter( // only results in TD
		int               num_cams,
		int               sel_sensors,
		float          ** gpu_clt,            // [num_cams] ->[TILES-Y][TILES-X][colors][DTT_SIZE*DTT_SIZE]
		float          ** gpu_clt_ref,        // [num_cams] ->[TILES-Y][TILES-X][colors][DTT_SIZE*DTT_SIZE]
		int               colors,             // number of colors (3/1)
		float             scale0,             // scale for R
		float             scale1,             // scale for B
		float             scale2,             // scale for G
		float            * gpu_ftasks,        // flattened tasks, 27 floats for quad EO, 99 floats for LWIR16
		int               num_tiles,          // number of tiles in task
		int               tilesx,             // number of tile rows
		int             * gpu_corr_indices,   // packed tile+pair
		int             * pnum_corr_tiles,    // pointer to a number of correlation tiles to process
		size_t            corr_stride,        // in floats
		float           * gpu_corrs);         // correlation output data


extern "C" __global__ void corr2D_normalize(
extern "C" __global__ void corr2D_normalize(
		int               num_corr_tiles,     // number of correlation tiles to process
		int               num_corr_tiles,     // number of correlation tiles to process
		const size_t      corr_stride_td,     // in floats
		const size_t      corr_stride_td,     // in floats