Request two VFs for primary UDN SR-IOV pods

Count one VF for the cluster default network and one for the
primary UDN. Preserve existing resource counts for regular SR-IOV
and secondary-network pods.

Signed-off-by: Venkat Kunaparaju <vkunaparaju@nvidia.com>

Author: vkunaparaju

manifests/sriov-pod.yaml.j2

@@ -25,11 +25,11 @@ spec:
     imagePullPolicy: {{ image_pull_policy }}{% if has_resource_name or has_resources %}
     resources:
       requests: {% if has_resource_name %}
-        {{ resource_name }}: '1'{% endif %}{% if cpu_request %}
+        {{ resource_name }}: {{ sriov_resource_quantity }}{% endif %}{% if cpu_request %}
         cpu: {{ cpu_request }}{% endif %}{% if mem_request %}
         memory: {{ mem_request }}{% endif %}
       limits: {% if has_resource_name %}
-        {{ resource_name }}: '1'{% endif %}{% if cpu_limit %}
+        {{ resource_name }}: {{ sriov_resource_quantity }}{% endif %}{% if cpu_limit %}
         cpu: {{ cpu_limit }}{% endif %}{% if mem_limit %}
         memory: {{ mem_limit }}{% endif %}{% endif %}
     securityContext:

task.py

@@ -451,6 +451,8 @@ def _get_sriov_resource_quantity(
         self,
         secondary_network_nads: tuple[str, ...],
     ) -> str:
+        if self.pod_type == PodType.SRIOV:
+            return str(1 + int(self.ts.test_case_id.is_udn_primary))
         return str(len(secondary_network_nads) + int(self.node.sriov))
 
     def get_template_args(self) -> dict[str, str | list[str] | bool]: