[PyBindings] Wait for the device instead of per-result flags, and fix bfloat16 result layout and input errors - #23999

[PyBindings] Wait for the device instead of per-result flags, and fix bfloat16 result layout and input errors#23999

shoumikhin merged 1 commit into

pytorch:mainpytorch/executorch:mainfrom

shoumikhin:pybindings-cuda-wait-followupshoumikhin/executorch:pybindings-cuda-wait-followupCopy head branch name to clipboard