mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
9c49bb9ddf
Summary: This does 6 things: - add c10/util/Registry.h as the unified registry util - cleaned up some APIs such as export condition - fully remove aten/core/registry.h - fully remove caffe2/core/registry.h - remove a bogus aten/registry.h - unifying all macros - set up registry testing in c10 Also, an important note that we used to mark the templated Registry class as EXPORT - this should not happen, because one should almost never export a template class. This PR fixes that. Pull Request resolved: https://github.com/pytorch/pytorch/pull/12077 Reviewed By: ezyang Differential Revision: D10050771 Pulled By: Yangqing fbshipit-source-id: 417b249b49fed6a67956e7c6b6d22374bcee24cf
13 lines
274 B
C++
13 lines
274 B
C++
#include "caffe2/opt/passes.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
C10_DEFINE_REGISTRY(
|
|
WorkspaceOptimizationPassRegistry,
|
|
WorkspaceOptimizationPass,
|
|
NNModule*,
|
|
Workspace*);
|
|
C10_DEFINE_REGISTRY(OptimizationPassRegistry, OptimizationPass, NNModule*);
|
|
|
|
} // namespace caffe2
|