Use -si for diskspd prefill

PiperOrigin-RevId: 839570314

Author: Arushi-07

perfkitbenchmarker/configs/default_benchmark_config.yaml

@@ -516,8 +516,10 @@ diskspd_single_thread_latency:
   name: diskspd
   flags: &diskspd_single_thread_latency_flags
     num_striped_disks: 1
-    data_disk_size: 500
-    diskspd_file_size: 400G
+    data_disk_size: 1024
+    diskspd_file_size: 1000G
+    provisioned_iops: 16000
+    provisioned_throughput: 1000
     diskspd_thread_number_per_file: 1
     diskspd_outstanding_io: 1
     diskspd_block_size: 8k

perfkitbenchmarker/windows_packages/diskspd.py

@@ -487,8 +487,9 @@ def Prefill(running_vm):
 
 
 def RunDiskspdPrefillCommandForDuration(vm, diskspd_exe_dir, prefill_duration):
+  # using -si so that all the threads share the same offset and don't overlap.
   diskspd_options = (
-      f'-c{FLAGS.diskspd_file_size} -t16 -w100'
+      f'-c{FLAGS.diskspd_file_size} -t16 -w100 -si'
       f' -b{DISKSPD_PREFILL_BLOCK_SIZE.value} -d{prefill_duration} -Rxml -Sh'
       f' -o16 -Zr C:\\scratch\\{DISKSPD_TMPFILE} > {DISKSPD_XMLFILE}'
   )

tests/windows_packages/diskspd_test.py

@@ -16,6 +16,7 @@
 
 import os
 import unittest
+from unittest import mock
 import xml.etree.ElementTree as ET
 
 from absl import flags
@@ -184,6 +185,26 @@ def testGenerateDiskspdConfigBooleans(self):
     self.assertIn('-l', config)
     self.assertIn('-n', config)
 
+  @flagsaver.flagsaver(diskspd_file_size='1G', diskspd_prefill_block_size='1M')
+  def testRunDiskspdPrefillCommandForDuration(self):
+    vm = mock.Mock()
+    with mock.patch.object(
+        diskspd, '_CatXml', return_value='<xml>result</xml>'
+    ) as cat_xml_mock:
+      with mock.patch.object(diskspd, '_RemoveXml') as remove_xml_mock:
+        result = diskspd.RunDiskspdPrefillCommandForDuration(
+            vm, '/tmp/x86', 100
+        )
+        self.assertEqual(result, '<xml>result</xml>')
+        command = (
+            'cd /tmp/x86; .\\diskspd.exe -c1G -t16 -w100 -si -b1M -d100 -Rxml'
+            f' -Sh -o16 -Zr C:\\scratch\\{diskspd.DISKSPD_TMPFILE} >'
+            f' {diskspd.DISKSPD_XMLFILE}'
+        )
+        vm.RobustRemoteCommand.assert_called_with(command)
+        cat_xml_mock.assert_called_with(vm)
+        remove_xml_mock.assert_called_with(vm)
+
 
 if __name__ == '__main__':
   unittest.main()