diff --git a/core/main/py_wrapper.cc b/core/main/py_wrapper.cc index fe9730d..08697ae 100644 --- a/core/main/py_wrapper.cc +++ b/core/main/py_wrapper.cc @@ -114,14 +114,10 @@ PYBIND11_MODULE(_pywrap_tn, m) { return py::reinterpret_steal(obj); }) - .def("create_sparse_table", [](py::object obj, std::string name, int dimension, bool use_cvm) { + .def("create_sparse_table", [](py::object obj, std::string name, int dimension) { OptimizerBase* opt = static_cast(PyCapsule_GetPointer(obj.ptr(), nullptr)); - opt->SetUseCvm(use_cvm); - - std::cout << "Cvm plugin is: " << opt->ShouldUseCvm() << std::endl; - PsCluster* cluster = PsCluster::Instance(); SparseTable* table = CreateSparseTable(opt, name, dimension, cluster->RankNum(), cluster->Rank());