Patch embedding4b

larryliu0820 · larryliu0820 · commit 9dab3e8ade5a · 2024-04-17T18:55:43.000-07:00
Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:
diff --git a/kernels/quantized/cpu/op_embedding4b.cpp b/kernels/quantized/cpu/op_embedding4b.cpp
@@ -195,7 +195,7 @@ void resize_out_tensor(
   for (size_t i = 0; i < indices.dim(); i++) {
     expected_output_size[i] = indices.size(i);
   }
-  const size_t embedding_dim = weight.size(1);
+  const size_t embedding_dim = weight.size(1) * 2;
   expected_output_size[out.dim() - 1] = embedding_dim;
 
   exec_aten::ArrayRef<exec_aten::SizesType> output_size{

Original file line number	Diff line number	Diff line change
`@@ -195,7 +195,7 @@ void resize_out_tensor(`
`195`	`195`	`for (size_t i = 0; i < indices.dim(); i++) {`
`196`	`196`	`expected_output_size[i] = indices.size(i);`
`197`	`197`	`}`
`198`		`- const size_t embedding_dim = weight.size(1);`
	`198`	`+ const size_t embedding_dim = weight.size(1) * 2;`
`199`	`199`	`expected_output_size[out.dim() - 1] = embedding_dim;`
`200`	`200`
`201`	`201`	`exec_aten::ArrayRef<exec_aten::SizesType> output_size{`