Unverified Commit 6cc1e7d9 authored by Li, Jiang's avatar Li, Jiang Committed by GitHub
Browse files

[CPU] Update custom ops for the CPU backend (#20255)


Signed-off-by: default avatarjiang1.li <jiang1.li@intel.com>
parent 9909726d
This diff is collapsed.
......@@ -43,7 +43,7 @@ class UnquantizedEmbeddingMethod(QuantizeMethodBase):
layer: torch.nn.Module,
x: torch.Tensor,
bias: Optional[torch.Tensor] = None) -> torch.Tensor:
return dispatch_unquantized_gemm()(x, layer.weight, bias)
return dispatch_unquantized_gemm()(layer, x, layer.weight, bias)
def embedding(self, layer: torch.nn.Module,
input_: torch.Tensor) -> torch.Tensor:
......
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment