projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 326c891) | patch
Add support for simpler for-in-list + tests (#16726)
authorNikolay Korovaiko <korovaikon@gmail.com>
Fri, 15 Feb 2019 19:38:26 +0000 (11:38 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 15 Feb 2019 19:41:20 +0000 (11:41 -0800)
commit82b269060c6937f3deee1a57e6db60dd41f27a61
treeab452daa24067eff4edc10d25fa71838d447958dtree | snapshot
parent326c891d32e3a786b350c2f08bc296106d73739ccommit | diff
Add support for simpler for-in-list + tests (#16726)

Summary:
This PR add supports for simpler for-in-list loops such as the example below:

```python
torch.ji.python
def sum_list(a):
    # type: (List[int]) -> int
    sum = 0
    for i in a:
        sum += i

    return sum
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16726

Differential Revision: D14070007

Pulled By: ezyang

fbshipit-source-id: b4d971ee647729a6caa3099ceac34ec5c4f143de
test/test_jit.py diff | blob | history
torch/csrc/jit/script/compiler.cpp diff | blob | history
Domain: Machine Learning / ML Framework;