From: pmikus <peter.mikus@protonmail.ch>
Date: Thu, 23 Nov 2023 09:08:23 +0000 (+0000)
Subject: fix(infra): QAT init
X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=commitdiff_plain;h=0efc3083ed2bacaf3445a50fae1679a5ae88d7d9

fix(infra): QAT init

Signed-off-by: pmikus <peter.mikus@protonmail.ch>
Change-Id: I862f9ca12e9149eab2d0701368cdc65f294629cc
---

diff --git a/resources/libraries/python/QATUtil.py b/resources/libraries/python/QATUtil.py
index 2afea8e29f..e16221fb30 100644
--- a/resources/libraries/python/QATUtil.py
+++ b/resources/libraries/python/QATUtil.py
@@ -76,15 +76,17 @@ class QATUtil:
 
         # Initialize QAT VFs.
         if int(device["numvfs"]) > 0:
+            path = f"drivers/{device['driver']}"
             DUTSetup.set_sriov_numvfs(
-                node, device["pci_address"], path="drivers/4xxx",
+                node, device["pci_address"], path=path,
                 numvfs=device["numvfs"]
             )
 
-        for cvf in range(int(device["numvfs"])):
-            DUTSetup.pci_vf_driver_unbind(
-                node, device["pci_address"], cvf
-            )
-            DUTSetup.pci_vf_driver_bind(
-                node, device["pci_address"], cvf, "vfio-pci"
-            )
+        if device["driver"] not in ["c4xxx"]:
+            for cvf in range(int(device["numvfs"])):
+                DUTSetup.pci_vf_driver_unbind(
+                    node, device["pci_address"], cvf
+                )
+                DUTSetup.pci_vf_driver_bind(
+                    node, device["pci_address"], cvf, "vfio-pci"
+                )