fix no need change

DeepLink-org · Apr 9, 2024 · 33bcd11 · 33bcd11
1 parent a580fbf
commit 33bcd11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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;
 }();