feat: auto-enable torch.compile/AMP on CUDA via apply_hardware_defaults - #23

Merged

Adithya-Thonse merged 22 commits into

TexasInstruments:mainfrom

musicalplatypus:pr/hardware-defaults

Aug 6, 2026

Commits on Jul 29, 2026

fix(ci): trigger on tinyml-tinyverse changes too

t5fkg8d44d-beep
and
claude
committed
c94f6dc
fix: validate torch.compile with a warmup pass so failures actually fall back to eager mode

t5fkg8d44d-beep
and
claude
committed
6b8d3c5
fix: restore training mode unconditionally after torch.compile warmup

t5fkg8d44d-beep
and
claude
committed
6f892fd
fix: pass input_shape to compile_model_if_enabled at all 4 call sites

t5fkg8d44d-beep
and
claude
committed
3fd2bed
docs: clarify compile warmup only validates one graph variant, not the training-mode graph
t5fkg8d44d-beep
committed
57ff586
fix: unwrap torch.compile wrapper before ONNX/TorchScript export
t5fkg8d44d-beep
committed
d0f420a
fix: recursively unwrap compiled submodules before export, not just top-level

t5fkg8d44d-beep
and
claude
committed
4ff3e62
fix: skip torch.compile when quantization is enabled — FX tracing can't handle a compiled model

t5fkg8d44d-beep
and
claude
committed
7f9c0f7
test: strengthen skip-compile tests with torch.compile mock and INFO-log assertion
t5fkg8d44d-beep
committed
106e2f9
fix: strip torch.compile wrapper prefix from saved checkpoints

t5fkg8d44d-beep
and
claude
committed
e0d03c3
fix: strip _orig_mod prefix from EMA checkpoint keys too, symmetric save+load

t5fkg8d44d-beep
and
claude
committed
1ed7eff
test: add asymmetric compiled-save/uncompiled-load EMA case — the real production shape
t5fkg8d44d-beep
committed
2268e22
fix(ci): run tinyml-tinyverse tests

t5fkg8d44d-beep
and
claude
committed
aff21fd
fix: strip _orig_mod prefix in load_weights() as defense-in-depth

t5fkg8d44d-beep
and
claude
committed
0326b50
fix: resume_from_checkpoint unsafe deserialization + old-checkpoint compat

t5fkg8d44d-beep
and
claude
committed
898bc4a
fix(tests): use real argparse.Namespace for checkpoint round-trip tests

t5fkg8d44d-beep
and
claude
committed
0179c2d
feat: add apply_hardware_defaults — auto-enable compile/AMP on CUDA

t5fkg8d44d-beep
and
claude
committed
a2c05eb
feat: wire apply_hardware_defaults into timeseries init_params
t5fkg8d44d-beep
committed
edbabba
fix(tests): rewrite native_amp override test to use production call pattern

t5fkg8d44d-beep
and
claude
committed
85873b3
fix: init_params crashes on non-dict first argument

t5fkg8d44d-beep
and
claude
committed
e27dd78

Commits on Aug 4, 2026

fix: train.py post-loop bugs in timeseries classification and forecasting

t5fkg8d44d-beep
and
claude
committed
9c2f625

Commits on Aug 6, 2026

Merge remote-tracking branch 'upstream/main' into pr/hardware-defaults
t5fkg8d44d-beep
committed
b7814cf