re-add impl for override_worker_cpu_count

erdos-project · Nov 23, 2024 · 5891dc3 · 5891dc3
1 parent e76c3fc
commit 5891dc3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/rpc/service.py b/rpc/service.py
@@ -386,9 +386,8 @@ async def RegisterWorker(self, request, context):
         cpu_resource = Resource(name="Slot")
         worker_resources = Resources(
             resource_vector={
-                # TODO(elton): handle override worker cpu count?
-                # cpu_resource: request.cores,
-                cpu_resource: 640,
+                cpu_resource: request.cores if not FLAGS.override_worker_cpu_count
+                else 640
             },
             _logger=self._logger,
         )