Richard Zou f21a176c03 Python Dispatcher should respect FuncTorchBatchedDecomposition key (#98328) ...

Fixes https://github.com/pytorch/pytorch/issues/97425.

Python Dispatcher's resolve_key function should be equivalent to
computeDispatchTableEntryWithDebug. We added a section to
computeDispatchTableEntryWithDebug but forgot to add it to resolve_key.

This PR fixes that discrepancy.

Test Plan:
- new test
Pull Request resolved: https://github.com/pytorch/pytorch/pull/98328
Approved by: https://github.com/Chillee, https://github.com/kshitij12345, https://github.com/Neilblaze

2023-04-05 20:32:53 +00:00

..

api

Uniformly use elem when checking ListType (#97873)

2023-04-05 12:06:03 +00:00

decompositions

Rename tools/codegen to torchgen (#76275)

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

dest

[BE] Enable flake8-simplify checks (#97984)

2023-03-31 03:40:21 +00:00

executorch

[executorch] Add RuntimeContext to generated C++ API Signature (#94570)

2023-02-16 02:43:18 +00:00

operator_versions

[BE] Remove unnecessary dict comprehensions (#97116)

2023-03-20 00:56:57 +00:00

selective_build

[BE] Enable flake8-comprehension rule C417 (#97880)

2023-03-30 14:34:24 +00:00

shape_functions

[JIT SSA] Allow updating shape functions without recompilation (#83629)

2022-08-22 18:03:44 +00:00

static_runtime

Make segment_reduce properly private. (#93166)

2023-02-06 18:32:23 +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

run buildifier on unified build files (#98141)

2023-04-04 00:37:19 +00:00

code_template.py

Apply ufmt linter to all py files under torchgen (#81570)

2022-07-16 03:52:25 +00:00

context.py

Change API type Tensor[] for structured kernels. (#73350)

2022-09-26 21:46:38 +00:00

gen_backend_stubs.py

Bump black version to 23.1.0 (#96578)

2023-03-15 06:27:59 +00:00

gen_executorch.py

[ET] Add RuntimeContext to ET Aten mode (#96084)

2023-03-08 02:51:47 +00:00

gen_functionalization_type.py

Bump black version to 23.1.0 (#96578)

2023-03-15 06:27:59 +00:00

gen_lazy_tensor.py

Simplify by using yield from (#97831)

2023-03-29 19:15:24 +00:00

gen_vmap_plumbing.py

fix: update error when tensor escapes vmap (#89077)

2022-12-06 05:52:09 +00:00

gen.py

[torchgen] Bump native function max namespace levels due for internal use case (#97381)

2023-03-23 00:40:37 +00:00

local.py

Change API type Tensor[] for structured kernels. (#73350)

2022-09-26 21:46:38 +00:00

model.py

Python Dispatcher should respect FuncTorchBatchedDecomposition key (#98328)

2023-04-05 20:32:53 +00:00

native_function_generation.py

Refactor NT offsets metadata to be a Tensor (#96909)

2023-03-21 18:51:35 +00:00

utils.py

Simplify by using yield from (#97831)

2023-03-29 19:15:24 +00:00