File: auxiliary.cl

package info (click to toggle)
tiny-dnn 1.0.0a3%2Bds-3
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 4,760 kB
  • sloc: cpp: 16,471; ansic: 11,829; lisp: 3,682; python: 3,422; makefile: 206
file content (10 lines) | stat: -rw-r--r-- 261 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
#ifndef __OPENCL_VERSION__
#include "header.cl"
#endif

__kernel void TEMPLATE(gpu_set,Dtype)(const int_tp n, const Dtype alpha, __global Dtype* y) {
  for (int_tp index = get_global_id(0); index < n; index += get_global_size(0)) {
    y[index] = alpha;
  }
}