Sha256: 58e2ab711e20f1b744503040d8825e685ca86fee9f52507dd5e1b291e69e7650

Contents?: true

Size: 509 Bytes

Versions: 23

Compression:

Stored size: 509 Bytes

Contents

% c_dtype = dtype_to_c_type(dtype)
 __kernel void argmin_<%= dtype %>(const int M, const int N, const int switch_op, __global const <%= c_dtype %> *A, __global const <%= c_dtype %> *B, __global <%= c_dtype %> *C) {
    // Get the index of the current element to be processed
    const int globalRow = get_global_id(0); // Row ID of C (0..M)
    const int globalCol = get_global_id(1); // Col ID of C (0..N)
    
    C[globalRow * N + globalCol] = A[globalRow * N + globalCol] + B[globalRow * N + globalCol];
}

Version data entries

23 entries across 23 versions & 2 rubygems

Version Path
tensor_stream-opencl-0.2.10 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.2.9 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.2.8 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.2.6 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.2.5 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.2.4 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.2.3 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.2.2 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.2.1 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.2.0 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.1.3 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.1.2 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.1.1 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-opencl-0.1.0 lib/tensor_stream/opencl/kernels/argmin.cl
tensor_stream-0.8.1 lib/tensor_stream/evaluator/opencl/kernels/argmin.cl
tensor_stream-0.8.0 lib/tensor_stream/evaluator/opencl/kernels/argmin.cl
tensor_stream-0.7.0 lib/tensor_stream/evaluator/opencl/kernels/argmin.cl
tensor_stream-0.6.1 lib/tensor_stream/evaluator/opencl/kernels/argmin.cl
tensor_stream-0.6.0 lib/tensor_stream/evaluator/opencl/kernels/argmin.cl
tensor_stream-0.5.1 lib/tensor_stream/evaluator/opencl/kernels/argmin.cl