• Lei Wang's avatar
    [Refactor] Remove small array reuse condition in shared memory allocation merging (#654) · 8205791d
    Lei Wang authored
    - Eliminated the condition that disabled the reuse of small arrays (const_nbits <= 32) in the `MergeSharedMemoryAllocations` function, allowing for more flexible memory management.
    - Added a comment in `OptimizeForTarget` to clarify the order of applying `MergeSharedMemoryAllocations` after `SplitHostDevice`, ensuring correct allocation site handling in device functions.
    8205791d
merge_shared_memory_allocations.cc 37.6 KB