frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-06 00:54:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Exception] [6/N] Remove use of torch::TypeError (#117964)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/117964
Approved by: https://github.com/albanD
This commit is contained in:
cyy
2024-01-25 03:35:53 +00:00
committed by PyTorch MergeBot
parent 67300a11cb
commit 87335fabae
12 changed files with 191 additions and 170 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
torch/csrc/utils/disable_torch_function.cpp
View File

@ -209,8 +209,8 @@ PyObject* THPModule_disable_torch_function(PyObject* self, PyObject* a) {
} else if (PyTuple_Check(args)) {
py_args = py::reinterpret_borrow<py::tuple>(args);
} else {
throw torch::TypeError(
"expected List or Tuple (got %s)", Py_TYPE(args)->tp_name);
TORCH_CHECK_TYPE(
false, "expected List or Tuple (got ", Py_TYPE(args)->tp_name, ")");
}
// These are all C-API calls so no exceptions will be raised
@ -243,8 +243,8 @@ PyObject* THPModule_disable_torch_dispatch(PyObject* self, PyObject* a) {
} else if (PyTuple_Check(args)) {
py_args = py::reinterpret_borrow<py::tuple>(args);
} else {
throw torch::TypeError(
"expected List or Tuple (got %s)", Py_TYPE(args)->tp_name);
TORCH_CHECK_TYPE(
false, "expected List or Tuple (got ", Py_TYPE(args)->tp_name, ")");
}
// This implementation is not completely correct. The moral