Alnis Murtovi e51c8ad369 AutoHeuristic: Heuristic that ranks choices for mm (#131714) ...

This PR adds a heuristic for tuned_mm that predicts the top 10 best choices. To be safe, aten.mm is always included.

Perf run: https://hud.pytorch.org/benchmark/compilers?dashboard=torchinductor&startTime=Thu%2C%2008%20Aug%202024%2020%3A20%3A28%20GMT&stopTime=Thu%2C%2015%20Aug%202024%2020%3A20%3A28%20GMT&granularity=hour&suite=torchbench&mode=inference&dtype=bfloat16&deviceName=cuda%20(a100)&lBranch=gh/AlnisM/22/head&lCommit=905826f4ab5344efb0bcaa87e3b27a25299927ab&rBranch=main&rCommit=79ca596dc6ea16b6cdd0f2517451e19840717d37

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131714
Approved by: https://github.com/eellison
ghstack dependencies: #131705, #131710

2024-08-16 16:20:38 +00:00

..

_autoheuristic

AutoHeuristic: Heuristic that ranks choices for mm (#131714)

2024-08-16 16:20:38 +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

Improve namespace for c10::MemoryFormat::Contiguous in torchgen/api/cpp.py (#131622)

2024-08-14 14:41:01 +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

[12/N] Use std::optional (#132361)

2024-08-02 13:46:46 +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

[12/N] Use std::optional (#132361)

2024-08-02 13:46:46 +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