Alnis Murtovi d3cefc9e3a AutoHeuristic: Collect data for mixed_mm (#131611) ...

This PR introduces a script that can be used to collect data for mixed_mm to learn a heuristic with AutoHeuristic. This PR also includes the following things:

Move pad_mm related AutoHeuristic files into subdirectory
Introduce an interface benchmark_runner.py that can be subclassed to introduce new scripts to run benchmarks in order to collect data with AutoHeuristic (see gen_data_pad_mm.py and gen_data_mixed_mm.py).
The idea behind the interface is that, in the end, it hopefully makes it easier to collect data for new optimizations, and thus makes it easier to learn a heuristic.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131611
Approved by: https://github.com/eellison
ghstack dependencies: #131610

2024-07-31 20:45:45 +00:00

..

_autoheuristic

AutoHeuristic: Collect data for mixed_mm (#131611)

2024-07-31 20:45:45 +00:00

aoti

[cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343)

2024-06-30 19:22:16 +00:00

api

[BE][Easy][5/19] enforce style for empty lines in import segments in tools/ and torchgen/ (#129756)

2024-07-17 06:44:35 +00:00

decompositions

[BE][Easy] eliminate relative import in torchgen (#128872)

2024-06-21 14:11:46 +00:00

dest

[Intel GPU] xpu-ops codegen via backend whitelist (#130082)

2024-07-31 16:31:38 +00:00

executorch

[BE][Easy][5/19] enforce style for empty lines in import segments in tools/ and torchgen/ (#129756)

2024-07-17 06:44:35 +00:00

fuse

[BE] update type annotations for basic utilities in torch/__init__.py (#129001)

2024-06-24 18:04:38 +00:00

operator_versions

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

selective_build

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

shape_functions

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

static_runtime

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

__init__.py

Package config/template files with torchgen (#78942)

2022-06-07 13:33:55 +00:00

BUCK.oss

rename BUILD.buck to BUCK.oss (#78792)

2022-06-03 07:23:16 +00:00

BUILD.bazel

Rename tools/codegen to torchgen (#76275)

2022-04-25 01:38:06 +00:00

build.bzl

update rules_python and let bazel install its own pip dependencies (#101405)

2023-05-23 06:20:33 +00:00

code_template.py

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

context.py

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

gen_aoti_c_shim.py

[cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343)

2024-06-30 19:22:16 +00:00

gen_backend_stubs.py

[BE][Easy] replace import pathlib with from pathlib import Path (#129426)

2024-06-30 01:36:07 +00:00

gen_executorch.py

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

gen_functionalization_type.py

propagate XLA's metadata after functional sync (#131076)

2024-07-31 18:20:00 +00:00

gen_lazy_tensor.py

[3/N] Change #include <c10/util/Optional.h> to #include <optional> (#130300)

2024-07-09 13:32:57 +00:00

gen_vmap_plumbing.py

[structural binding][7/N] Replace std::tie with structural binding (#130216)

2024-07-10 00:52:04 +00:00

gen.py

Include _native.h for structured_native_functions (#131208)

2024-07-24 02:55:36 +00:00

local.py

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

model.py

[Intel GPU] xpu-ops codegen via backend whitelist (#130082)

2024-07-31 16:31:38 +00:00

native_function_generation.py

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

utils.py

[torchgen] reference generated comment to actual location of the generator and template (#130020)

2024-07-05 21:47:14 +00:00

yaml_utils.py

[Reland] Update mypy to 1.4.1 (#105227)

2023-07-15 20:30:20 +00:00