Add two new public utilities to accelerate.utils — compile regions and has compiled regions — and extend…
Add two new public utilities to accelerate.utils — compile regions and has compiled regions — and extend…: a task in HF ML Tasksmith (Harbor dataset). compile regions(module, compile kwargs) applies regional compilation: - A torch.nn.ModuleList where all elements share the same class → each…
The task
**`compile_regions(module, **compile_kwargs)`** applies regional compilation: - A `torch.nn.ModuleList` where **all** elements share the same class → each element is compiled individually via `torch.compile(..., **compile_kwargs)`. - A `torch.nn.ModuleList` with **mixed** element types → the entire list is compiled…
Part of FineEnvs/HF_ML_Tasksmith.