mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
db6bd9d60b
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43889 1. rename input argunment `interested-folder` to `interest-only` -- be consistent with `run-only`, `coverage-only` and be shorted Test Plan: Test on devserver and linux docker. Reviewed By: malfet Differential Revision: D23417338 fbshipit-source-id: ce9711e75ca3a1c30801ad6bd1a620f3b06819c5
102 lines
2.5 KiB
Python
102 lines
2.5 KiB
Python
import argparse
|
|
import os
|
|
from typing import Any
|
|
|
|
from .setting import (
|
|
JSON_FOLDER_BASE_DIR,
|
|
LOG_DIR,
|
|
MERGED_FOLDER_BASE_DIR,
|
|
PROFILE_DIR,
|
|
SUMMARY_FOLDER_DIR,
|
|
Option,
|
|
)
|
|
from .utils import create_folder, get_raw_profiles_folder, remove_file
|
|
|
|
|
|
def remove_files() -> None:
|
|
# remove log
|
|
remove_file(os.path.join(LOG_DIR, "log.txt"))
|
|
|
|
|
|
def create_folders() -> None:
|
|
create_folder(
|
|
PROFILE_DIR,
|
|
MERGED_FOLDER_BASE_DIR,
|
|
JSON_FOLDER_BASE_DIR,
|
|
get_raw_profiles_folder(),
|
|
SUMMARY_FOLDER_DIR,
|
|
LOG_DIR,
|
|
)
|
|
|
|
|
|
def add_arguments_utils(parser: argparse.ArgumentParser) -> argparse.ArgumentParser:
|
|
parser.add_argument("--run", help="run the cpp test binaries", action="store_true")
|
|
parser.add_argument(
|
|
"--merge",
|
|
help="merge raw profiles (only apply to clang coverage)",
|
|
action="store_true",
|
|
)
|
|
parser.add_argument(
|
|
"--export", help="generate json report for each file", action="store_true"
|
|
)
|
|
parser.add_argument(
|
|
"--summary",
|
|
help="read json report and generate file/line-oriented summary",
|
|
action="store_true",
|
|
)
|
|
parser.add_argument(
|
|
"--interest-only",
|
|
help="Final report will be only about these folders and its sub-folders; for example: caff2/c10;",
|
|
nargs="+",
|
|
default=None,
|
|
)
|
|
parser.add_argument(
|
|
"--clean",
|
|
help="delete all files generated by coverage tool",
|
|
action="store_true",
|
|
default=False,
|
|
)
|
|
|
|
return parser
|
|
|
|
|
|
def have_option(have_stage: bool, option: int) -> int:
|
|
if have_stage:
|
|
return option
|
|
else:
|
|
return 0
|
|
|
|
|
|
def get_options(args: Any) -> Option:
|
|
option: Option = Option()
|
|
if args.__contains__("build"):
|
|
if args.build:
|
|
option.need_build = True
|
|
|
|
if args.__contains__("run"):
|
|
if args.run:
|
|
option.need_run = True
|
|
|
|
if args.__contains__("merge"):
|
|
if args.merge:
|
|
option.need_merge = True
|
|
|
|
if args.__contains__("export"):
|
|
if args.export:
|
|
option.need_export = True
|
|
|
|
if args.__contains__("summary"):
|
|
if args.summary:
|
|
option.need_summary = True
|
|
|
|
# user does not have specified stage like run
|
|
if not any(vars(option).values()):
|
|
option.need_build = True
|
|
option.need_run = True
|
|
option.need_merge = True
|
|
option.need_export = True
|
|
option.need_summary = True
|
|
option.need_pytest = True
|
|
|
|
return option
|