Function main

python/2_CoreConcepts/binarySearch/binarySearch.py:107–143

()

def main():
    import argparse

    parser = argparse.ArgumentParser(
        description="Parallel upper_bound / lower_bound via cuda.compute"
    )
    parser.add_argument("--device", type=int, default=0, help="CUDA device id")
    args = parser.parse_args()

    device = Device(args.device)
    device.set_current()
    print_gpu_info(device)
    print()

    ok = True

    # Case 1: values both inside and outside the data range; no duplicates
    # in the data. lower_bound and upper_bound agree on values not present.
    print("Case 1: distinct data, mixed queries")
    h_data1 = np.array([1, 3, 5, 7, 9], dtype=np.int32)
    h_values1 = np.array([0, 3, 4, 10], dtype=np.int32)
    ok &= run_binary_search(h_data1, h_values1)
    print()

    # Case 2: duplicates in the data so lower_bound and upper_bound diverge
    # on present values.
    print("Case 2: duplicates in data")
    h_data2 = np.array([1, 3, 3, 5, 7, 9], dtype=np.int32)
    h_values2 = np.array([3, 3, 5, 8], dtype=np.int32)
    ok &= run_binary_search(h_data2, h_values2)

    print()
    if ok:
        print("Done")
        return 0
    print("FAILED")
    return 1


if __name__ == "__main__":