[CoreEngine] get the gpu provider which will be showed to users when …

…launching jobs.
jrpizani · Aug 24, 2023 · 5bb1e82 · 5bb1e82
1 parent 23d7545
commit 5bb1e82
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/devops/dockerfile/device-image/Dockerfile-Dev b/devops/dockerfile/device-image/Dockerfile-Dev
@@ -1,6 +1,6 @@
 ARG VERSION=dev
 ARG IS_BUILDING_GPU_IMAGE=0
-ARG BASE_IMAGE=public.ecr.aws/x6k8q1x9/fedml-device-image:base
+ARG BASE_IMAGE=fedml/fedml-device-image:base
 FROM ${BASE_IMAGE}
 
 ADD ./devops/scripts/runner.sh ./fedml/runner.sh

diff --git a/python/fedml/__init__.py b/python/fedml/__init__.py
@@ -25,7 +25,7 @@
 _global_training_type = None
 _global_comm_backend = None
 
-__version__ = "0.8.8a43"
+__version__ = "0.8.8a44"
 
 
 def init(args=None):

diff --git a/python/fedml/cli/scheduler/job_manager.py b/python/fedml/cli/scheduler/job_manager.py
@@ -195,7 +195,7 @@ def __init__(self, gpu_device_json):
         self.gpu_count = gpu_device_json["got_gpu_count"]
         self.gpu_name = gpu_device_json["gpu_name"]
         self.gpu_instance = self.gpu_name
-        self.gpu_provider = "FedML"
+        self.gpu_provider = gpu_device_json["gpu_provider"]
 
 
 class FedMLJobModelList(object):

diff --git a/python/setup.py b/python/setup.py
@@ -94,7 +94,7 @@ def finalize_options(self):
 
 setup(
     name="fedml",
-    version="0.8.8a43",
+    version="0.8.8a44",
     author="FedML Team",
     author_email="ch@fedml.ai",
     description="A research and production integrated edge-cloud library for "