Use more sensible name

because we're not only making MNIST modules!
microsoft · Nov 19, 2019 · c46cd17 · c46cd17
1 parent 5cc496d
commit c46cd17
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/test/builds/build_pybind11_module.sh b/test/builds/build_pybind11_module.sh
@@ -28,6 +28,6 @@ build_pybind11_module() {
 	  -shared \
 	  -fPIC \
 	  -o $MODULE_FILE \
-	  -DMNISTCNNCPP_MODULE_NAME=$MODULE_NAME \
+	  -DPYTHON_MODULE_NAME=$MODULE_NAME \
 	  $CPP_FILE
 }
diff --git a/test/ksc/adbench-lstmpy.cpp b/test/ksc/adbench-lstmpy.cpp
@@ -36,7 +36,7 @@ void declare_vec(py::module &m, std::string typestr) {
 // possible because Python tuples map to std::tuples regardless of
 // their contents.  I'll look into it later.  For now I'll just have a
 // bunch of verbose replication.
-PYBIND11_MODULE(MNISTCNNCPP_MODULE_NAME, m) {
+PYBIND11_MODULE(PYTHON_MODULE_NAME, m) {
   declare_vec<double>(m, std::string("double"));
   declare_vec<std::tuple<ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>>>(m, std::string("tuple_vec10"));
   declare_vec<std::tuple<ks::vec<double>, ks::vec<double>>>(m, std::string("tuple_vec2"));

diff --git a/test/ksc/gmmpy.cpp b/test/ksc/gmmpy.cpp
@@ -34,7 +34,7 @@ void declare_vec(py::module &m, std::string typestr) {
 // possible because Python tuples map to std::tuples regardless of
 // their contents.  I'll look into it later.  For now I'll just have a
 // bunch of verbose replication.
-PYBIND11_MODULE(MNISTCNNCPP_MODULE_NAME, m) {
+PYBIND11_MODULE(PYTHON_MODULE_NAME, m) {
   declare_vec<double>(m, std::string("double"));
   declare_vec<std::tuple<ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>>>(m, std::string("tuple_vec10"));
   declare_vec<std::tuple<ks::vec<double>, ks::vec<double>>>(m, std::string("tuple_vec2"));

diff --git a/test/ksc/mnistcnnpy.cpp b/test/ksc/mnistcnnpy.cpp
@@ -24,7 +24,7 @@ void declare_vec(py::module &m, std::string typestr) {
 
 // In the future it might make more sense to move the vec type
 // definitions to a general Knossos CPP types Python module.
-PYBIND11_MODULE(MNISTCNNCPP_MODULE_NAME, m) {
+PYBIND11_MODULE(PYTHON_MODULE_NAME, m) {
   declare_vec<double>(m, std::string("double"));
   declare_vec<ks::vec<double> >(m, std::string("vec_double"));
   declare_vec<ks::vec<ks::vec<double> > >(m, std::string("vec_vec_double"));