frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE][Easy] enable postponed annotations in tools (#129375)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/129375
Approved by: https://github.com/malfet
This commit is contained in:
Xuehai Pan
2024-06-29 12:48:06 +08:00
committed by PyTorch MergeBot
parent 58f346c874
commit 8a67daf283
123 changed files with 1274 additions and 1053 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tools/setup_helpers/generate_code.py
View File

@ -1,8 +1,10 @@
from __future__ import annotations
import argparse
import os
import pathlib
import sys
from typing import Any, cast, Optional
from typing import Any, cast
import yaml
@ -18,10 +20,10 @@ TAGS_PATH = "aten/src/ATen/native/tags.yaml"
def generate_code(
gen_dir: pathlib.Path,
native_functions_path: Optional[str] = None,
tags_path: Optional[str] = None,
install_dir: Optional[str] = None,
subset: Optional[str] = None,
native_functions_path: str | None = None,
tags_path: str | None = None,
install_dir: str | None = None,
subset: str | None = None,
disable_autograd: bool = False,
force_schema_registration: bool = False,
operator_selector: Any = None,
@ -102,8 +104,8 @@ def get_selector_from_legacy_operator_selection_list(
def get_selector(
selected_op_list_path: Optional[str],
operators_yaml_path: Optional[str],
selected_op_list_path: str | None,
operators_yaml_path: str | None,
) -> Any:
# cwrap depends on pyyaml, so we can't import it earlier
root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))