pytorch

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

Files

Shangdi Yu 02990fe36b Populate nn.module.stack in _fuse_conv_bn_qat (#141400 )

Summary:
Populate nn.module.stack in _fuse_conv_bn_qat for replacement nodes that correspond to a `get_attr` node in the original graph.

In new training ir , `get_attr` nodes don't have `nn_module_stack` in node meta anymore (because the get_attr nodes are de-duplicated, so one get_attr node can potential have users in different module stacks).

We populate it by checking if "conv_input" or "conv_weight" replacement node has nn_module_stack. If not, we copy it from the conv node.

Test Plan:
CI

```
buck run fbcode//caffe2/test:quantization_pt2e -- -r test_preserve_nn_module_stack
```

Differential Revision: D66393517

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141400
Approved by: https://github.com/angelayi

2024-11-25 23:41:28 +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

Revert "Add option to split Linear gates for Quantizable LSTM into separate ops (#140868 )"

2024-11-22 15:54:05 +00:00

eager

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

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