Skip to content

Commit 086646d

Browse files
author
morelos
committed
[ET-VK][Ops] dequantize_per_token.default test setup
Pull Request resolved: #11482 # Context In order to enhance my own understanding of these operators, I needed to create a reference implementation, and also build out the vulkan testing framework which creates the necessary build up when I need to call the vulkan implementations. I won't explain what the dequantize operator actually is in this diff, but will rather opt to explain the operator in a future diff where I implement the glsl shader, however, the reference implementation is heavily inspired by the cpu implementation and aims to create similar checks when using the zero points and scales and performing the dequantization with the given parameters. # Changes The main changes were the include of the reference implementation that is used for my own learning, and the necessary wrapper functions that will be called later when the vulkan implementation is successfully completed. It has everything necessary for this purpose, including calling the operator by its appropriate name as when defined in the C++ implementation header, and staging components correctly from the GPU and then the CPU which will be where the comparison is done. I have also included comprehensive failure print statements that prints the tensor size along with relevant parameters such as the zero points or scales passed in. ghstack-source-id: 290376484 @exported-using-ghexport Differential Revision: [D76267037](https://our.internmc.facebook.com/intern/diff/D76267037/)
1 parent 5968ac4 commit 086646d

File tree

1 file changed

+494
-0
lines changed

1 file changed

+494
-0
lines changed

0 commit comments

Comments
 (0)