Simplify invocation a bit

browsermt · Aug 2, 2023 · 3d37edf · 3d37edf
1 parent 1e80e79
commit 3d37edf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/bindings/python/translator.py b/bindings/python/translator.py
@@ -125,9 +125,6 @@ def main():
         num_gpus = []
     else:
         num_gpus = args.num_gpus[0]
-        print(num_gpus)
-        print(type(num_gpus))
-        print(args.num_workers)
     translator = Translator(args.config, args.num_workers, num_gpus, args.cache_size, args.logging, args.terminology_tsv, args.force_terminology, args.terminology_form)
 
     if args.path_to_input is None:

diff --git a/src/translator/service.cpp b/src/translator/service.cpp
@@ -158,15 +158,16 @@ AsyncService::AsyncService(const AsyncService::Config &config)
       safeBatchingPool_(),
       cache_(makeOptionalCache(config_.cacheSize, /*mutexBuckets=*/config_.numWorkers)),
       logger_(config.logger) {
-  if (config_.gpuWorkers.size() != 0) {
-    ABORT_IF(config_.numWorkers != 0, "Unable to mix GPU and CPU workers.");
+  if (config_.gpuWorkers.empty()) {
+    ABORT_IF(config_.numWorkers == 0, "Number of workers should be at least 1 in a threaded workflow");
+    workers_.reserve(config_.numWorkers);
+  } else {
+    if (config_.numWorkers != 0)
+      LOG(info, "Unable to mix GPU and CPU workers, using GPU workers only...");
     workers_.reserve(config_.gpuWorkers.size());
     // VERY VERY HACKY. EVERYTHING USES NUM_WORKERS AS A REFERENCE FOR THE NUMBER OF WORKERS,
     // REFACTOR TO USE gpuWorkers directly...
     config_.numWorkers = config_.gpuWorkers.size();
-  } else {
-    ABORT_IF(config_.numWorkers == 0, "Number of workers should be at least 1 in a threaded workflow");
-    workers_.reserve(config_.numWorkers);
   }
   // Initiate terminology map if present
   if (!config_.terminologyFile.empty()) {