-
Notifications
You must be signed in to change notification settings - Fork 604
Commit 086646d

morelos
[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 086646dCopy full SHA for 086646d
File tree
Expand file treeCollapse file tree
1 file changed
+494
-0
lines changedFilter options
- backends/vulkan/test/op_tests
Expand file treeCollapse file tree
1 file changed
+494
-0
lines changed
0 commit comments