python_benchmark_messages.cc 1.2 KB

1234567891011121314151617181920212223242526272829303132
  1. #include <Python.h>
  2. #include "benchmarks.pb.h"
  3. #include "datasets/google_message1/proto2/benchmark_message1_proto2.pb.h"
  4. #include "datasets/google_message1/proto3/benchmark_message1_proto3.pb.h"
  5. #include "datasets/google_message2/benchmark_message2.pb.h"
  6. #include "datasets/google_message3/benchmark_message3.pb.h"
  7. #include "datasets/google_message4/benchmark_message4.pb.h"
  8. static struct PyModuleDef _module = {PyModuleDef_HEAD_INIT,
  9. "libbenchmark_messages",
  10. "Benchmark messages Python module",
  11. -1,
  12. NULL,
  13. NULL,
  14. NULL,
  15. NULL,
  16. NULL};
  17. extern "C" {
  18. PyMODINIT_FUNC
  19. PyInit_libbenchmark_messages() {
  20. benchmarks::BenchmarkDataset().descriptor();
  21. benchmarks::proto3::GoogleMessage1().descriptor();
  22. benchmarks::proto2::GoogleMessage1().descriptor();
  23. benchmarks::proto2::GoogleMessage2().descriptor();
  24. benchmarks::google_message3::GoogleMessage3().descriptor();
  25. benchmarks::google_message4::GoogleMessage4().descriptor();
  26. return PyModule_Create(&_module);
  27. }
  28. }