Add support for Kokkos::parallel_for in the fwd mode
#4260
| Job | Run time |
|---|---|
| 2m 45s | |
| 2m 16s | |
| 2m 24s | |
| 2m 30s | |
| 2m 45s | |
| 6m 24s | |
| 4m 17s | |
| 6m 14s | |
| 3m 55s | |
| 6m 24s | |
| 4m 52s | |
| 4m 54s | |
| 5m 7s | |
| 5m 30s | |
| 5m 46s | |
| 5m 9s | |
| 20m 31s | |
| 6m 17s | |
| 8m 0s | |
| 1m 58s | |
| 2m 49s | |
| 2m 35s | |
| 2m 31s | |
| 3m 17s | |
| 2m 34s | |
| 3m 11s | |
| 2m 45s | |
| 2m 45s | |
| 2m 29s | |
| 3m 7s | |
| 2m 59s | |
| 3m 0s | |
| 2m 56s | |
| 2m 53s | |
| 2m 41s | |
| 2m 54s | |
| 2m 49s | |
| 2m 55s | |
| 3m 50s | |
| 3m 16s | |
| 3m 16s | |
| 2m 45s | |
| 2m 51s | |
| 2m 46s | |
| 2m 45s | |
| 2m 51s | |
| 2m 57s | |
| 3m 45s | |
| 2m 53s | |
| 2m 46s | |
| 2m 44s | |
| 2m 49s | |
| 2m 54s | |
| 2m 41s | |
| 2m 58s | |
| 3m 3s | |
| 3m 29s | |
| 2m 55s | |
| 3m 23s | |
| 3m 11s | |
| 2m 54s | |
| 3m 5s | |
| 3m 25s | |
| 3m 1s | |
| 3m 6s | |
| 2m 50s | |
| 2m 52s | |
| 3m 6s | |
| 3m 16s | |
| 4m 4s | |
| 2m 52s | |
| 2m 56s | |
| 3m 4s | |
| 3m 6s | |
| 3m 1s | |
| 2m 52s | |
| 2m 57s | |
| 3m 26s | |
| 3m 31s | |
| 3m 36s | |
| 3m 26s | |
| 3m 18s | |
| 3m 28s | |
| 2m 33s | |
| 5h 1m 41s |