torch/csrc/jit/pass_manager.cpp

#include <torch/csrc/jit/pass_manager.h>

namespace torch {
namespace jit {

std::vector<Pass>& getCustomPostFusionPasses() {
  static std::vector<Pass> passes;
  return passes;
}

std::vector<Pass>& getCustomPreFusionPasses() {
  static std::vector<Pass> passes;
  return passes;
}

RegisterPostFusionPass::RegisterPostFusionPass(Pass p) {
  getCustomPostFusionPasses().emplace_back(std::move(p));
}

RegisterPreFusionPass::RegisterPreFusionPass(Pass p) {
  getCustomPreFusionPasses().emplace_back(std::move(p));
}

} // namespace jit
} // namespace torch