未验证 提交 3ca4bc10 编写于 作者: L Leo Chen 提交者: GitHub

[NPU] fix allocator min chunk size (#31632)

上级 11f78877
...@@ -163,8 +163,11 @@ size_t NPUInitAllocSize() { return NPUAllocSize(/* realloc = */ false); } ...@@ -163,8 +163,11 @@ size_t NPUInitAllocSize() { return NPUAllocSize(/* realloc = */ false); }
size_t NPUReallocSize() { return NPUAllocSize(/* realloc = */ true); } size_t NPUReallocSize() { return NPUAllocSize(/* realloc = */ true); }
size_t NPUMinChunkSize() { size_t NPUMinChunkSize() {
// Allow to allocate the minimum chunk size is 256 bytes. // NOTE(zhiqiu): It seems the min chunk size should be 512 on NPU,
return 1 << 8; // though no document specify that explicitly.
// See https://gitee.com/zhiqiuchen/Ascend/tree/master/test_reduce_sum_d for
// details.
return 1 << 9;
} }
size_t NPUMaxChunkSize() { size_t NPUMaxChunkSize() {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册