Marc Sun bc2478a472 fix (#3808)

2025-10-08 15:32:18 +02:00

..

deepspeed

xpu enablement on left cases (#3654)

2025-07-07 18:10:53 +02:00

fsdp

feat: allow mixed precision policy as dtype (#3751)

2025-09-08 23:29:20 +02:00

test_configs

ENH: Allow FSDP ignored modules to be regex (#3698)

2025-08-05 14:23:14 +02:00

test_samples

Introduce a pod-config command (#802)

2022-11-01 10:00:48 -04:00

tp

(Part 1) fix: make TP training compatible with new transformers (#3457)

2025-04-11 18:31:28 +02:00

__init__.py

Torchao float8 training (#3348)

2025-02-17 11:51:47 -05:00

test_accelerator.py

Fix tests (#3722)

2025-08-07 16:59:29 +02:00

test_big_modeling.py

enable test_dispatch_model_tied_weights_memory_with_nested_offload_cpu on xpu (#3569)

2025-05-15 11:40:55 +02:00

test_cli.py

Fix: typo makes tests fail (#3765)

2025-09-09 12:06:05 +02:00

test_compile.py

Fix tests (#3722)

2025-08-07 16:59:29 +02:00

test_cpu.py

Add test for ops and fix reduce (#1122)

2023-02-28 09:18:09 -05:00

test_data_loader.py

Fix seeding of new generator for multi GPU (#3459)

2025-03-28 12:48:05 -04:00

test_dataclasses.py

Protect import for device_mesh (#3742)

2025-08-22 15:44:56 +02:00

test_examples.py

fix (#3808)

2025-10-08 15:32:18 +02:00

test_fp8.py

Add support for TE MXFP8 recipe in accelerate (#3688)

2025-08-27 14:08:34 +02:00

test_grad_sync.py

HPU support (#3378)

2025-03-11 11:16:57 -04:00

test_hooks.py

enable 2 model hook ut cases on XPU (#3774)

2025-09-17 01:32:59 +02:00

test_imports.py

Do not import transformer_engine on import (#3056)

2024-08-28 09:06:13 -04:00

test_kwargs_handlers.py

HPU support (#3378)

2025-03-11 11:16:57 -04:00

test_launch.py

Feat: context parallel v2.0 (#3700)

2025-08-05 16:17:13 +02:00

test_load_checkpoint_and_dispatch_with_broadcast.py

Update Gaudi Runners (#3593)

2025-06-03 12:36:56 +02:00

test_logging.py

HPU support (#3378)

2025-03-11 11:16:57 -04:00

test_memory_utils.py

“Stop Halving My Batch!” · Default back-off 0.5 → 0.9 (#3684)

2025-07-16 12:32:46 +02:00

test_metrics.py

HPU support (#3378)

2025-03-11 11:16:57 -04:00

test_modeling_utils.py

warn for invalid keys (#3613)

2025-07-16 12:23:41 +02:00

test_multidevice.py

switch XPU ccl backend to torch-builtin xccl in test_zero3_integration (#3773)

2025-09-18 11:50:32 +02:00

test_offload.py

Make test assertions more idiomatic (#2420)

2024-02-13 14:23:18 -05:00

test_optimizer.py

HPU support (#3378)

2025-03-11 11:16:57 -04:00

test_quantization.py

enable test_dispatch_model_tied_weights_memory_with_nested_offload_cpu on xpu (#3569)

2025-05-15 11:40:55 +02:00

test_sagemaker.py

Add copyright + some ruff lint things (#2523)

2024-03-04 09:14:31 -05:00

test_scheduler.py

Bump ruff to 0.11.2 (#3471)

2025-04-01 11:57:06 +02:00

test_state_checkpointing.py

Bump ruff to 0.11.2 (#3471)

2025-04-01 11:57:06 +02:00

test_tpu.py

Remove unnecessary env=os.environ.copy()s (#2449)

2024-03-06 06:36:56 -05:00

test_tracking.py

fix (#3808)

2025-10-08 15:32:18 +02:00

test_utils.py

Bye bye torch <2 (#3331)

2025-01-09 12:11:08 -05:00

xla_spawn.py

Bump ruff to 0.11.2 (#3471)

2025-04-01 11:57:06 +02:00