Skip to content

Commit 1f7f390

Browse files
author
ssjia
committed
Update on "[ET-VK] AOT logic for quantized conv2d"
### Changes As title; this diff adds the necessary export logic required to enable fusing quantized convolution patterns into the custom ops introduced in the below diff. Differential Revision: [D83437826](https://our.internmc.facebook.com/intern/diff/D83437826/) [ghstack-poisoned]
2 parents 8733004 + b0c0bbc commit 1f7f390

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backends/vulkan/test/custom_ops/CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,4 +96,6 @@ if(TARGET vulkan_backend)
9696
add_operator_prototype(q4gsw_linear)
9797
add_operator_prototype(choose_qparams_per_row)
9898
add_operator_prototype(qdq8ta_conv2d_activations)
99+
add_operator_prototype(q8ta_q8csw_q8to_conv2d)
100+
add_operator_prototype(q8ta_q8csw_q8to_conv2d_dw)
99101
endif()

0 commit comments

Comments
 (0)