[ET-VK] 6/n Split dispatches between multiple command buffers. Repurpose submit_current_cmd_and_wait to wait based on input flag.

[trivedivivek]

Stack from ghstack (oldest at bottom):

• [ET-VK] 7/n Split dispatches between multiple command buffers. Split execute dispatch into multiple commands based on dispatch count. #12530
• -> [ET-VK] 6/n Split dispatches between multiple command buffers. Repurpose submit_current_cmd_and_wait to wait based on input flag. #12529
• [ET-VK] 5/n Split dispatches between multiple command buffers. Track previously submitted command buffers in context and add function to execute all previous command buffers. #12527
• [ET-VK] 4/n Split dispatches between multiple command buffers. Removing flush function call from submit_current_cmd_and_wait. #12525
• [ET-VK] 3/n Split dispatches between multiple command buffers. Track previous semaphore in context. #12523
• [ET-VK] 2/n Split dispatches between multiple command buffers. Add semaphore support to Adapter::submit_cmd. #12522
• [ET-VK] 1/n Split dispatches between multiple command buffers. Add semaphore support in command buffer. #12519

This diff makes changes to the submit_current_cmd_and_wait function in the ComputeGraph class to repurpose it to wait for command buffer completion based on an input flag. The function is renamed to submit_current_cmd and now takes an additional wait parameter. If wait is true, the function submits the command buffer to the GPU and waits for its completion. Otherwise, it only submits the command buffer without waiting.

Differential Revision: D78360042

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/12529

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 2 New Failures, 2 Unrelated Failures

As of commit c103fd7 with merge base 4551a56 ():

NEW FAILURES - The following jobs have failed:

• pull / unittest / linux / linux-job (gh)
  RuntimeError: Command docker exec -t e12e07f62e1bb06aa1d8ec5b8c0c05bad5c1c4d81bd32b96aac90b142402899f /exec failed with exit code 1
• pull / unittest-editable / linux / linux-job (gh)
  RuntimeError: Command docker exec -t 8be81e7f5b444ec5c6d5429a8a1e3de3f173195f15c8128f7a90200689441220 /exec failed with exit code 1

BROKEN TRUNK - The following jobs failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• pull / unittest / macos / macos-job (gh) (trunk failure)
  RuntimeError: Command bash /Users/ec2-user/runner/_work/_temp/exec_script failed with exit code 1
• pull / unittest-editable / macos / macos-job (gh) (trunk failure)
  RuntimeError: Command bash /Users/ec2-user/runner/_work/_temp/exec_script failed with exit code 1

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D78360042

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D78360042

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D78360042