opencl.cl: Add add, multiply and divide

ufo-kit · Jan 28, 2021 · cf4d5c4 · cf4d5c4
1 parent b480263
commit cf4d5c4
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/src/kernels/opencl.cl b/src/kernels/opencl.cl
@@ -118,3 +118,30 @@ diff (global float *x,
     const size_t idx = get_global_id (1) * get_global_size (0) + get_global_id (0);
     output[idx] = x[idx] - y[idx];
 }
+
+kernel void
+add (global float *x,
+     global float *y,
+     global float *output)
+{
+    const size_t idx = get_global_id (1) * get_global_size (0) + get_global_id (0);
+    output[idx] = x[idx] + y[idx];
+}
+
+kernel void
+multiply (global float *x,
+          global float *y,
+          global float *output)
+{
+    const size_t idx = get_global_id (1) * get_global_size (0) + get_global_id (0);
+    output[idx] = x[idx] * y[idx];
+}
+
+kernel void
+divide (global float *x,
+        global float *y,
+        global float *output)
+{
+    const size_t idx = get_global_id (1) * get_global_size (0) + get_global_id (0);
+    output[idx] = x[idx] / y[idx];
+}