add collapse() to omp directives