Skip to content

Commit 3f4a2ee

Browse files
author
jiangxinglei
committed
remove test code
1 parent 7fd2871 commit 3f4a2ee

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

core/main/py_wrapper.cc

+1-5
Original file line numberDiff line numberDiff line change
@@ -114,14 +114,10 @@ PYBIND11_MODULE(_pywrap_tn, m) {
114114

115115
return py::reinterpret_steal<py::object>(obj);
116116
})
117-
.def("create_sparse_table", [](py::object obj, std::string name, int dimension, bool use_cvm) {
117+
.def("create_sparse_table", [](py::object obj, std::string name, int dimension) {
118118
OptimizerBase* opt =
119119
static_cast<OptimizerBase*>(PyCapsule_GetPointer(obj.ptr(), nullptr));
120120

121-
opt->SetUseCvm(use_cvm);
122-
123-
std::cout << "Cvm plugin is: " << opt->ShouldUseCvm() << std::endl;
124-
125121
PsCluster* cluster = PsCluster::Instance();
126122

127123
SparseTable* table = CreateSparseTable(opt, name, dimension, cluster->RankNum(), cluster->Rank());

0 commit comments

Comments
 (0)