From fb0d1e15d33c151f6df923903a769d711329e90e Mon Sep 17 00:00:00 2001
From: "wangang.wa" <wangang.wa@alibaba-inc.com>
Date: Tue, 16 Jul 2024 10:46:15 +0800
Subject: [PATCH] Adapt to torchacc 2.3

---
 .../pytorch_blade/compiler/jit/torch/constant_propagation.cpp   | 1 -
 .../pytorch_blade/compiler/jit/torch/freeze_module.cpp          | 2 +-
 2 files changed, 1 insertion(+), 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
index 20b81c9c2af..39a4bd814bb 100644
--- 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
index c828dc993fb..cf308891e0a 100644
--- 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();