Function

main() — pytorch Function Reference

Architecture documentation for the main() function in analyze_templates.py from the pytorch codebase.

Function python

optimize_templates() — pytorch Function Reference

Architecture documentation for the optimize_templates() function in analyze_templates.py from the pytorch codebase.

Function python

parse_log_file() — pytorch Function Reference

Architecture documentation for the parse_log_file() function in analyze_templates.py from the pytorch codebase.

Function python

bench() — pytorch Function Reference

Architecture documentation for the bench() function in bench_mm_fusion.py from the pytorch codebase.

Function python

fusion_type() — pytorch Function Reference

Architecture documentation for the fusion_type() function in bench_mm_fusion.py from the pytorch codebase.

Function python

mm_add_relu() — pytorch Function Reference

Architecture documentation for the mm_add_relu() function in bench_mm_fusion.py from the pytorch codebase.

Function python

mm_add() — pytorch Function Reference

Architecture documentation for the mm_add() function in bench_mm_fusion.py from the pytorch codebase.

Function python

mm_relu() — pytorch Function Reference

Architecture documentation for the mm_relu() function in bench_mm_fusion.py from the pytorch codebase.

Function python

mm() — pytorch Function Reference

Architecture documentation for the mm() function in bench_mm_fusion.py from the pytorch codebase.

Function python

_benchmark_tile_reduce_single() — pytorch Function Reference

Architecture documentation for the _benchmark_tile_reduce_single() function in bench_nvshmem_tile_reduce.py from the pytorch codebase.

Function python

device() — pytorch Function Reference

Architecture documentation for the device() function in bench_nvshmem_tile_reduce.py from the pytorch codebase.

Function python

_init_device() — pytorch Function Reference

Architecture documentation for the _init_device() function in bench_nvshmem_tile_reduce.py from the pytorch codebase.

Function python

requires_nvshmem() — pytorch Function Reference

Architecture documentation for the requires_nvshmem() function in bench_nvshmem_tile_reduce.py from the pytorch codebase.

Function python

test_benchmark_tile_reduce_various_sizes() — pytorch Function Reference

Architecture documentation for the test_benchmark_tile_reduce_various_sizes() function in bench_nvshmem_tile_reduce.py from the pytorch codebase.

Function python

torch() — pytorch Function Reference

Architecture documentation for the torch() function in bench_nvshmem_tile_reduce.py from the pytorch codebase.

Function python

time_with_torch_timer() — pytorch Function Reference

Architecture documentation for the time_with_torch_timer() function in benchmark_helper.py from the pytorch codebase.

Function python

allequal() — pytorch Function Reference

Architecture documentation for the allequal() function in benchmark.py from the pytorch codebase.

Function python

allgather_object() — pytorch Function Reference

Architecture documentation for the allgather_object() function in benchmark.py from the pytorch codebase.

Function python

allgather_run() — pytorch Function Reference

Architecture documentation for the allgather_run() function in benchmark.py from the pytorch codebase.

Function python

benchmark_process_group() — pytorch Function Reference

Architecture documentation for the benchmark_process_group() function in benchmark.py from the pytorch codebase.

Function python

create_model() — pytorch Function Reference

Architecture documentation for the create_model() function in benchmark.py from the pytorch codebase.

Function python

create_model() — pytorch Function Reference

Architecture documentation for the create_model() function in benchmark.py from the pytorch codebase.

Function python

generate_inputs() — pytorch Function Reference

Architecture documentation for the generate_inputs() function in benchmark.py from the pytorch codebase.

Function python

generate_inputs() — pytorch Function Reference

Architecture documentation for the generate_inputs() function in benchmark.py from the pytorch codebase.

Function python

generate_target() — pytorch Function Reference

Architecture documentation for the generate_target() function in benchmark.py from the pytorch codebase.

Function python

generate_target() — pytorch Function Reference

Architecture documentation for the generate_target() function in benchmark.py from the pytorch codebase.

Function python

__init__() — pytorch Function Reference

Architecture documentation for the __init__() function in benchmark.py from the pytorch codebase.

Function python

__init__() — pytorch Function Reference

Architecture documentation for the __init__() function in benchmark.py from the pytorch codebase.

Function python

list_benchmarks() — pytorch Function Reference

Architecture documentation for the list_benchmarks() function in benchmark.py from the pytorch codebase.

Function python

main() — pytorch Function Reference

Architecture documentation for the main() function in benchmark.py from the pytorch codebase.

Function python

main() — pytorch Function Reference

Architecture documentation for the main() function in benchmark.py from the pytorch codebase.

Function python

run_all_benchmarks() — pytorch Function Reference

Architecture documentation for the run_all_benchmarks() function in benchmark.py from the pytorch codebase.

Function python

_run_benchmark() — pytorch Function Reference

Architecture documentation for the _run_benchmark() function in benchmark.py from the pytorch codebase.

Function python

run_benchmark() — pytorch Function Reference

Architecture documentation for the run_benchmark() function in benchmark.py from the pytorch codebase.

Function python

run_benchmark() — pytorch Function Reference

Architecture documentation for the run_benchmark() function in benchmark.py from the pytorch codebase.

Function python

show_environment_info() — pytorch Function Reference

Architecture documentation for the show_environment_info() function in benchmark.py from the pytorch codebase.

Function python

__str__() — pytorch Function Reference

Architecture documentation for the __str__() function in benchmark.py from the pytorch codebase.

Function python

__str__() — pytorch Function Reference

Architecture documentation for the __str__() function in benchmark.py from the pytorch codebase.

Function python

sweep() — pytorch Function Reference

Architecture documentation for the sweep() function in benchmark.py from the pytorch codebase.

Function python

args() — pytorch Function Reference

Architecture documentation for the args() function in benchmarks.py from the pytorch codebase.

Function python

AssertionError() — pytorch Function Reference

Architecture documentation for the AssertionError() function in benchmarks.py from the pytorch codebase.

Function python

model_names() — pytorch Function Reference

Architecture documentation for the model_names() function in benchmarks.py from the pytorch codebase.

Function python

parse_args() — pytorch Function Reference

Architecture documentation for the parse_args() function in benchmarks.py from the pytorch codebase.

Function python

huge_graph() — pytorch Function Reference

Architecture documentation for the huge_graph() function in cache_debug_microbenchmarks.py from the pytorch codebase.

Function python

main() — pytorch Function Reference

Architecture documentation for the main() function in cache_debug_microbenchmarks.py from the pytorch codebase.

Function python

huge_graph() — pytorch Function Reference

Architecture documentation for the huge_graph() function in cache_hit_microbenchmarks.py from the pytorch codebase.

Function python

main() — pytorch Function Reference

Architecture documentation for the main() function in cache_hit_microbenchmarks.py from the pytorch codebase.

Function python

get_compile_time() — pytorch Function Reference

Architecture documentation for the get_compile_time() function in cachebench.py from the pytorch codebase.

Function python