pytorch
72f36880 - [Pytorch][Vulkan] Update spv generation script to embed shader parameters (#88321)

Commit
2 years ago
[Pytorch][Vulkan] Update spv generation script to embed shader parameters (#88321) This diffs adds shader parameters such as tile size, weight storage type and format to the generated spv.cpp file. This is used in ShaderInfo struct that ops such as convolution will use to determine, the workgroup size and how to pack weights. Differential Revision: [D40280337](https://our.internmc.facebook.com/intern/diff/D40280337/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/88321 Approved by: https://github.com/jmdetloff, https://github.com/mcr229
Author
Committer
Parents
Loading