Reapply 418a6d6 Fix nvptx_target_teams_distribute_parallel_for_simd_codegen failure

Changes were accidentally dropped after merge conflict resolution.

Author: vmaksimo