123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- /* StarPU --- Runtime system for heterogeneous multicore architectures.
- *
- * Copyright (C) 2011,2012 Inria
- * Copyright (C) 2012,2015,2017 CNRS
- *
- * StarPU is free software; you can redistribute it and/or modify
- * it under the terms of the GNU Lesser General Public License as published by
- * the Free Software Foundation; either version 2.1 of the License, or (at
- * your option) any later version.
- *
- * StarPU is distributed in the hope that it will be useful, but
- * WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
- *
- * See the GNU Lesser General Public License in COPYING.LGPL for more details.
- */
- __kernel void tensor_opencl(__global int *tensor,
- int nx, int ny, int nz, int nt,
- int ldy, int ldz, int ldt,
- int factor, __global int *err)
- {
- const int id = get_global_id(0);
- if (id > 0)
- return;
- unsigned int i, j, k, l;
- int val = 0;
- for (l = 0; l < nt; l++)
- {
- for (k = 0; k < nz; k++)
- {
- for (j = 0; j < ny; j++)
- {
- for (i = 0; i < nx; i++)
- {
- if (tensor[(k*ldz)+(j*ldy)+i] != factor * val)
- {
- *err = 1;
- return;
- }
- else
- {
- tensor[(k*ldz)+(j*ldy)+i] *= -1;
- val++;
- }
- }
- }
- }
- }
- }
|