mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01: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
17 lines
299 B
C++
17 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
|