We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 11f7887 commit 3ca4bc1Copy full SHA for 3ca4bc1
paddle/fluid/platform/npu_info.cc
@@ -163,8 +163,11 @@ size_t NPUInitAllocSize() { return NPUAllocSize(/* realloc = */ false); }
163
size_t NPUReallocSize() { return NPUAllocSize(/* realloc = */ true); }
164
165
size_t NPUMinChunkSize() {
166
- // Allow to allocate the minimum chunk size is 256 bytes.
167
- return 1 << 8;
+ // NOTE(zhiqiu): It seems the min chunk size should be 512 on NPU,
+ // though no document specify that explicitly.
168
+ // See https://gitee.com/zhiqiuchen/Ascend/tree/master/test_reduce_sum_d for
169
+ // details.
170
+ return 1 << 9;
171
}
172
173
size_t NPUMaxChunkSize() {
0 commit comments