Support for basic list comprehensions (#17267)
authorNikolay Korovaiko <korovaikon@gmail.com>
Fri, 22 Mar 2019 22:22:23 +0000 (15:22 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 22 Mar 2019 22:25:13 +0000 (15:25 -0700)
commit2ad2b2c7b10aa00822164c61845d910917d91ff4
tree55ddb01f7a883e00593a8af72f457940a33682c2
parente20894fce5592b87170da678019bb798893a42e2
Support for basic list comprehensions (#17267)

Summary:
Supports the following syntax:
```
        torch.jit.script
        def comp(l):
            # type: (List[float]) -> List[float]

            n = [x * 3 for x in l]
            return n
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17267

Differential Revision: D14581119

Pulled By: Krovatkin

fbshipit-source-id: 6fd091a8a9ab607386ac58fda6ad88bf8aea380e
test/test_jit.py
torch/csrc/jit/script/compiler.cpp
torch/csrc/jit/script/lexer.cpp
torch/csrc/jit/script/lexer.h
torch/csrc/jit/script/parser.cpp
torch/csrc/jit/script/python_tree_views.cpp
torch/csrc/jit/script/tree_views.h
torch/jit/frontend.py