| /aosp_15_r20/external/pytorch/c10/core/impl/ |
| H A D | PyObjectSlot.h | 28 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 D | PyObjectSlot.cpp | 27 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 D | GPUTrace.h | 9 static std::atomic<const PyInterpreter*> gpuTraceState;
19 static void set_trace(const PyInterpreter*);
21 static const PyInterpreter* get_trace() { in get_trace()
|
| H A D | PythonDispatcherTLS.h | 9 static void set_state(PyInterpreter* state);
10 static PyInterpreter* get_state();
21 PyInterpreter* old_;
|
| H A D | PythonDispatcherTLS.cpp | 7 thread_local PyInterpreter* pythonDispatcherState;
9 void PythonDispatcherTLS::set_state(PyInterpreter* state) { in set_state()
19 PyInterpreter* PythonDispatcherTLS::get_state() { in get_state()
|
| H A D | PyInterpreter.h | 31 struct C10_API PyInterpreter; variable
220 struct C10_API PyInterpreter { struct
223 PyInterpreter(const PyInterpreterVTable* vtable) : vtable_(vtable){}; in PyInterpreter() argument
|
| H A D | GPUTrace.cpp | 6 std::atomic<const PyInterpreter*> GPUTrace::gpuTraceState{nullptr};
10 void GPUTrace::set_trace(const PyInterpreter* trace) { in set_trace()
|
| H A D | PyInterpreter.cpp | 143 void PyInterpreter::disarm() noexcept { in disarm()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | SafePyObject.h | 24 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 D | PyHandleCache.h | 48 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 D | SafePyObject.cpp | 5 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 D | PythonOpRegistrationTrampoline.cpp | 11 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 D | PythonOpRegistrationTrampoline.h | 11 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 D | XPUEvent.h | 27 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 D | CUDAEvent.h | 48 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 D | CUDAGuardImpl.h | 99 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 D | XPUGuardImpl.h | 78 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 D | CUDAFunctions.h | 87 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 D | XPUStream.cpp | 112 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 D | XPUStream.h | 100 const c10::impl::PyInterpreter* interp = c10::impl::GPUTrace::get_trace(); in synchronize()
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | PyInterpreter.cpp | 157 : 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 D | PyInterpreter.h | 12 TORCH_PYTHON_API c10::impl::PyInterpreter* getPyInterpreter();
|
| /aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
| H A D | python_variable_simple.h | 49 TORCH_PYTHON_API c10::impl::PyInterpreter* getPyInterpreter();
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | ThreadLocalState.h | 68 c10::impl::PyInterpreter* python_dispatcher_state_;
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/ |
| H A D | kernel_holder.h | 64 c10::impl::PyInterpreter* pyinterpreter_;
|