/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_;
|