-
Notifications
You must be signed in to change notification settings - Fork 6
Expand file tree
/
Copy pathadjacent_difference.cpp
More file actions
25 lines (21 loc) · 973 Bytes
/
adjacent_difference.cpp
File metadata and controls
25 lines (21 loc) · 973 Bytes
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
#include "adjacent_difference.h"
bool TRTC_Adjacent_Difference(const DVVectorLike& vec_in, DVVectorLike& vec_out)
{
static TRTC_For s_for({ "view_vec_in", "view_vec_out"}, "idx",
" auto value = view_vec_in[idx ];\n"
" if (idx>0) value -= view_vec_in[idx - 1]; \n"
" view_vec_out[idx] = (decltype(view_vec_out)::value_t) value;\n"
);
const DeviceViewable* args[] = { &vec_in, &vec_out };
return s_for.launch_n(vec_in.size(), args);
}
bool TRTC_Adjacent_Difference(const DVVectorLike& vec_in, DVVectorLike& vec_out, const Functor& binary_op)
{
static TRTC_For s_for({ "view_vec_in", "view_vec_out", "binary_op" }, "idx",
" auto value = view_vec_in[idx];\n"
" if (idx>0) value = (decltype(view_vec_in)::value_t) binary_op(value, view_vec_in[idx - 1]); \n"
" view_vec_out[idx] = (decltype(view_vec_out)::value_t) value;\n"
);
const DeviceViewable* args[] = { &vec_in, &vec_out, &binary_op };
return s_for.launch_n(vec_in.size(), args);
}