csarofeen / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
http://pytorch.org
Other
26 stars 7 forks source link

Lower unroll cleanup, make it support IfThenElse #2496

Closed zasdfgbnm closed 1 year ago

zasdfgbnm commented 1 year ago

Split from https://github.com/csarofeen/pytorch/pull/2488

zasdfgbnm commented 1 year ago

Reopen PR to trigger CI