Skip to content

Commit dac86fe

Browse files
authored
Fix undefined bias (#81)
1 parent 59eb860 commit dac86fe

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

torch_ipex/csrc/cpu/DevOPs.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,9 +180,11 @@ at::Tensor AtenIpexCPUDev::dil_convolution_overrideable(const at::Tensor & input
180180
std::vector<at::Tensor> dnnl_input_tensors;
181181
dnnl_input_tensors.push_back(input);
182182
dnnl_input_tensors.push_back(weight);
183-
dnnl_input_tensors.push_back(bias);
183+
if (bias.defined()) {
184+
dnnl_input_tensors.push_back(bias);
185+
}
184186
if (dbl::chk::dnnl_support_the_tensors(dnnl_input_tensors))
185-
return AtenIpexCPUDev::dil_convolution(input.is_contiguous() ? input : input.contiguous(), weight.is_contiguous() ? weight : weight.contiguous(), bias.is_contiguous() ? bias : bias.contiguous(), stride, padding, dilation, groups);
187+
return AtenIpexCPUDev::dil_convolution(input.is_contiguous() ? input : input.contiguous(), weight.is_contiguous() ? weight : weight.contiguous(), bias.defined() && !bias.is_contiguous() ? bias.contiguous() : bias, stride, padding, dilation, groups);
186188
}
187189
} catch (std::exception& e) {
188190
#if defined(_DEBUG)

0 commit comments

Comments
 (0)