update test_vulkan (#62795)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62795
Test Plan: Imported from OSS
Reviewed By: driazati
Differential Revision: D30124421
Pulled By: tktrungna
fbshipit-source-id: 235ba166b02f7334e89cb2493024067851bf5b9b
