chilli 938f37b745 Added batching rule for sdpa_math, sdpa_efficient_attention forward, cudnn, and flash attention (#133964) ...

Fixes https://github.com/pytorch/pytorch/issues/117016, https://github.com/pytorch/pytorch/issues/102457, https://github.com/pytorch/pytorch/issues/110525, https://github.com/pytorch/pytorch/issues/108065,

Pull Request resolved: https://github.com/pytorch/pytorch/pull/133964
Approved by: https://github.com/Skylion007

2024-08-22 05:29:49 +00:00

..

_autoheuristic

AutoHeuristic: documentation for mm (#133611)

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_schema_utils.py

[HOP] support generating schema for hop (#133521)

2024-08-21 17:34:21 +00:00

gen_vmap_plumbing.py

Added batching rule for sdpa_math, sdpa_efficient_attention forward, cudnn, and flash attention (#133964)

2024-08-22 05:29:49 +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

[HOP] support generating schema for hop (#133521)

2024-08-21 17:34:21 +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