diff --git a/torch/csrc/jit/tensorexpr/external_functions.cpp b/torch/csrc/jit/tensorexpr/external_functions.cpp index a9c283e9b8c..be4d2930e00 100644 --- a/torch/csrc/jit/tensorexpr/external_functions.cpp +++ b/torch/csrc/jit/tensorexpr/external_functions.cpp @@ -941,16 +941,28 @@ const static RegisterNNCExternalFunction nnc_quantized_conv2d( const static RegisterNNCExternalFunction nnc_quantized_conv2d_relu( "nnc_aten_quantized_conv2d_relu", nnc_aten_quantized_conv2d_relu); +const static RegisterNNCExternalFunction nnc_quantized_linear( + "nnc_aten_quantized_linear", + nnc_aten_quantized_linear); #ifndef _WIN32 const static RegisterNNCExternalFunction nnc_quantized_add( "nnc_aten_quantized_add", nnc_aten_quantized_add); +const static RegisterNNCExternalFunction nnc_quantized_mul( + "nnc_aten_quantized_mul", + nnc_aten_quantized_mul); +const static RegisterNNCExternalFunction nnc_quantized_mul_scalar( + "nnc_aten_quantized_mul_scalar", + nnc_aten_quantized_mul_scalar); const static RegisterNNCExternalFunction nnc_quantized_sigmoid( "nnc_aten_quantized_sigmoid", nnc_aten_quantized_sigmoid); const static RegisterNNCExternalFunction nnc_quantized_cat( "nnc_aten_quantized_cat", nnc_aten_quantized_cat); +const static RegisterNNCExternalFunction nnc_quantized_relu( + "nnc_aten_quantized_relu", + nnc_aten_quantized_relu); #endif // _WIN32 const static RegisterNNCExternalFunction nnc_quantize_per_tensor( "nnc_aten_quantize_per_tensor", diff --git a/torch/csrc/jit/tensorexpr/operators/quantization.cpp b/torch/csrc/jit/tensorexpr/operators/quantization.cpp index 0a798c2f49a..2451287a1d3 100644 --- a/torch/csrc/jit/tensorexpr/operators/quantization.cpp +++ b/torch/csrc/jit/tensorexpr/operators/quantization.cpp @@ -558,7 +558,7 @@ Tensor computeQuantizedCat( out_qscale, out_qzero); StmtPtr s = - ExternalCall::make(ResultBuf, "nnc_quantized_cat", args, extra_args); + ExternalCall::make(ResultBuf, "nnc_aten_quantized_cat", args, extra_args); return Tensor(ResultBuf.node(), s); }