Add aws_instance_bandwidth_weighting flag.

PiperOrigin-RevId: 822664868

Author: andyz422

perfkitbenchmarker/providers/aws/aws_virtual_machine.py

@@ -962,6 +962,13 @@ def _GenerateCreateCommand(self):
           '{CapacityReservationId=%s}' % reservation_id]
     else:
       reservation_args = []
+
+    bandwidth_weighting_args = []
+    if aws_flags.AWS_INSTANCE_BANDWIDTH_WEIGHTING.value:
+      bandwidth_weighting_args = [
+          '--network-performance-options',
+          f'BandwidthWeighting={aws_flags.AWS_INSTANCE_BANDWIDTH_WEIGHTING.value}'
+      ]
     create_cmd = util.AWS_PREFIX + [
         'ec2',
         'run-instances',
@@ -972,7 +979,7 @@ def _GenerateCreateCommand(self):
         '--key-name=%s' % AwsKeyFileManager.GetKeyNameForRun(),
         '--tag-specifications=%s'
         % util.FormatTagSpecifications('instance', self.aws_tags),
-    ] + reservation_args
+    ] + reservation_args + bandwidth_weighting_args
 
     if FLAGS.aws_vm_hibernate:
       create_cmd.extend([
@@ -1607,6 +1614,9 @@ def GetResourceMetadata(self):
       result['efa_count'] = FLAGS.aws_efa_count
       result['nic_count'] = FLAGS.aws_efa_count
     result['preemptible'] = self.use_spot_instance
+    result['instance_bandwidth_weighting'] = (
+        aws_flags.AWS_INSTANCE_BANDWIDTH_WEIGHTING.value or 'default'
+    )
     return result
 
   def DiskTypeCreatedOnVMCreation(self, disk_type):

perfkitbenchmarker/providers/aws/flags.py

@@ -39,7 +39,13 @@
     'The queue count of each NIC. Specify a list of key=value pairs, where key'
     ' is the network device name and value is the queue count.',
 )
-
+AWS_INSTANCE_BANDWIDTH_WEIGHTING = flags.DEFINE_enum(
+    'aws_instance_bandwidth_weighting',
+    None,
+    ['vpc-1', 'ebs-1'],
+    'The bandwidth weighting of each instance, increasing one of vpc and ebs'
+    ' bandwidth at the expense of the other. Valid options are vpc-1, ebs-1.',
+)
 flags.DEFINE_string(
     'aws_dax_node_type',
     'dax.r4.large',