[BUG] Fix databricks-aws user profiling tool error with `--gpu_cluste…

…r` argument (#707)


Fixes #695
---------

Signed-off-by: cindyyuanjiang <cindyj@nvidia.com>

data_validation/src/spark_rapids_validation_tool/utilities.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2023, NVIDIA CORPORATION.
+# Copyright (c) 2023-2024, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -110,7 +110,7 @@ def gen_random_string(str_length: int) -> str:
 
 
 def get_gpu_device_list():
-    return ['T4', 'V100', 'K80', 'A100', 'P100']
+    return ['T4', 'V100', 'K80', 'A100', 'P100', 'A10', 'A10G']
 
 
 def is_valid_gpu_device(val):

user_tools/src/spark_rapids_pytools/cloud_api/sp_types.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2023, NVIDIA CORPORATION.
+# Copyright (c) 2023-2024, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -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)