Function main

python/1_GettingStarted/vectorAdd/vectorAdd.py:158–192

Main entry point for the vector addition sample.

()

def main():
    """
    Main entry point for the vector addition sample.
    """
    import argparse

    parser = argparse.ArgumentParser(description="Vector Addition using CUDA Core API")
    parser.add_argument(
        "--elements",
        type=int,
        default=50000,
        help="Number of elements in vectors (default: 50000)",
    )
    parser.add_argument(
        "--device", type=int, default=0, help="CUDA device ID (default: 0)"
    )
    parser.add_argument(
        "--no-verify", action="store_true", help="Skip result verification"
    )

    args = parser.parse_args()

    if args.elements <= 0:
        print("Error: Number of elements must be positive")
        return 1

    success = vector_add_cuda_core(
        num_elements=args.elements, device_id=args.device, verify=not args.no_verify
    )

    if success:
        print("\nDone")
        return 0
    else:
        return 1


if __name__ == "__main__":