parser() — pytorch Function Reference

Architecture documentation for the parser() function in generate_kernels.py from the pytorch codebase.

Function python ComputeKernels QuantizedCPU calls 1

Entity Profile

ComputeKernels→ QuantizedCPU→ parser() — pytorch Function Reference

Dependency Diagram

graph TD
  7fedab6d_8981_08c3_ca38_2f0eddd64dcc["parser()"]
  20303473_0948_19b1_a212_748886b73572["main()"]
  7fedab6d_8981_08c3_ca38_2f0eddd64dcc -->|calls| 20303473_0948_19b1_a212_748886b73572
  style 7fedab6d_8981_08c3_ca38_2f0eddd64dcc fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/generate_kernels.py lines 401–414

    parser = argparse.ArgumentParser(
        prog="generate_kernels",
        description="Generate the mem-eff kernels template instantiations",
    )
    # Set an optional output directory
    parser.add_argument(
        "-o",
        "--output_dir",
        required=False,
        help="Where to generate the kernels "
        " will default to <ATen/native/transformers/cuda/mem_eff_attention/kernels/> ",
    )
    args = parser.parse_args()
    main(args.output_dir)