Skip to content

Tags: sywangyi/pytorch

Tags

ciflow/trunk/90856

Toggle ciflow/trunk/90856's commit message
Update on "Revert "Revert "[functorch] Refactor life handle storage (p…

…ytorch#90317)"""

Adds the fix for -Wsign-compare.

See original PR (pytorch#90317) for
commit message

[ghstack-poisoned]

ciflow/trunk/90825

Toggle ciflow/trunk/90825's commit message
Update on "Deeply rework WeakIdKeyDictionary"

In the prior patch, I just YOLOed a mutable mapping implementation.
Many edge cases were not handled correctly.  In this PR, I just
copy paste the WeakKeyDictionary from CPython and the hacked it up
to use WeakIdRef instead of weakref.ref.  You can see each line
I changed with the comment CHANGED; there aren't many.

Being exactly API compatible with WeakKeyDictionary means I can also
rob all of the tests from CPython, which I also did for
test/test_weak.py

How to review?  You could either try taking the delta from CPython
(recommended), or review everything from scratch (not recommended).
Can post diff representing delta on request.

Signed-off-by: Edward Z. Yang <ezyangfb.com>

[ghstack-poisoned]

ciflow/trunk/90738

Toggle ciflow/trunk/90738's commit message
Update on "[inductor] Rewrite Triton templates + epilogue fusion"

Differential Revision: [D41474473](https://our.internmc.facebook.com/intern/diff/D41474473/)

**NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D41474473/)!

cc mlazos soumith voznesenskym yanboliang penguinwu anijain2305 EikanWang jgong5 Guobing-Chen chunyuan-w XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx peterbell10 desertfire

[ghstack-poisoned]

ciflow/trunk/90628

Toggle ciflow/trunk/90628's commit message
Update on "Don't put tracing state on Tensor"

Fixes pytorch#89626

Signed-off-by: Edward Z. Yang <ezyangfb.com>

[ghstack-poisoned]

ciflow/trunk/90575

Toggle ciflow/trunk/90575's commit message
Update on "Handle tensor default func args when inlining"


    Handle tensor default func/method args when inlining
    
    Previously, when inlining a function, its default arguments
    were only wrapped with VariableTrackers if non-tensor. Now,
    tensor default args are also handled by adding them to the
    parent InstructionTranslator as an attribute.
    
    - also patches up a missing source in nnmodule call_function,
      needed to properly guard on a default arg in its methods
    - adds new 'DefaultsSource' type which guards either a `__defaults__`
      or `__kwdefaults__` entry on a function

Fixes pytorch#90361  https://github.com/pytorch/torchdynamo/issues/1968    

cc mlazos soumith voznesenskym yanboliang penguinwu anijain2305 EikanWang jgong5 Guobing-Chen chunyuan-w XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire

[ghstack-poisoned]

ciflow/trunk/90566

Toggle ciflow/trunk/90566's commit message
Make test_binary_shape_functions test ops

ciflow/trunk/90265

Toggle ciflow/trunk/90265's commit message
Update on "[inductor] add conv_transpose2d unary fusion for cpu in in…

…ference mode"



An FX transformation is added to fuse ConvTranspose2d with eltwise OPs in torchinductor for CPU in inference mode, following the implementation in pytorch#87063.

The fusion OP is implemented in pytorch#90264 and will be treated as an extern kernel call in torchinductor.

The fusion of ConvTranspose2d with the below OPs is supported:

- relu
- sigmoid
- tanh
- hardswish
- leaky_relu
- hardtanh
- gelu


cc mlazos soumith voznesenskym yanboliang penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx peterbell10 desertfire

[ghstack-poisoned]

ciflow/trunk/90264

Toggle ciflow/trunk/90264's commit message
Update on "add conv_transpose2d pointwise(unary) fusion kernel"


This PR adds `torch.ops.mkldnn._convolution_transpose_pointwise` that supports ConvTranspose fusion with the below unary pointwise OPs:

- relu
- sigmoid
- tanh
- hardswish
- leaky_relu
- hardtanh
- gelu


cc VitalyFedyunin jgong5 mingfeima XiaobingSuper sanchitintel ashokei jingxu10

[ghstack-poisoned]

ciflow/trunk/90257

Toggle ciflow/trunk/90257's commit message
Update on "[Composable API] `replicate`: cleanup _ddp.py"

[ghstack-poisoned]

ciflow/trunk/90247

Toggle ciflow/trunk/90247's commit message
Rename pooling_avg to pooling_avg_exclude_padding to align with onedn…

…n 3.0