Sha256: 87edaf257fa41f0eaa20d87e09d2f8b35166816f2b432ba1c92864db3d915294
Contents?: true
Size: 547 Bytes
Versions: 8
Compression:
Stored size: 547 Bytes
Contents
% c_dtype = dtype_to_c_type(dtype)
__kernel void where_<%= dtype %>(const int M, const int N, __global const short *PRED, __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] = PRED[globalRow * N + globalCol] ? A[globalRow * N + globalCol] : B[globalRow * N + globalCol];
}
Version data entries
8 entries across 8 versions & 2 rubygems