mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
b1539412db
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18587 ghimport-source-id: 80d753f7046a2a719e0c076684f44fa2059a0921 Differential Revision: D14901227 Pulled By: bwasti fbshipit-source-id: 56511d0313419b63945a36b80e9ea51abdef2bd4
16 lines
299 B
C++
16 lines
299 B
C++
#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
|