module.h

#ifndef CAFFE2_CORE_MODULE_H_
#define CAFFE2_CORE_MODULE_H_

#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <functional>
#include <memory>
#include <mutex>

#include "caffe2/core/common.h"
#include <c10/util/typeid.h>

namespace caffe2 {

class CAFFE2_API ModuleSchema {
 public:
  ModuleSchema(const char* name, const char* description);

 private:
  const char* name_;
  const char* description_;
};


CAFFE2_API const CaffeMap<string, const ModuleSchema*>& CurrentModules();

CAFFE2_API bool HasModule(const string& name);

CAFFE2_API void LoadModule(const string& name, const string& filename="");


#define CAFFE2_MODULE(name, description)                                    \
  extern "C" {                                                              \
    bool gCaffe2ModuleSanityCheck##name() { return true; }                  \
  }                                                                         \
  namespace {                                                               \
    static ::caffe2::ModuleSchema module_schema_##name(#name, description); \
  }

}  // namespace caffe2
#endif  // CAFFE2_CORE_MODULE_H_