mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
3a5bf0bc36
Fixes #144302 After checking the code of my third-party devices, I think these APIs are also relied on by us, so I exposed them according to the discussion in the issue. Pull Request resolved: https://github.com/pytorch/pytorch/pull/144746 Approved by: https://github.com/albanD
61 lines
1.5 KiB
C
61 lines
1.5 KiB
C
#ifndef THP_STORAGE_INC
|
|
#define THP_STORAGE_INC
|
|
|
|
#include <Python.h>
|
|
#include <c10/core/Storage.h>
|
|
#include <torch/csrc/Exceptions.h>
|
|
#include <torch/csrc/Export.h>
|
|
#include <torch/csrc/Types.h>
|
|
|
|
#define THPStorageStr "torch.UntypedStorage"
|
|
|
|
struct THPStorage {
|
|
PyObject_HEAD
|
|
c10::MaybeOwned<c10::Storage> cdata;
|
|
bool is_hermetic;
|
|
};
|
|
|
|
TORCH_PYTHON_API PyObject* THPStorage_Wrap(c10::Storage storage);
|
|
TORCH_PYTHON_API PyObject* THPStorage_NewWithStorage(
|
|
PyTypeObject* type,
|
|
c10::Storage _storage,
|
|
c10::impl::PyInterpreterStatus status,
|
|
bool allow_preexisting_pyobj = false);
|
|
TORCH_PYTHON_API extern PyTypeObject* THPStorageClass;
|
|
|
|
inline bool THPStorage_CheckTypeExact(PyTypeObject* tp) {
|
|
return tp == THPStorageClass;
|
|
}
|
|
|
|
inline bool THPStorage_CheckExact(PyObject* obj) {
|
|
return THPStorage_CheckTypeExact(Py_TYPE(obj));
|
|
}
|
|
|
|
inline bool THPStorage_Check(PyObject* obj) {
|
|
if (!THPStorageClass)
|
|
return false;
|
|
|
|
const auto result = PyObject_IsInstance(obj, (PyObject*)THPStorageClass);
|
|
if (result == -1)
|
|
throw python_error();
|
|
return result;
|
|
}
|
|
|
|
bool THPStorage_init(PyObject* module);
|
|
void THPStorage_postInit(PyObject* module);
|
|
|
|
void THPStorage_assertNotNull(THPStorage* storage);
|
|
TORCH_PYTHON_API void THPStorage_assertNotNull(PyObject* obj);
|
|
|
|
TORCH_PYTHON_API extern PyTypeObject THPStorageType;
|
|
|
|
inline const c10::Storage& THPStorage_Unpack(THPStorage* storage) {
|
|
return *storage->cdata;
|
|
}
|
|
|
|
inline const c10::Storage& THPStorage_Unpack(PyObject* obj) {
|
|
return THPStorage_Unpack(reinterpret_cast<THPStorage*>(obj));
|
|
}
|
|
|
|
#endif
|