blob: 7dea678e5b1f4cb313f653449091a65e720acde6 [file] [log] [blame]
#pragma once
#include "torch/csrc/python_headers.h"
#include "torch/csrc/Exceptions.h"
#include "torch/csrc/utils/object_ptr.h"
#include "torch/csrc/utils/python_numbers.h"
inline void THPUtils_packInt64Array(PyObject *tuple, size_t size, const int64_t *sizes) {
for (size_t i = 0; i != size; ++i) {
PyObject *i64 = THPUtils_packInt64(sizes[i]);
if (!i64) {
throw python_error();
}
PyTuple_SET_ITEM(tuple, i, THPUtils_packInt64(sizes[i]));
}
}
inline PyObject* THPUtils_packInt64Array(size_t size, const int64_t *sizes) {
THPObjectPtr tuple(PyTuple_New(size));
if (!tuple) throw python_error();
THPUtils_packInt64Array(tuple.get(), size, sizes);
return tuple.release();
}