whitelist accelerate (#1999)

## Changes whitelist accelerate ### Linked issues Progresses #1901 ### Functionality None ### Tests - [x] manually tested Co-authored-by: Eric Vergnaud <[email protected]>
databrickslabs · Jul 2, 2024 · e3a5e31 · e3a5e31
1 parent 70d0846
commit e3a5e31
Showing 1 changed file with 80 additions and 0 deletions.
diff --git a/src/databricks/labs/ucx/source_code/known.json b/src/databricks/labs/ucx/source_code/known.json
@@ -2,6 +2,86 @@
   "absl-py": {
     "absl": []
   },
+  "accelerate": {
+    "accelerate": [],
+    "accelerate.accelerator": [],
+    "accelerate.big_modeling": [],
+    "accelerate.checkpointing": [],
+    "accelerate.commands": [],
+    "accelerate.commands.accelerate_cli": [],
+    "accelerate.commands.config": [],
+    "accelerate.commands.config.cluster": [],
+    "accelerate.commands.config.config": [],
+    "accelerate.commands.config.config_args": [],
+    "accelerate.commands.config.config_utils": [],
+    "accelerate.commands.config.default": [],
+    "accelerate.commands.config.sagemaker": [],
+    "accelerate.commands.config.update": [],
+    "accelerate.commands.env": [],
+    "accelerate.commands.estimate": [],
+    "accelerate.commands.launch": [],
+    "accelerate.commands.menu": [],
+    "accelerate.commands.menu.cursor": [],
+    "accelerate.commands.menu.helpers": [],
+    "accelerate.commands.menu.input": [],
+    "accelerate.commands.menu.keymap": [],
+    "accelerate.commands.menu.selection_menu": [],
+    "accelerate.commands.merge": [],
+    "accelerate.commands.test": [],
+    "accelerate.commands.tpu": [],
+    "accelerate.commands.utils": [],
+    "accelerate.data_loader": [],
+    "accelerate.hooks": [],
+    "accelerate.inference": [],
+    "accelerate.launchers": [],
+    "accelerate.local_sgd": [],
+    "accelerate.logging": [],
+    "accelerate.memory_utils": [],
+    "accelerate.optimizer": [],
+    "accelerate.scheduler": [],
+    "accelerate.state": [],
+    "accelerate.test_utils": [],
+    "accelerate.test_utils.examples": [],
+    "accelerate.test_utils.scripts": [],
+    "accelerate.test_utils.scripts.external_deps": [],
+    "accelerate.test_utils.scripts.external_deps.test_checkpointing": [],
+    "accelerate.test_utils.scripts.external_deps.test_metrics": [],
+    "accelerate.test_utils.scripts.external_deps.test_peak_memory_usage": [],
+    "accelerate.test_utils.scripts.external_deps.test_performance": [],
+    "accelerate.test_utils.scripts.external_deps.test_pippy": [],
+    "accelerate.test_utils.scripts.external_deps.test_zero3_integration": [],
+    "accelerate.test_utils.scripts.test_cli": [],
+    "accelerate.test_utils.scripts.test_distributed_data_loop": [],
+    "accelerate.test_utils.scripts.test_merge_weights": [],
+    "accelerate.test_utils.scripts.test_notebook": [],
+    "accelerate.test_utils.scripts.test_ops": [],
+    "accelerate.test_utils.scripts.test_script": [],
+    "accelerate.test_utils.scripts.test_sync": [],
+    "accelerate.test_utils.testing": [],
+    "accelerate.test_utils.training": [],
+    "accelerate.tracking": [],
+    "accelerate.utils": [],
+    "accelerate.utils.bnb": [],
+    "accelerate.utils.constants": [],
+    "accelerate.utils.dataclasses": [],
+    "accelerate.utils.deepspeed": [],
+    "accelerate.utils.environment": [],
+    "accelerate.utils.fsdp_utils": [],
+    "accelerate.utils.imports": [],
+    "accelerate.utils.launch": [],
+    "accelerate.utils.megatron_lm": [],
+    "accelerate.utils.memory": [],
+    "accelerate.utils.modeling": [],
+    "accelerate.utils.offload": [],
+    "accelerate.utils.operations": [],
+    "accelerate.utils.other": [],
+    "accelerate.utils.random": [],
+    "accelerate.utils.rich": [],
+    "accelerate.utils.torch_xla": [],
+    "accelerate.utils.tqdm": [],
+    "accelerate.utils.transformer_engine": [],
+    "accelerate.utils.versions": []
+  },
   "aiohttp": {
     "aiohttp": []
   },