30 lines
963 B
C++
30 lines
963 B
C++
#include <Python.h>
|
|
|
|
#include "benchmarks.pb.h"
|
|
#include "datasets/google_message1/benchmark_message1_proto2.pb.h"
|
|
#include "datasets/google_message1/benchmark_message1_proto3.pb.h"
|
|
#include "datasets/google_message2/benchmark_message2.pb.h"
|
|
#include "datasets/google_message3/benchmark_message3.pb.h"
|
|
#include "datasets/google_message4/benchmark_message4.pb.h"
|
|
|
|
static PyMethodDef python_benchmark_methods[] = {
|
|
{NULL, NULL, 0, NULL} /* Sentinel */
|
|
};
|
|
|
|
|
|
PyMODINIT_FUNC
|
|
initlibbenchmark_messages() {
|
|
benchmarks::BenchmarkDataset().descriptor();
|
|
benchmarks::proto3::GoogleMessage1().descriptor();
|
|
benchmarks::proto2::GoogleMessage1().descriptor();
|
|
benchmarks::proto2::GoogleMessage2().descriptor();
|
|
benchmarks::google_message3::GoogleMessage3().descriptor();
|
|
benchmarks::google_message4::GoogleMessage4().descriptor();
|
|
|
|
PyObject *m;
|
|
|
|
m = Py_InitModule("libbenchmark_messages", python_benchmark_methods);
|
|
if (m == NULL)
|
|
return;
|
|
}
|