Adapt to torchacc 2.3

alibaba · Jul 16, 2024 · fb0d1e1 · fb0d1e1
1 parent 72eec1c
commit fb0d1e1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/pytorch_blade/pytorch_blade/compiler/jit/torch/constant_propagation.cpp b/pytorch_blade/pytorch_blade/compiler/jit/torch/constant_propagation.cpp
@@ -29,7 +29,6 @@
 #include <torch/csrc/jit/passes/dead_code_elimination.h>
 #include <torch/csrc/jit/runtime/operator.h>
 #include <torch/csrc/jit/runtime/vararg_functions.h>
-#include <torch/csrc/utils/memory.h>
 #endif
 
 namespace torch {

diff --git a/pytorch_blade/pytorch_blade/compiler/jit/torch/freeze_module.cpp b/pytorch_blade/pytorch_blade/compiler/jit/torch/freeze_module.cpp
@@ -254,7 +254,7 @@ class AttributePropagator {
   void recordMutableAttrs(std::shared_ptr<Graph>& graph) {
     std::stack<Block*> blocks({graph->block()});
     std::unique_ptr<AliasDb> aliasDb =
-        torch::make_unique<AliasDb>(graph, /* isFrozen */ true);
+        std::make_unique<AliasDb>(graph, /* isFrozen */ true);
     while (!blocks.empty()) {
       Block* block = blocks.top();
       blocks.pop();