mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
b140ed6848
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/35625 Python 2 has reached end-of-life and is no longer supported by PyTorch. This function was already ifdef'ed out in Python 2. Added a comment about when we might be able to remove this entire file. Test Plan: CI Differential Revision: D20842885 Pulled By: dreiss fbshipit-source-id: 1fd3b1b2ff5a82caaf3bc11344dde2941427cfc0
72 lines
2.1 KiB
C++
72 lines
2.1 KiB
C++
/* Copyright Python Software Foundation
|
|
*
|
|
* This file is copy-pasted from CPython source code with modifications:
|
|
* https://github.com/python/cpython/blob/master/Objects/structseq.c
|
|
* https://github.com/python/cpython/blob/2.7/Objects/structseq.c
|
|
*
|
|
* The purpose of this file is to overwrite the default behavior
|
|
* of repr of structseq to provide better printting for returned
|
|
* structseq objects from operators, aka torch.return_types.*
|
|
*
|
|
* For more information on copyright of CPython, see:
|
|
* https://github.com/python/cpython#copyright-and-license-information
|
|
*/
|
|
|
|
#include <torch/csrc/utils/structseq.h>
|
|
#include <torch/csrc/utils/six.h>
|
|
#include <structmember.h>
|
|
#include <sstream>
|
|
|
|
namespace torch {
|
|
namespace utils {
|
|
|
|
// NOTE: The built-in repr method from PyStructSequence was updated in
|
|
// https://github.com/python/cpython/commit/c70ab02df2894c34da2223fc3798c0404b41fd79
|
|
// so this function might not be required in Python 3.8+.
|
|
PyObject *returned_structseq_repr(PyStructSequence *obj) {
|
|
PyTypeObject *typ = Py_TYPE(obj);
|
|
THPObjectPtr tup = six::maybeAsTuple(obj);
|
|
if (tup == nullptr) {
|
|
return nullptr;
|
|
}
|
|
|
|
std::stringstream ss;
|
|
ss << typ->tp_name << "(\n";
|
|
Py_ssize_t num_elements = Py_SIZE(obj);
|
|
|
|
for (Py_ssize_t i = 0; i < num_elements; i++) {
|
|
const char *cname = typ->tp_members[i].name;
|
|
if (cname == nullptr) {
|
|
PyErr_Format(PyExc_SystemError, "In structseq_repr(), member %zd name is nullptr"
|
|
" for type %.500s", i, typ->tp_name);
|
|
return nullptr;
|
|
}
|
|
|
|
PyObject* val = PyTuple_GetItem(tup.get(), i);
|
|
if (val == nullptr) {
|
|
return nullptr;
|
|
}
|
|
|
|
auto repr = THPObjectPtr(PyObject_Repr(val));
|
|
if (repr == nullptr) {
|
|
return nullptr;
|
|
}
|
|
|
|
const char* crepr = PyUnicode_AsUTF8(repr);
|
|
if (crepr == nullptr) {
|
|
return nullptr;
|
|
}
|
|
|
|
ss << cname << '=' << crepr;
|
|
if (i < num_elements - 1) {
|
|
ss << ",\n";
|
|
}
|
|
}
|
|
ss << ")";
|
|
|
|
return PyUnicode_FromString(ss.str().c_str());
|
|
}
|
|
|
|
}
|
|
}
|