diff --git a/cuvec/src/pybind11.cu b/cuvec/src/pybind11.cu index 064c30b..f8e7fc2 100644 --- 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>(m, "Shape"); + pybind11::bind_vector>(m, "Shape", pybind11::module_local(true)); pybind11::implicitly_convertible>(); PYBIND11_BIND_NDCUVEC(signed char, b); PYBIND11_BIND_NDCUVEC(unsigned char, B);