pytorch
192e806c - [Pytorch][vulkan] Generate shader with parameters (#88322)

Commit
2 years ago
[Pytorch][vulkan] Generate shader with parameters (#88322) Parametsr such as tile size and weight type and format is embedded within the shader code. This is used to generate ShaderInfo. For now we will maintain both ShaderSrc and ShaderInfo so as to transition from VK_KERNEL to VK_SHADER incremental. Otherwise we will have to switch multiple of them at the same time. Differential Revision: [D40280338](https://our.internmc.facebook.com/intern/diff/D40280338/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/88322 Approved by: https://github.com/jmdetloff, https://github.com/mcr229
Author
Committer
Parents
Loading