[LLGA] remove infer_shape due to backend change (#176)

intel · Aug 31, 2021 · c1fe854 · c1fe854
1 parent 568c4d9
commit c1fe854
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/third_party/llga b/third_party/llga
diff --git a/torch_ipex/csrc/jit/codegen/onednn/kernel.cpp b/torch_ipex/csrc/jit/codegen/onednn/kernel.cpp
@@ -70,14 +70,10 @@ ArgSpecs LlgaKernel::specializeInputSpecs(const TensorArgs &inputs) const {
 
 ArgSpecs LlgaKernel::specializeOutputSpecs(const partition &partition,
                                            const ArgSpecs &inputSpecs) const {
-  auto inputs = fmap(inputSpecs, toLogicalTensor);
-  auto outputs = fmap(graph_->outputs(), toLogicalTensor);
-  partition.infer_shape(inputs, outputs);
-
   ArgSpecs outputSpecs;
   outputSpecs.reserve(nOutputs_);
   for (size_t i = 0; i < nOutputs_; i++) {
-    auto spec = ArgSpec(outputs[i]);
+    auto spec = ArgSpec(graph_->outputs()[i]);
 
     if (spec.is_quantized())
       spec = getQuantizedSpec(spec, i);