xla
Work with latest PyTorch
#60
Merged

Commits
  • Initial changes for latest PyTorch
    asuhan committed 7 years ago
  • Keep track of evaluated aten::size
    asuhan committed 7 years ago
  • Add pass to remove in-place operations on output parameters
    asuhan committed 7 years ago
  • Add pass to replace inplace operators with non-inplace counterparts
    asuhan committed 7 years ago
  • Remove nearly all dependencies on PyTorch trace shapes
    asuhan committed 7 years ago
  • Add and use XlaHelpers::SizesOfXlaOp helper
    asuhan committed 7 years ago
  • Address some of the logging PR feedback
    asuhan committed 7 years ago
  • Adjust signature of RemoveUnusedForwardOutputs
    asuhan committed 7 years ago
  • Additional cleanups from PR feedback
    asuhan committed 7 years ago
  • Store gradient descriptor in XlaModule
    asuhan committed 7 years ago
  • Additional XlaModule fixes per review feedback
    asuhan committed 7 years ago
  • Even more cleanups from review feedback
    asuhan committed 7 years ago
  • Fix the construction of argument list for backward
    asuhan committed 7 years ago
  • Address more review feedback
    asuhan committed 7 years ago
  • Simplify APIs and remove redundant logging
    asuhan committed 7 years ago
Loading