From 33bcd11161f51f484ef531ef7acab75141c8c5b8 Mon Sep 17 00:00:00 2001
From: zhaoguochun1995 <zhaoguochun1995@163.com>
Date: Tue, 9 Apr 2024 07:02:19 +0000
Subject: [PATCH] fix no need change

---
 .../csrc_dipu/vendor/ascend/AscendCustomAllocatorStrategy.cpp   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dipu/torch_dipu/csrc_dipu/vendor/ascend/AscendCustomAllocatorStrategy.cpp b/dipu/torch_dipu/csrc_dipu/vendor/ascend/AscendCustomAllocatorStrategy.cpp
index de80e58ee6..79682dde9c 100644
--- a/dipu/torch_dipu/csrc_dipu/vendor/ascend/AscendCustomAllocatorStrategy.cpp
+++ b/dipu/torch_dipu/csrc_dipu/vendor/ascend/AscendCustomAllocatorStrategy.cpp
@@ -4,7 +4,7 @@ const static int ascend_init = []() {
   // – 内存大小向上对齐成32整数倍+32字节(m=ALIGN_UP[len,32]+32字节);
   // – 内存起始地址需满足64字节对齐(ALIGN_UP[m,64])。
   //  nbytes = align_64(1 * nbytes + 32);
-  static dipu::MemoryAlignmentStrategy memoryAlignStrategy(512, 1, 32);
+  static dipu::MemoryAlignmentStrategy memoryAlignStrategy(64, 1, 32);
   dipu::setMemoryAlignmentStrategy(&memoryAlignStrategy);
   return 0;
 }();