IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CPU] windows_Interpolate_fused-FQ_nearest-mode_nspc-layout_fix (#5317)

Browse Source
This commit is contained in:
Chenhu Wang
2021-05-03 19:48:42 +08:00
committed by GitHub
parent 7d2ec02d65
commit 0808975a37
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
inference-engine/src/mkldnn_plugin/nodes/mkldnn_interpolate_node.cpp
View File

@@ -386,6 +386,13 @@ private:
Xbyak::Label out_loop_label;
Xbyak::Label out_loop_end;
Xbyak::Reg64 reg_work_amount_bk = reg_src_aux2;
Xbyak::Reg64 reg_oc_off_bk = rsi;
mov(reg_work_amount_bk, ptr[reg_params + GET_OFF(work_amount)]);
if (attr_.post_ops_.len() != 0) {
mov(reg_oc_off_bk, ptr[reg_params + GET_OFF(oc_off)]);
}
Xbyak::Reg64 reg_work_amount_out = reg_src_aux1;
mov(reg_work_amount_out, jcp_.OW);
L(out_loop_label);
@@ -410,9 +417,9 @@ private:
mov(reg_index_offset, dword[reg_index]);
add(reg_src_aux, reg_index_offset);
mov(reg_work_amount, ptr[reg_params + GET_OFF(work_amount)]);
mov(reg_work_amount, reg_work_amount_bk);
if (attr_.post_ops_.len() != 0)
mov(reg_oc_off, ptr[reg_params + GET_OFF(oc_off)]);
mov(reg_oc_off, reg_oc_off_bk);
L(nn_loop_label);
{

1
inference-engine/tests/functional/plugin/cpu/single_layer_tests/interpolate.cpp
View File

@@ -236,6 +236,7 @@ const std::vector<fusingSpecificParams> interpolateFusingParamsSet{
emptyFusingSpec,
fusingRelu,
fusingSwish,
fusingFakeQuantizePerChannelRelu,
};
std::vector<std::map<std::string, std::string>> filterAdditionalConfig() {