mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
5dbca58bd0
Another attempt at merging https://github.com/pytorch/pytorch/pull/164597 due to CLA signing failure. Differential Revision: [D84397377](https://our.internmc.facebook.com/intern/diff/D84397377) Pull Request resolved: https://github.com/pytorch/pytorch/pull/165200 Approved by: https://github.com/anijain2305, https://github.com/mlazos
41 lines
920 B
C
41 lines
920 B
C
#pragma once
|
|
|
|
#include <torch/csrc/utils/python_compat.h>
|
|
|
|
// Functions that need to be copied from the CPython source
|
|
// should go in cpython_defs.c. Copying is required when, e.g.,
|
|
// we need to call internal CPython functions that are not exposed.
|
|
|
|
#if IS_PYTHON_3_11_PLUS
|
|
|
|
typedef struct _PyInterpreterFrame _PyInterpreterFrame;
|
|
|
|
PyFunctionObject* _PyFunction_CopyWithNewCode(
|
|
PyFunctionObject* o,
|
|
PyCodeObject* code);
|
|
|
|
void THP_PyFrame_Clear(_PyInterpreterFrame* frame);
|
|
|
|
_PyInterpreterFrame* THP_PyThreadState_BumpFramePointerSlow(
|
|
PyThreadState* tstate,
|
|
size_t size);
|
|
|
|
void THP_PyThreadState_PopFrame(
|
|
PyThreadState* tstate,
|
|
_PyInterpreterFrame* frame);
|
|
|
|
#endif
|
|
|
|
// pointers to _PyOpcode_Caches for C++
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
extern const uint8_t* THP_PyOpcode_Caches;
|
|
extern int THP_PyOpcode_Caches_size;
|
|
void init_THPCaches();
|
|
|
|
#ifdef __cplusplus
|
|
} // extern "C"
|
|
#endif
|