[Vulkan] Improve error handling in a few places. (#51423)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/51423
Test Plan: Imported from OSS
Reviewed By: SS-JIA
Differential Revision: D26179604
Pulled By: AshkanAliabadi
fbshipit-source-id: 2e270423bf7e960e9303b17e0ca1a1530b760ad3
