jit.h

#pragma once

#include <torch/csrc/jit/script/compiler.h>
#include <torch/csrc/jit/script/module.h>
#include <ATen/core/stack.h>
#include <torch/csrc/WindowsTorchApiMacro.h>

#include <string>
#include <memory>

namespace torch {
namespace jit {

TORCH_API std::shared_ptr<script::Module> compile(const std::string& source);

} // namespace jit
} // namespace torch