From cf8d1a9d533ad52c4b026b94692cf815cffe2563 Mon Sep 17 00:00:00 2001 From: Chen Lai Date: Mon, 22 Apr 2024 13:09:28 -0700 Subject: [PATCH] backout the schema definition change Summary: The schame was changed to avoid double register, but it was hiding the symptons by using a differnt schema. Resume the correct the schema Differential Revision: D56432559 --- kernels/quantized/quantized.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kernels/quantized/quantized.yaml b/kernels/quantized/quantized.yaml index ade5575e32e..ca2360b7d80 100644 --- a/kernels/quantized/quantized.yaml +++ b/kernels/quantized/quantized.yaml @@ -40,7 +40,7 @@ - arg_meta: null kernel_name: torch::executor::quantized_embedding_byte_out -- func: quantized_decomposed::embedding_byte.dtype_out(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, int weight_quant_min, int weight_quant_max, Tensor indices, ScalarType? dtype=None, *, Tensor(a!) out) -> Tensor(a!) +- func: quantized_decomposed::embedding_byte.dtype_out(Tensor weight, Tensor weight_scales, Tensor? weight_zero_points, int weight_quant_min, int weight_quant_max, Tensor indices, *, ScalarType? dtype=None, Tensor(a!) out) -> Tensor(a!) variants: function kernels: - arg_meta: null