Skip to content

Commit f4af2b9

Browse files
authored
revert onednn update (#189)
* Revert "[LLGA] remove infer_shape due to backend change (#176)" This reverts commit c1fe854. * Revert "[LLGA] ww35 update backend (#175)" This reverts commit eccafbe.
1 parent c1fe854 commit f4af2b9

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

third_party/llga

Submodule llga updated from 5f36faf to 1c0bff1

torch_ipex/csrc/jit/codegen/onednn/kernel.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,10 +70,14 @@ ArgSpecs LlgaKernel::specializeInputSpecs(const TensorArgs &inputs) const {
7070

7171
ArgSpecs LlgaKernel::specializeOutputSpecs(const partition &partition,
7272
const ArgSpecs &inputSpecs) const {
73+
auto inputs = fmap(inputSpecs, toLogicalTensor);
74+
auto outputs = fmap(graph_->outputs(), toLogicalTensor);
75+
partition.infer_shape(inputs, outputs);
76+
7377
ArgSpecs outputSpecs;
7478
outputSpecs.reserve(nOutputs_);
7579
for (size_t i = 0; i < nOutputs_; i++) {
76-
auto spec = ArgSpec(graph_->outputs()[i]);
80+
auto spec = ArgSpec(outputs[i]);
7781

7882
if (spec.is_quantized())
7983
spec = getQuantizedSpec(spec, i);

0 commit comments

Comments
 (0)