MCPcopy Create free account

hub / github.com/StanfordAHA/aha / functions

Functions108 in github.com/StanfordAHA/aha

↓ 35 callersFunctionparse_layer_parametrized_test
(testname, keyword, layer_in="")
aha/util/regress_util.py:370
↓ 17 callersFunctionbuildkite_call
(command, env={}, return_output=False, out_file=None)
aha/util/regress_util.py:27
↓ 11 callersFunctionreport_ongoing_failures
Keep track of which tests have failed already, if any
aha/util/regress.py:17
↓ 9 callersFunctionsubprocess_call_log
(cmd, cwd, env, log, log_file_path, use_shell=False)
aha/util/map.py:28
↓ 5 callersFunctionsubprocess_call_log
(cmd, cwd, env, log, log_file_path)
aha/util/test.py:26
↓ 4 callersFunctionbuildkite_call
(command)
aha/util/my_regress.py:21
↓ 4 callersFunctionsummarize_and_print_info
Print a readable summary of the info table. Try not to swizzle it!
aha/util/regress_info.py:28
↓ 4 callersFunctiontest_dense_app
( test, tgroup, width, height, env_parameters, extra_args, dense_only=False, using_matrix_unit=False, mu_d
aha/util/regress_util.py:412
↓ 4 callersFunctiontrack_performance
()
aha/util/regress_util.py:377
↓ 3 callersFunctionbuildkite_filter
(s)
aha/util/my_regress.py:17
↓ 3 callersMethodconfigs_template
()
aha/util/regress_tests/tests.py:54
↓ 3 callersFunctiongen_garnet
(width, height, dense_only=False, using_matrix_unit=False, mu_datawidth=16, num_fabric_cols_removed=0, mu_oc_0
aha/util/regress_util.py:68
↓ 3 callersFunctiongenerate_sparse_bitstreams
(sparse_tests, width, height, seed_flow, data_tile_pairs, kernel_name, opal_workaround=False, unroll=1, using_
aha/util/regress_util.py:110
↓ 3 callersFunctionsubprocess_call_log
Can set e.g. do_cmd=Popen to run job in background
aha/util/pnr.py:48
↓ 3 callersFunctiontest_sparse_app
(testname, seed_flow, data_tile_pairs, pipeline_num_l=None, opal_worka
aha/util/regress_util.py:237
↓ 2 callersFunction_load_hex_txt
(txt_path: str, block_index: int = 0)
aha/util/test.py:126
↓ 2 callersFunction_set_deterministic_cpu_env
()
aha/util/regress_util.py:16
↓ 2 callersFunctionbfbin2float
(bfstr)
aha/util/test.py:48
↓ 2 callersFunctionbuild_dependency_graph
(deps)
aha/util/deps.py:134
↓ 2 callersFunctionformat_concat_tiles
(test, data_tile_pairs, kernel_name, pipeline_num=32, unroll=1)
aha/util/regress_util.py:199
↓ 2 callersMethodget_config
(config_name='', zircon=True)
aha/util/regress_tests/tests.py:1619
↓ 2 callersFunctionhhmm
Turn nsec into hhmm e.g. hhmm(3600)="1h00" or hhmm(120)="0h02"
aha/util/regress_info.py:98
↓ 2 callersFunctionparse_E64_MB_mode
(testname)
aha/util/regress_util.py:349
↓ 2 callersFunctionparse_E64_mode
(testname)
aha/util/regress_util.py:341
↓ 2 callersFunctionparse_RV_mode
(testname)
aha/util/regress_util.py:333
↓ 2 callersMethodprefix_lines
Attach the indicated prefix to each line in "lines
aha/util/regress_tests/tests.py:1549
↓ 2 callersFunctionretry
(r_func, *args, **kwargs)
aha/util/pnr.py:23
↓ 2 callersFunctionunpack_output
(arr)
aha/util/test.py:107
↓ 1 callersFunction_strait_compile_full_model
Run strait compilation (protobuf->coreir) for the given model.
aha/util/strait.py:220
↓ 1 callersFunction_strait_kernel_fp_output_map
Return a mapping {kernel_name: has_bfloat16_output}. Uses the scheduled_ops.json emitted by strait, which records per-kernel output datatypes
aha/util/strait.py:263
↓ 1 callersFunction_voyager_compile_full_model
Run voyager-compiler (PyTorch -> protobuf) for the given model. Creates a dedicated Python 3.10+ venv under strait_path so regression can sta
aha/util/strait.py:97
↓ 1 callersFunctionappgroups
# - calculate grouptimes # - add blank lines before each "APP GROUP" and garnet "NO Zircon" compilation # - indent app names # - make
aha/util/regress_info.py:70
↓ 1 callersMethodbuild_skip_cgra_map_list
(DBG=False)
aha/util/regress_tests/tests.py:1598
↓ 1 callersFunctioncheck_for_dupes
(DBG=0)
aha/util/regress_tests/tests.py:1670
↓ 1 callersFunctionclear_unwanted_groups
Clear all {tests_dict} groups that are not in [wanted_groups]
aha/util/regress.py:61
↓ 1 callersMethoddetect_and_process_json
if "config" is a parsable json string, add it to selfdict and return True
aha/util/regress_tests/tests.py:1554
↓ 1 callersMethoddetect_and_process_yaml
if "config" is a parsable yaml string, add it to selfdict and return True
aha/util/regress_tests/tests.py:1564
↓ 1 callersFunctiondo_gold_check
(args, post_silicon_check=False, post_silicon_base_dir="", post_silicon_full_app_name="")
aha/util/test.py:145
↓ 1 callersFunctiondump_to_csv
(report_items, csv_path)
aha/util/report.py:162
↓ 1 callersFunctioneliminate_skips
Delete SKIP lines and add their info to the first non-skip line after e.g. BEFORE: 49 resnet18-submodule -> zircon_dequantize_relu
aha/util/regress_info.py:42
↓ 1 callersFunctiongen_garnet
(width, height)
aha/util/my_regress.py:39
↓ 1 callersFunctionget_absolute_time
(report_items)
aha/util/report.py:149
↓ 1 callersFunctionget_array_dimension
()
aha/util/report.py:94
↓ 1 callersFunctionget_glb_results
(aha_glb_log_path)
aha/util/report.py:68
↓ 1 callersFunctionget_install_requires
(expr)
aha/util/deps.py:85
↓ 1 callersFunctionget_map_results
(aha_map_log_path)
aha/util/report.py:20
↓ 1 callersFunctionget_num_tracks
()
aha/util/report.py:127
↓ 1 callersFunctionget_reqs
(module)
aha/util/deps.py:115
↓ 1 callersMethodget_skip_candidates
()
aha/util/regress_tests/tests.py:1580
↓ 1 callersFunctionget_sta_results
(aha_sta_log_path)
aha/util/report.py:43
↓ 1 callersFunctionin_docker
()
aha/util/docker.py:103
↓ 1 callersFunctioninstall
(package)
aha/util/deps.py:29
↓ 1 callersFunctionload_environmental_vars
(env, app, layer=None, env_parameters=None)
aha/util/test.py:77
↓ 1 callersFunctionload_environmental_vars
(env, app, layer=None, env_parameters=None)
aha/util/map.py:51
↓ 1 callersFunctionload_environmental_vars
(env, app, layer=None, env_parameters=None)
aha/util/pnr.py:72
↓ 1 callersFunctionmain
()
aha/aha.py:10
↓ 1 callersFunctionorder_deps
(deps)
aha/util/deps.py:145
↓ 1 callersFunctionparse_install_requires
(elts)
aha/util/deps.py:69
↓ 1 callersFunctionparse_mu_cgra_test
(testname)
aha/util/regress_util.py:357
↓ 1 callersFunctionparse_setup
(filename)
aha/util/deps.py:94
↓ 1 callersFunctionparse_voyager_cgra_test
(testname)
aha/util/regress_util.py:364
↓ 1 callersFunctionprint_report_items
(report_items)
aha/util/report.py:87
↓ 1 callersFunctionretry
(r_func, *args, **kwargs)
aha/util/garnet.py:13
↓ 1 callersFunctionrun_test
(testname, width, height)
aha/util/my_regress.py:55
↓ 1 callersFunctionset_group_filter
Accomodations for e.g. "--group sparse_tests" or "--groups glb_tests,resnet_tests" Returns e.g. ["glb_tests","resnet_tests"]
aha/util/regress.py:46
↓ 1 callersMethodshow_config
(config_name='', zircon=True)
aha/util/regress_tests/tests.py:1659
↓ 1 callersFunctionsubprocess_call_log
(cmd, cwd, log, log_file_path)
aha/util/sta.py:15
↓ 1 callersFunctiontest_dense_ml_model
( model, width, height, env_parameters, extra_args, mu_datawidth=16, num_fabric_cols_removed=0, mu_oc_
aha/util/regress_util.py:729
↓ 1 callersFunctiontest_hardcoded_dense_app
( test, width, height, env_parameters, extra_args, using_matrix_unit=False, mu_datawidth=16,
aha/util/regress_util.py:924
↓ 1 callersFunctiontrack_ml_model_performance
()
aha/util/regress_util.py:390
↓ 1 callersFunctionvoyager_run_layer
(model, layer)
aha/util/map.py:85
Method__init__
(self, testname="BLANK", zircon=True)
aha/util/regress_tests/tests.py:78
Function_natural_sort_key
(s)
aha/util/regress_util.py:764
Functionadd_subparser
(subparser)
aha/util/config.py:5
Functionadd_subparser
(subparser)
aha/util/regress.py:29
Functionadd_subparser
(subparser)
aha/util/test.py:9
Functionadd_subparser
(subparser)
aha/util/report.py:11
Functionadd_subparser
(subparser)
aha/util/strait.py:8
Functionadd_subparser
(subparser)
aha/util/garnet.py:7
Functionadd_subparser
(subparser)
aha/util/pd.py:6
Functionadd_subparser
(subparser)
aha/util/map.py:12
Functionadd_subparser
(subparser)
aha/util/docker.py:8
Functionadd_subparser
(subparser)
aha/util/sta.py:8
Functionadd_subparser
(subparser)
aha/util/pnr.py:9
Functionadd_subparser
(subparser)
aha/util/app.py:12
Functionadd_subparser
(subparser)
aha/util/deps.py:16
Functionadd_subparser
(subparser)
aha/util/my_regress.py:9
Functiondispatch
(args, extra_args=None)
aha/util/config.py:12
Functiondispatch
(args, extra_args=None)
aha/util/regress.py:71
Functiondispatch
(args, extra_args=None)
aha/util/test.py:501
Functiondispatch
(args, extra_args=None)
aha/util/report.py:172
Functiondispatch
Entry point for `aha strait`. Runs voyager-compiler (PyTorch -> protobuf) followed by strait (protobuf -> CoreIR) for the requested dense ML
aha/util/strait.py:55
Functiondispatch
(args, extra_args=None)
aha/util/garnet.py:38
Functiondispatch
(args, extra_args=None)
aha/util/pd.py:13
Functiondispatch
(args, extra_args=None)
aha/util/map.py:116
Functiondispatch
(args, extra_args=None)
aha/util/docker.py:19
Functiondispatch
(args, extra_args=None)
aha/util/sta.py:34
Functiondispatch
(args, extra_args=None)
aha/util/pnr.py:108
Functiondispatch
(args, extra_args=None)
aha/util/app.py:16
Functiondispatch
(args, extra_args=None)
aha/util/deps.py:23
next →1–100 of 108, ranked by callers