From 7a5f20e96191c49e5783b89e11784f3a9360c728 Mon Sep 17 00:00:00 2001
From: Joji Mekkattuparamban <jojim@nvidia.com>
Date: Tue, 25 Jun 2024 14:31:49 -0700
Subject: [PATCH] address review comments

Signed-off-by: Joji Mekkattuparamban <jojim@nvidia.com>
---
 assets/state-vfio-manager/0400_configmap.yaml | 26 +++++++++++--------
 1 file changed, 15 insertions(+), 11 deletions(-)

diff --git a/assets/state-vfio-manager/0400_configmap.yaml b/assets/state-vfio-manager/0400_configmap.yaml
index aa330ccbb..85df8f8ca 100644
--- a/assets/state-vfio-manager/0400_configmap.yaml
+++ b/assets/state-vfio-manager/0400_configmap.yaml
@@ -129,20 +129,24 @@ data:
         local gpu=$1
         device_class_file=$(readlink -f "/sys/bus/pci/devices/$gpu/class")
         device_class=$(cat "$device_class_file")
-        if [ "$device_class" == "0x030000" ]; then
-          if ls "/sys/bus/pci/devices/$gpu" | grep consumer >& /dev/null; then
-            aux_dev=$(ls "/sys/bus/pci/devices/$gpu" | grep consumer | awk -Fconsumer:pci: '{print $2}')
-            if [ "$device_class" == "" ]; then
-              echo "NONE"
-              return
-            fi
+        if [ "$device_class" != "0x030000" ]; then
+          echo "NONE"
+          return
+        fi
 
-            if ls "/sys/bus/pci/devices/$aux_dev/" >& /dev/null; then
-              echo "$aux_dev"
-              return
-            fi
+        if ls "/sys/bus/pci/devices/$gpu" | grep consumer >& /dev/null; then
+          aux_dev=$(ls "/sys/bus/pci/devices/$gpu" | grep consumer | awk -Fconsumer:pci: '{print $2}')
+          if [ "$aux_dev" == "" ]; then
+            echo "NONE"
+            return
+          fi
+
+          if ls "/sys/bus/pci/devices/$aux_dev/" >& /dev/null; then
+            echo "$aux_dev"
+            return
           fi
         fi
+
         echo "NONE"
     }