(size, group)
| 23 | |
| 24 | |
| 25 | def set_context_parallel_group(size, group): |
| 26 | global _CONTEXT_PARALLEL_GROUP |
| 27 | global _CONTEXT_PARALLEL_SIZE |
| 28 | _CONTEXT_PARALLEL_GROUP = group |
| 29 | _CONTEXT_PARALLEL_SIZE = size |
| 30 | |
| 31 | |
| 32 | def initialize_context_parallel(context_parallel_size): |
no outgoing calls
no test coverage detected