Sha256: 5448cc03a39c28ae4075699eb6d885fd4ec2b87b412e9d368478601c477da0da

Contents?: true

Size: 424 Bytes

Versions: 10

Compression:

Stored size: 424 Bytes

Contents

% c_dtype = dtype_to_c_type(dtype)
__kernel void floor_<%= dtype %>(const int M, const int N, __global const <%= c_dtype %> *A, __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] = floor(A[globalRow * N + globalCol]);
}

Version data entries

10 entries across 10 versions & 2 rubygems

Version Path
tensor_stream-opencl-0.1.3 lib/tensor_stream/opencl/kernels/floor.cl
tensor_stream-opencl-0.1.2 lib/tensor_stream/opencl/kernels/floor.cl
tensor_stream-opencl-0.1.1 lib/tensor_stream/opencl/kernels/floor.cl
tensor_stream-opencl-0.1.0 lib/tensor_stream/opencl/kernels/floor.cl
tensor_stream-0.8.1 lib/tensor_stream/evaluator/opencl/kernels/floor.cl
tensor_stream-0.8.0 lib/tensor_stream/evaluator/opencl/kernels/floor.cl
tensor_stream-0.7.0 lib/tensor_stream/evaluator/opencl/kernels/floor.cl
tensor_stream-0.6.1 lib/tensor_stream/evaluator/opencl/kernels/floor.cl
tensor_stream-0.6.0 lib/tensor_stream/evaluator/opencl/kernels/floor.cl
tensor_stream-0.5.1 lib/tensor_stream/evaluator/opencl/kernels/floor.cl