hooks_for_testing.h

#pragma once
#include <torch/csrc/WindowsTorchApiMacro.h>
#include <functional>
#include <memory>

namespace torch {
namespace jit {
namespace script {
struct Module;
}
TORCH_API void didFinishEmitModule(std::shared_ptr<script::Module> module);
TORCH_API void setEmitModuleHook(
    std::function<void(std::shared_ptr<script::Module> module)> cb);
} // namespace jit
} // namespace torch