Home
last modified time | relevance | path

Searched refs:PyInterpreter (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DPyObjectSlot.h28 PyInterpreter* self_interpreter, in init_pyobj()
31 impl::PyInterpreter* expected = nullptr; in init_pyobj()
78 PyInterpreter* pyobj_interpreter();
98 PyInterpreter* self_interpreter,
101 impl::PyInterpreter* interpreter =
129 void unchecked_clear_pyobj(PyInterpreter* interpreter);
131 PyInterpreter& load_pyobj_interpreter() const;
135 bool check_interpreter(PyInterpreter* interpreter);
167 std::atomic<PyInterpreter*> pyobj_interpreter_;
H A DPyObjectSlot.cpp27 PyInterpreter* PyObjectSlot::pyobj_interpreter() { in pyobj_interpreter()
37 void PyObjectSlot::unchecked_clear_pyobj(PyInterpreter* interpreter) { in unchecked_clear_pyobj()
42 PyInterpreter& PyObjectSlot::load_pyobj_interpreter() const { in load_pyobj_interpreter()
53 bool PyObjectSlot::check_interpreter(PyInterpreter* interpreter) { in check_interpreter()
H A DGPUTrace.h9 static std::atomic<const PyInterpreter*> gpuTraceState;
19 static void set_trace(const PyInterpreter*);
21 static const PyInterpreter* get_trace() { in get_trace()
H A DPythonDispatcherTLS.h9 static void set_state(PyInterpreter* state);
10 static PyInterpreter* get_state();
21 PyInterpreter* old_;
H A DPythonDispatcherTLS.cpp7 thread_local PyInterpreter* pythonDispatcherState;
9 void PythonDispatcherTLS::set_state(PyInterpreter* state) { in set_state()
19 PyInterpreter* PythonDispatcherTLS::get_state() { in get_state()
H A DPyInterpreter.h31 struct C10_API PyInterpreter; variable
220 struct C10_API PyInterpreter { struct
223 PyInterpreter(const PyInterpreterVTable* vtable) : vtable_(vtable){}; in PyInterpreter() argument
H A DGPUTrace.cpp6 std::atomic<const PyInterpreter*> GPUTrace::gpuTraceState{nullptr};
10 void GPUTrace::set_trace(const PyInterpreter* trace) { in set_trace()
H A DPyInterpreter.cpp143 void PyInterpreter::disarm() noexcept { in disarm()
/aosp_15_r20/external/pytorch/c10/core/
H A DSafePyObject.h24 SafePyObject(PyObject* data, c10::impl::PyInterpreter* pyinterpreter) in SafePyObject()
60 c10::impl::PyInterpreter& pyinterpreter() const { in pyinterpreter()
63 PyObject* ptr(const c10::impl::PyInterpreter*) const;
74 c10::impl::PyInterpreter* pyinterpreter_;
82 SafePyObjectT(PyObject* data, c10::impl::PyInterpreter* pyinterpreter) in SafePyObjectT()
98 SafePyHandle(PyObject* data, c10::impl::PyInterpreter* pyinterpreter) in SafePyHandle()
101 c10::impl::PyInterpreter& pyinterpreter() const { in pyinterpreter()
104 PyObject* ptr(const c10::impl::PyInterpreter*) const;
115 c10::impl::PyInterpreter* pyinterpreter_;
H A DPyHandleCache.h48 PyObject* ptr_or(impl::PyInterpreter* self_interpreter, F slow_accessor) in ptr_or()
51 impl::PyInterpreter* interpreter = in ptr_or()
57 impl::PyInterpreter* expected = nullptr; in ptr_or()
72 mutable std::atomic<impl::PyInterpreter*> pyinterpreter_;
H A DSafePyObject.cpp5 PyObject* SafePyObject::ptr(const c10::impl::PyInterpreter* interpreter) const { in ptr()
10 PyObject* SafePyHandle::ptr(const c10::impl::PyInterpreter* interpreter) const { in ptr()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DPythonOpRegistrationTrampoline.cpp11 std::atomic<c10::impl::PyInterpreter*>
14 c10::impl::PyInterpreter* PythonOpRegistrationTrampoline::getInterpreter() { in getInterpreter()
19 c10::impl::PyInterpreter* interp) { in registerInterpreter()
20 c10::impl::PyInterpreter* expected = nullptr; in registerInterpreter()
H A DPythonOpRegistrationTrampoline.h11 static std::atomic<c10::impl::PyInterpreter*> interpreter_;
16 static bool registerInterpreter(c10::impl::PyInterpreter*);
19 static c10::impl::PyInterpreter* getInterpreter();
/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/
H A DXPUEvent.h27 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in ~XPUEvent()
90 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in record()
107 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in record()
121 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in block()
149 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in synchronize()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDAEvent.h48 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in ~CUDAEvent()
122 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in record()
138 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in block()
165 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in synchronize()
195 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in createEvent()
/aosp_15_r20/external/pytorch/c10/cuda/impl/
H A DCUDAGuardImpl.h99 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in createEvent()
114 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in destroyEvent()
149 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in record()
172 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in block()
213 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in synchronizeEvent()
/aosp_15_r20/external/pytorch/c10/xpu/impl/
H A DXPUGuardImpl.h78 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in destroyEvent()
109 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in record()
125 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in block()
158 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in synchronizeEvent()
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAFunctions.h87 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in memcpy_and_sync()
105 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in stream_synchronize()
/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUStream.cpp112 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in initDeviceStreamState()
289 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in syncStreamsOnDevice()
H A DXPUStream.h100 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in synchronize()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DPyInterpreter.cpp157 : impl_(new c10::impl::PyInterpreter( in PyInterpreterHolder()
168 c10::impl::PyInterpreter* get() const noexcept { in get()
176 c10::impl::PyInterpreter* impl_;
982 c10::impl::PyInterpreter* getPyInterpreter() { in getPyInterpreter()
H A DPyInterpreter.h12 TORCH_PYTHON_API c10::impl::PyInterpreter* getPyInterpreter();
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Dpython_variable_simple.h49 TORCH_PYTHON_API c10::impl::PyInterpreter* getPyInterpreter();
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DThreadLocalState.h68 c10::impl::PyInterpreter* python_dispatcher_state_;
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/
H A Dkernel_holder.h64 c10::impl::PyInterpreter* pyinterpreter_;

12