Function main

python/2_CoreConcepts/simpleZeroCopy/simpleZeroCopy.py:225–271

Parse CLI, call ``run()``, and exit with validation status.

()

def main():
    """Parse CLI, call ``run()``, and exit with validation status."""
    parser = argparse.ArgumentParser(
        description="Demonstrate zero-copy memory access with CUDA",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Examples:
  python simpleZeroCopy.py
  python simpleZeroCopy.py --num_elements 2097152
What is Zero-Copy Memory?
  Zero-copy allows the GPU to directly access host (CPU) memory without
  explicit memory transfers. This is useful for:
  - Small data that doesn't benefit from explicit transfers
  - Data that is accessed infrequently
  - Integrated GPUs that share memory with CPU

  Trade-offs:
  - Slower than device memory (PCIe bandwidth limited)
  - No explicit transfers needed (simpler code)
  - Good for discrete GPUs with small data
  - Excellent for integrated GPUs (e.g., Tegra)
        """,
    )

    parser.add_argument(
        "--num_elements",
        type=int,
        default=1048576,
        help="Number of elements in vectors (default: 1048576)",
    )

    args = parser.parse_args()

    if args.num_elements <= 0:
        print("Error: num_elements must be positive")
        sys.exit(1)

    try:
        exit_code = run(num_elements=args.num_elements)
    except Exception as e:
        print(f"\nError: {e}")
        import traceback

        traceback.print_exc()
        exit_code = 1

    sys.exit(exit_code)


if __name__ == "__main__":