FM / python

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/python/

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/python

Name	Size	Mode	Actions
dlpack.h	4693	0644	edit dl rm
pybind_state.h	15621	0644	edit dl rm
pybind_state_dlpack.h	4143	0644	edit dl rm
pybind_state_registry.h	1021	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/python/pybind_state_registry.h (1021B)

#pragma once

#include <pybind11/pybind11.h>
#include "c10/util/Registry.h"

namespace caffe2 {
namespace python {

namespace py = pybind11;

struct PybindAddition {
  PybindAddition() {}
  PybindAddition(py::module&) {}
  virtual ~PybindAddition(){};
};

C10_DECLARE_REGISTRY(PybindAdditionRegistry, PybindAddition, py::module&);

#define REGISTER_PYBIND_ADDITION(funcname)                                    \
  namespace {                                                                 \
  struct funcname##Impl : public PybindAddition {                             \
    funcname##Impl(py::module& m) {                                           \
      funcname(m);                                                            \
    }                                                                         \
  };                                                                          \
  C10_REGISTER_CLASS(PybindAdditionRegistry, funcname##Impl, funcname##Impl); \
  }

} // namespace python
} // namespace caffe2

#pragma once #include #include "c10/util/Registry.h" namespace caffe2 { namespace python { namespace py = pybind11; struct PybindAddition { PybindAddition() {} PybindAddition(py::module&) {} virtual ~PybindAddition(){}; }; C10_DECLARE_REGISTRY(PybindAdditionRegistry, PybindAddition, py::module&); #define REGISTER_PYBIND_ADDITION(funcname) \ namespace { \ struct funcname##Impl : public PybindAddition { \ funcname##Impl(py::module& m) { \ funcname(m); \ } \ }; \ C10_REGISTER_CLASS(PybindAdditionRegistry, funcname##Impl, funcname##Impl); \ } } // namespace python } // namespace caffe2