[vulkan] Make vulkan ops selective (#58332)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58332
Test Plan: Imported from OSS
Reviewed By: iseeyuan
Differential Revision: D28454976
Pulled By: IvanKobzarev
fbshipit-source-id: 445c1f326be76e3530a4884aa5fe749d636e0ae5