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

PyLong_{As/From}{Long/UnsignedLong} lint checks (#49280)

Browse Source 
Summary:
Fixes https://github.com/pytorch/pytorch/issues/45581

Pull Request resolved: https://github.com/pytorch/pytorch/pull/49280

Reviewed By: mruberry

Differential Revision: D25592330

Pulled By: ezyang

fbshipit-source-id: 5c16d6aed88ad1feaa7f129b4cd44c0561be2de2
This commit is contained in:
peterjc123
2020-12-17 09:30:02 -08:00
committed by Facebook GitHub Bot
parent c20b916cbd
commit 815d38395a
9 changed files with 84 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/csrc/autograd/python_cpp_function.cpp
View File

@ -11,6 +11,7 @@
#include <torch/csrc/autograd/python_hook.h>
#include <torch/csrc/autograd/python_anomaly_mode.h>
#include <pybind11/pybind11.h>
#include <torch/csrc/utils/python_numbers.h>
#include <torch/csrc/utils/python_strings.h>
#include <torch/csrc/DynamicTypes.h>
#include <torch/csrc/Exceptions.h>
@ -114,7 +115,7 @@ PyObject* THPCppFunction_next_functions(THPCppFunction* self, PyObject* hook)
PyObject *py_fn = functionToPyObject(c_tuple.function);
if (!py_fn) return nullptr;
PyTuple_SET_ITEM(tuple.get(), 0, py_fn);
PyObject *py_idx = PyLong_FromLong(c_tuple.input_nr);
PyObject *py_idx = THPUtils_packUInt32(c_tuple.input_nr);
if (!py_idx) return nullptr;
PyTuple_SET_ITEM(tuple.get(), 1, py_idx);
PyTuple_SET_ITEM(py_functions.get(), i, tuple.release());