mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
8967d55b01
Follows #132753 Pull Request resolved: https://github.com/pytorch/pytorch/pull/132963 Approved by: https://github.com/Skylion007
27 lines
638 B
C++
27 lines
638 B
C++
#pragma once
|
|
|
|
#include <vector>
|
|
|
|
#include <torch/csrc/jit/mobile/code.h>
|
|
#include <torch/csrc/jit/mobile/frame.h>
|
|
|
|
namespace torch::jit::mobile {
|
|
|
|
struct InterpreterState {
|
|
TORCH_API explicit InterpreterState(const Code& code);
|
|
TORCH_API bool run(Stack& stack);
|
|
|
|
private:
|
|
void enterFrame(const Code&);
|
|
void leaveFrame();
|
|
void saveExceptionDebugHandles();
|
|
void callFunction(torch::jit::Function& f, Stack& stack);
|
|
|
|
c10::IValue& reg(size_t reg);
|
|
std::vector<c10::IValue> registers_;
|
|
std::vector<Frame> frames_;
|
|
};
|
|
|
|
const std::vector<DebugHandle>& getInterpretersExceptionDebugHandles();
|
|
} // namespace torch::jit::mobile
|