frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Python Dispatcher integration with C++ dispatcher (#84826)"

Browse Source 
This reverts commit 35f6a69191ef762cf22b6cbfe94b8d9406e16674.

Reverted https://github.com/pytorch/pytorch/pull/84826 on behalf of https://github.com/malfet due to Broke dynamo, see 35f6a69191
This commit is contained in:
PyTorch MergeBot
2022-09-14 14:07:58 +00:00
parent 74ead61944
commit 706b990306
33 changed files with 160 additions and 686 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
torch/csrc/jit/python/init.cpp
View File

@ -1713,11 +1713,13 @@ void initJITBindings(PyObject* module) {
return _get_operation_for_overload_or_packet(
{op}, symbol, args, kwargs, /*is_overload*/ true);
});
auto func_dk = py::cpp_function(
[op, symbol, allow_numbers_as_tensors](
c10::DispatchKey dk_, py::args args, py::kwargs kwargs) {
auto func_dk =
py::cpp_function([op, symbol, allow_numbers_as_tensors](
const std::string& str_dk,
py::args args,
py::kwargs kwargs) {
c10::optional<c10::DispatchKey> dk =
c10::make_optional(dk_);
c10::make_optional(c10::parseDispatchKey(str_dk));
ToIValueAllowNumbersAsTensors g(allow_numbers_as_tensors);
return _get_operation_for_overload_or_packet(
{op}, symbol, args, kwargs, /*is_overload*/ true, dk);