OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Update base for Update on "WIP - Quantized BI Bytedoc workarounds"

Browse Source 
[ghstack-poisoned]
This commit is contained in:
Ivan Kobzarev 2021-11-24 14:35:59 -08:00
parent ea0ccc85ff
commit 84e5a1529c
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
torch/csrc/jit/tensorexpr/external_functions.cpp
View File

@ -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",

2
torch/csrc/jit/tensorexpr/operators/quantization.cpp
View File

@ -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);
}