The raw speed_mode string was forwarded to _enable_fp16_accumulation, so a case-variant like MAX failed the speed_mode != SPEED_MAX check and wrongly disabled fp16 accumulation on float16 pipelines. Forward the normalized mode and cover the case-insensitive path in the test. |
||
|---|---|---|
| .. | ||
| data_recipe | ||
| export | ||
| inference | ||
| rag | ||
| training | ||
| __init__.py | ||
| _torchao_stub.py | ||
| import_guards.py | ||
| tool_healing.py | ||