Skip to content
This repository was archived by the owner on Nov 1, 2024. It is now read-only.

Update reference to torch::TypeError #72

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Update reference to torch::TypeError #72

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ec39610
Update reference to torch::TypeError
mehtanirav Jan 4, 2023
82a32b8
formatting fix
mehtanirav Jan 4, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 2 additions & 5 deletions src/python/torchdistx/_C/deferred_init.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,6 @@ using at::Tensor;

using c10::impl::PyInterpreterStatus;

using torch::TypeError;

} // namespace torchdistx

namespace torchdistx::python {
Expand Down Expand Up @@ -60,9 +58,8 @@ py::object makeVariable(PyTypeObject* type, Tensor data) {
py::object materializeVariable(const py::object& var) {
PyObject* naked_var = var.ptr();

if (!THPVariable_Check(naked_var)) {
throw TypeError{"`var` has to be a `Variable`, but got `%s`.", Py_TYPE(naked_var)->tp_name};
}
TORCH_CHECK_TYPE(THPVariable_Check(naked_var), "`var` has to be a `Variable`, but got `%s`.",
Py_TYPE(naked_var)->tp_name);

const Tensor& data = THPVariable_Unpack(naked_var);

Expand Down