| commit | 57ed94804e8195f227c7a75899a319cc0a3b833a | [log] [tgz] |
|---|---|---|
| author | Edward Z. Yang <ezyang@fb.com> | Mon Nov 21 16:04:46 2022 -0500 |
| committer | PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> | Tue Nov 22 00:32:30 2022 +0000 |
| tree | a65dfd4f729033687941fee832e71c0162b5242f | |
| parent | b189a7444da8b17c535e7d04c9ab705289ec53e1 [diff] |
Bind DispatchKey.Functionalonalize in pybind11 (#89452) Signed-off-by: Edward Z. Yang <ezyang@fb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/89452 Approved by: https://github.com/albanD, https://github.com/bdhirsh
diff --git a/torch/csrc/utils/python_dispatch.cpp b/torch/csrc/utils/python_dispatch.cpp index 381e82e..e4ce9cc 100644 --- a/torch/csrc/utils/python_dispatch.cpp +++ b/torch/csrc/utils/python_dispatch.cpp
@@ -495,6 +495,7 @@ DEF_ONE(FuncTorchDynamicLayerFrontMode) DEF_ONE(FuncTorchDynamicLayerBackMode) DEF_ONE(PythonDispatcher) + DEF_ONE(Functionalize) // clang-format on #define DEF_SINGLE(n, prefix) .value(#prefix #n, c10::DispatchKey::prefix##n)