add support for thread team based loop

We're starting having more and more thread team based stuff with kokkos, and some ugly code associated with it.