pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

andrewor14 dd5df002b9 [pt2e][quant] Make move_exported_model_to_train/eval idempotent (#142239 )

Summary: Before we would recompile the model unnecessarily even
if the model is already in the desired mode. For training
frameworks that assume `model.train()` is idempotent and calls
this before every single training step, this led to a bunch of
tiny graphs and poor performance. This commit makes these calls
no-ops if we're already in the target train/eval mode.

Test Plan:
python test/test_quantization -k TestQuantizePT2E.test_allow_exported_model_train_eval_idempotent
Pull Request resolved: https://github.com/pytorch/pytorch/pull/142239
Approved by: https://github.com/jerryzh168

2024-12-09 16:50:20 +00:00

ao_migration

Enable UFMT on all of test/quantization/ao_migration &bc (#123994 )

2024-04-13 06:36:10 +00:00

Fix failures when default is flipped for weights_only (#127627 )

2024-08-16 00:22:43 +00:00

core

[ci, 3.13] disable some quantization tests affected by numpy 2.1 overflow error (#141621 )

2024-12-05 00:24:29 +00:00

eager

Replace clone.detach with detach.clone (#140264 )

2024-11-13 07:01:02 +00:00