pybind11: local Shape

AMYPAD · Mar 14, 2024 · bf7b524 · bf7b524
1 parent 21edd59
commit bf7b524
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cuvec/src/pybind11.cu b/cuvec/src/pybind11.cu
@@ -9,7 +9,7 @@
 
 PYBIND11_MODULE(cuvec_pybind11, m) {
   m.doc() = "PyBind11 external module.";
-  pybind11::bind_vector<std::vector<size_t>>(m, "Shape");
+  pybind11::bind_vector<std::vector<size_t>>(m, "Shape", pybind11::module_local(true));
   pybind11::implicitly_convertible<pybind11::tuple, std::vector<size_t>>();
   PYBIND11_BIND_NDCUVEC(signed char, b);
   PYBIND11_BIND_NDCUVEC(unsigned char, B);