def gpu_mark_found_float(arr, val, out, not_found): i = cuda.grid(1) if i < arr.size: if check_equals_float(arr[i], val): out[i] = i else: out[i] = not_found
def gpu_mark_segment_begins_float(arr, markers): i = cuda.grid(1) if i == 0: markers[0] = 1 elif 0 < i < markers.size: if not markers[i]: markers[i] = not check_equals_float(arr[i], arr[i - 1])
def gpu_mark_found_float(arr, val, out, not_found): i = cuda.grid(1) if i < arr.size: # TODO: Remove val typecast to float(val) # once numba minimum version is pinned # at 0.51.1, this will have a very slight # performance improvement. Related # discussion in : https://github.com/rapidsai/cudf/pull/6073 if check_equals_float(arr[i], float(val)): out[i] = i else: out[i] = not_found