()
| 52 | |
| 53 | @contextmanager |
| 54 | def switch_sequence_parallel_mode(): |
| 55 | prev_mode = gpc.config.parallel.sequence_parallel |
| 56 | try: |
| 57 | gpc.config.parallel.sequence_parallel = False |
| 58 | yield |
| 59 | finally: |
| 60 | gpc.config.parallel.sequence_parallel = prev_mode |
| 61 | |
| 62 | |
| 63 | def evaluate_on_val_dls( |
no outgoing calls
no test coverage detected