[SWDEV-512393] Added amdsmi_get_cpu_affinity_with_scope (#198)

Signed-off-by: Maisam Arif <Maisam.Arif@amd.com>
Signed-off-by: Deepak Mewar <deepak.mewar@amd.com>

amdsmi_cli/amdsmi_commands.py

@@ -832,8 +832,24 @@ class AMDSMICommands():
                     numa_affinity = "N/A"
                     logging.debug("Failed to get numa affinity for gpu %s | %s", gpu_id, e.get_error_info())
 
+                try:
+                    cpu_set = amdsmi_interface.amdsmi_get_cpu_affinity_with_scope(args.gpu, amdsmi_interface.AmdSmiAffinityScope.NUMA_SCOPE)
+                except amdsmi_exception.AmdSmiLibraryException as e:
+                    cpu_set = []
+                    cpu_set.append(-1)
+                    logging.debug("Failed to get cpu affinity for gpu %s | %s", gpu_id, e.get_error_info())
+
+                try:
+                    cpusockets = amdsmi_interface.amdsmi_get_cpu_affinity_with_scope(args.gpu, amdsmi_interface.AmdSmiAffinityScope.SOCKET_SCOPE)
+                except amdsmi_exception.AmdSmiLibraryException as e:
+                    cpusockets = []
+                    cpusockets.append(-1)
+                    logging.debug("Failed to get socket affinity for gpu %s | %s", gpu_id, e.get_error_info())
+
                 static_dict['numa'] = {'node' : numa_node_number,
-                                        'affinity' : numa_affinity}
+                                        'affinity' : numa_affinity,
+                                        'CPU affinity' : [hex(cpus) for cpus in cpu_set],
+                                        'Socket affinity' : [socket for socket in set(cpusockets)]}
         if args.vram:
             vram_info_dict = {"type" : "N/A",
                               "vendor" : "N/A",