Add quantize_per_channel and dequantize_per_channel to q_dq_ops target

Differential Revision: D66400800 Pull Request resolved: pytorch#7045
kirklandsign · Dec 2, 2024 · 2d499b3 · 2d499b3
1 parent a347665
commit 2d499b3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/kernels/quantized/targets.bzl b/kernels/quantized/targets.bzl
@@ -69,6 +69,8 @@ def define_common_targets():
                 "quantized_decomposed::dequantize_per_tensor.Tensor_out",
                 "quantized_decomposed::quantize_per_tensor.out",
                 "quantized_decomposed::quantize_per_tensor.Tensor_out",
+                "quantized_decomposed::dequantize_per_channel.out",
+                "quantized_decomposed::quantize_per_channel.out",
             ],
     )