Added components for poc mode. (NVIDIA#414)

pxLi · Apr 18, 2022 · ceba1da · ceba1da
1 parent e851f70
commit ceba1da
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 2 deletions.
diff --git a/nvflare/poc/client/startup/fed_client.json b/nvflare/poc/client/startup/fed_client.json
@@ -30,5 +30,21 @@
       "sp_end_point": "localhost:8002:8003",
       "heartbeat_interval": 6
     }
-  }
+  },
+  "components": [
+    {
+      "id": "resource_manager",
+      "path": "nvflare.apis.impl.list_resource_manager.ListResourceManager",
+      "args":  {
+        "resources": {"gpu": [0,1,2,3]}
+        }
+    },
+    {
+      "id": "resource_consumer",
+      "path": "nvflare.apis.impl.gpu_resource_consumer.GPUResourceConsumer",
+      "args":  {
+        "gpu_resource_key": "gpu"
+        }
+    }
+  ]
 }
diff --git a/nvflare/poc/server/startup/fed_server.json b/nvflare/poc/server/startup/fed_server.json
@@ -45,5 +45,38 @@
       "sp_end_point": "localhost:8002:8003",
       "heartbeat_interval": 6
     }
-  }
+  },
+  "components": [
+    {
+      "id": "job_scheduler",
+      "path": "nvflare.apis.impl.job_scheduler.DefaultJobScheduler",
+      "args": {
+        "max_jobs": 4
+      }
+    },
+    {
+      "id": "job_manager",
+      "path": "nvflare.apis.impl.job_def_manager.SimpleJobDefManager",
+      "args": {
+        "uri_root": "/tmp/jobs-storage",
+        "job_store_id": "job_store",
+        "temp_dir": "/tmp/jobs_temp"
+      }
+    },
+    {
+      "id": "job_store",
+      "name": "FilesystemStorage",
+      "args": {}
+    },
+    {
+      "id": "study_manager",
+      "path": "nvflare.apis.impl.study_manager.StudyManager",
+      "args": {
+        "storage": {
+          "name": "FilesystemStorage",
+          "args": {}
+        }
+      }
+    }
+  ]
 }