We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a310c3b commit f4732b8Copy full SHA for f4732b8
torch_ipex/csrc/cpu/dbl/Common.cpp
@@ -36,7 +36,11 @@ at::Tensor dil_tensor_to_dense(const at::Tensor& tensor) {
36
37
dil::tensor try_gen_dil_tensor(const at::Tensor &input) {
38
if (cpu::ShadeDataContext::isDilTensor(input)) {
39
- return cpu::ShadeDataContext::getDilTensor(input);
+ auto dil_tensor = cpu::ShadeDataContext::getDilTensor(input);
40
+ if (dil_tensor.is_public_format()) {
41
+ dil_tensor.set_dims_and_strides(input.sizes().vec(), input.strides().vec());
42
+ }
43
+ return dil_tensor;
44
} else {
45
return dil_tensor_from_dense(input);
46
}
0 commit comments