fix: Using toGraphFunction API to get graph

Signed-off-by: Anurag Dixit <[email protected]>

Author: Anurag Dixit

core/lowering/passes/linear_to_addmm.cpp

@@ -7,6 +7,7 @@
 #include "torch/csrc/jit/passes/guard_elimination.h"
 #include "torch/csrc/jit/passes/peephole.h"
 #include "torch/csrc/jit/runtime/graph_executor.h"
+#include "torch/csrc/jit/api/function_impl.h"
 
 #include "core/util/prelude.h"
 #include "torch/csrc/jit/passes/subgraph_rewrite.h"
@@ -34,7 +35,7 @@ void replaceLinearWithBiasNonePattern(std::shared_ptr<torch::jit::Graph> graph)
         continue;
       } else {
         torch::jit::WithInsertPoint guard(*it);
-        std::shared_ptr<torch::jit::Graph> d_graph = decompose_funcs.get_function("linear").graph();
+        std::shared_ptr<torch::jit::Graph> d_graph = toGraphFunction(decompose_funcs.get_function("linear")).graph();
         torch::jit::Value* new_output = insertGraph(*it->owningGraph(), *d_graph, it->inputs()).at(0);
         new_output->setType(it->output()->type());
         it->output()->replaceAllUsesWith(new_output);