Vulkan tweaks (#47261)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47261
Test Plan: Imported from OSS
Reviewed By: SS-JIA
Differential Revision: D24837714
Pulled By: AshkanAliabadi
fbshipit-source-id: 221258c03a7f2304a3b34ad550c458c49a108cd0