#include <torch/csrc/jit/pass_manager.h> | |
namespace torch { | |
namespace jit { | |
std::vector<Pass>& getCustomPasses() { | |
static std::vector<Pass> passes; | |
return passes; | |
} | |
RegisterPass::RegisterPass(Pass p) { | |
getCustomPasses().emplace_back(std::move(p)); | |
} | |
} // namespace jit | |
} // namespace torch |