From b3fd46f61cc00b52820356349e3f79b0edd54a84 Mon Sep 17 00:00:00 2001 From: cindyyuanjiang Date: Thu, 28 Dec 2023 10:35:27 -0800 Subject: [PATCH] added new gpu name got emr nodes Signed-off-by: cindyyuanjiang --- user_tools/src/spark_rapids_pytools/cloud_api/sp_types.py | 6 +++--- user_tools/src/spark_rapids_tools/enums.py | 2 -- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/user_tools/src/spark_rapids_pytools/cloud_api/sp_types.py b/user_tools/src/spark_rapids_pytools/cloud_api/sp_types.py index 62905d451..99856322f 100644 --- a/user_tools/src/spark_rapids_pytools/cloud_api/sp_types.py +++ b/user_tools/src/spark_rapids_pytools/cloud_api/sp_types.py @@ -50,6 +50,7 @@ class GpuDevice(EnumeratedType): P4 = 'P4' L4 = 'l4' A10 = 'a10' + A10G = 'a10g' @classmethod def get_default_gpu(cls): @@ -64,7 +65,8 @@ def get_gpu_mem(self) -> list: self.K80: [12288], self.V100: [16384], self.P100: [16384], - self.A10: [24576] + self.A10: [24576], + self.A10G: [24576] } return memory_hash.get(self) @@ -115,8 +117,6 @@ class GpuHWInfo: gpu_device: GpuDevice = GpuDevice.get_default_gpu() def get_gpu_device_name(self) -> str: - print("self.gpu_device =", self.gpu_device) - print("GpuDevice.get_default_gpu()=", GpuDevice.get_default_gpu()) return GpuDevice.tostring(self.gpu_device) diff --git a/user_tools/src/spark_rapids_tools/enums.py b/user_tools/src/spark_rapids_tools/enums.py index 1d195177d..58fa8cd0b 100644 --- a/user_tools/src/spark_rapids_tools/enums.py +++ b/user_tools/src/spark_rapids_tools/enums.py @@ -37,9 +37,7 @@ def tostring(cls, value: Union[Enum, str]) -> str: :return: the uppercase string that represents the state object :rtype: str """ - print("value =", value) value = cast(Enum, value) - print("value =", value) return str(value._value_).upper() # pylint: disable=protected-access @classmethod