fix memory option on interfaces

Xtra-Computing · Oct 28, 2018 · a824cce · a824cce
1 parent 235979e
commit a824cce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/thundersvm/svm_interface_api.cpp b/src/thundersvm/svm_interface_api.cpp
@@ -140,7 +140,7 @@ extern "C" {
         CUDA_CHECK(cudaSetDevice(parser.gpu_id));
 #endif
 
-        model->set_max_memory_size(parser.param_cmd.max_mem_size);
+        model->set_max_memory_size_Byte(parser.param_cmd.max_mem_size);
         model->load_from_file(model_file_path);
         file.close();
         file.open(output_file_path, fstream::out);