added split/join to relu
This commit is contained in:
Binary file not shown.
@@ -12,9 +12,11 @@ void kernel_body(int __DIVERGENT__ task_id, kernel_arg_t* arg) {
|
|||||||
|
|
||||||
int32_t ref_value = src_ptr[task_id];
|
int32_t ref_value = src_ptr[task_id];
|
||||||
int ref_negative = ref_value < 0;
|
int ref_negative = ref_value < 0;
|
||||||
|
vx_split(ref_negative);
|
||||||
if (ref_negative) {
|
if (ref_negative) {
|
||||||
ref_value = 0;
|
ref_value = 0;
|
||||||
}
|
}
|
||||||
|
vx_join();
|
||||||
|
|
||||||
dst_ptr[task_id] = ref_value;
|
dst_ptr[task_id] = ref_value;
|
||||||
}
|
}
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
Binary file not shown.
Reference in New Issue
Block a user