Skip to content

Commit 8963b53

Browse files
committed
Update base for Update on "Support of implicit fallback"
(Split out the large PR #46) Support the implicit replication fallback startegy. How to use Implicit replication fallback: ```python from autoparallel.dtensor_util import strategy_pool with strategy_pool.replicate_for_unsupported_operators(): ... # (missing ops will use replicated strategy if possible) ``` Note: StrategyPool reuses the _op_dispatcher.sharding_propagator.op_strategy_funcs/op_to_rules/op_to_schema_info by reference now. [ghstack-poisoned]
2 parents 8df62c4 + 22e663f commit 8963b53

17 files changed

+1079
-1192
lines changed

.pre-commit-config.yaml

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,12 @@ repos:
3434
additional_dependencies: [toml]
3535
args: ["--profile", "black"]
3636

37-
- repo: https://github.com/pre-commit/mirrors-mypy
38-
rev: 'v1.10.0'
37+
- repo: local
3938
hooks:
4039
- id: mypy
40+
name: mypy
41+
entry: mypy
42+
language: system
43+
types: [python]
44+
exclude: (docs|examples)
45+
args: ["--ignore-missing-imports", "--scripts-are-modules", "--pretty"]

0 commit comments

Comments
 (0)