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
