PR #60 Work with latest PyTorch

Work with latest PyTorch #60

asuhan merged 15 commits into master from asuhan/upgrade_pytorch

asuhan requested a review from

dlibenzi 7 years ago

asuhan force pushed from 2d661315 to 62a960b5 7 years ago

asuhan force pushed from 62a960b5 to 08f9fc60 7 years ago

asuhan force pushed from 08f9fc60 to 2d2c0318 7 years ago

dlibenzi requested changes on 2018-12-13

Initial changes for latest PyTorch

67a6cdf0

Keep track of evaluated aten::size

e1ebd434

Add pass to remove in-place operations on output parameters

a135f4c8

Add pass to replace inplace operators with non-inplace counterparts

d43b3335

Remove nearly all dependencies on PyTorch trace shapes

3315d799

Add and use XlaHelpers::SizesOfXlaOp helper

7edd227f

asuhan force pushed from 2d2c0318 to 07946928 7 years ago

Address some of the logging PR feedback

6ac04791

Adjust signature of RemoveUnusedForwardOutputs

e09dcb26

asuhan force pushed from 07946928 to e09dcb26 7 years ago

asuhan force pushed from cb5d590f to 414aca8e 7 years ago

Additional cleanups from PR feedback

7ee5b904

Store gradient descriptor in XlaModule

1f79b18d

Additional XlaModule fixes per review feedback

eb9212a0

dlibenzi requested changes on 2018-12-13

asuhan force pushed from 414aca8e to eb9212a0 7 years ago

Even more cleanups from review feedback

6f03fb2f

Fix the construction of argument list for backward

c7fea96a

dlibenzi requested changes on 2018-12-14

asuhan force pushed from 75131d63 to a2a5d391 7 years ago

Address more review feedback

447ce022

asuhan force pushed from a2a5d391 to 447ce022 7 years ago

dlibenzi requested changes on 2018-12-16

Simplify APIs and remove redundant logging

eb3f6383

dlibenzi approved these changes on 2018-12-17

asuhan merged 0288df9a into master 7 years ago

Reviewers

dlibenzi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

xla Work with latest PyTorch #60 Merged

Work with latest PyTorch #60

xla
Work with latest PyTorch
#60

Merged