Make throughput benchmark compatible with both shallow and deep storage stacks