Tags: sywangyi/pytorch
Tags
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]
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]
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]
Update on "Don't put tracing state on Tensor" Fixes pytorch#89626 Signed-off-by: Edward Z. Yang <ezyangfb.com> [ghstack-poisoned]
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]
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]
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]
Update on "[Composable API] `replicate`: cleanup _ddp.py" [ghstack-poisoned]
Rename pooling_avg to pooling_avg_exclude_padding to align with onedn… …n 3.0
PreviousNext