X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FQATUtil.py;h=e16221fb30c6e10bd8f52842a18057be1a901b9e;hb=fcd0677317970062b37e196b4d1a15a135f51cca;hp=2afea8e29fccf901a418a4e95ff8c6ed6d900480;hpb=837b144cac331214edfe48dfbffe97475e4f4e11;p=csit.git

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"
+                )