[OpenMP][FIX] Ensure recording works properly w/ late allocations

GitOrigin-RevId: 41566fb8520a1284678d5a344632930b0d3d1fbf
1 file changed