Improve configs - the rest! (#17562)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
2025-05-09 23:18:44 +01:00
parent 7e3571134f
commit 4b2ed7926a
14 changed files with 456 additions and 340 deletions
--- a/tests/distributed/test_sequence_parallel.py
+++ b/tests/distributed/test_sequence_parallel.py
@@ -206,7 +206,7 @@ def _compare_sp(
        'compile_sizes': [4, 8],
        'splitting_ops': [],
        'pass_config': {
-            'enable_sequence_parallism': sp_enabled,
+            'enable_sequence_parallelism': sp_enabled,
            'enable_noop': True,
            'enable_fusion': True,
        },
@@ -223,7 +223,7 @@ def _compare_sp(
        "--distributed-executor-backend",
        distributed_backend,
        "--compilation_config",
-        str(compilation_config),
+        json.dumps(compilation_config),
    ]

    tp_env = {