pytorch
adb76ef5 - Expose API for backward execution order (#87507)

Commit

2 years ago

Expose API for backward execution order (#87507) In this PR: - graph_task stores graph roots on construction so that we can later traverse through the graph - before the nodes are returned, they needed to be converted from raw_ptr to shared_ptr, and this should be OK because the graph is guaranteed to be alive Pull Request resolved: https://github.com/pytorch/pytorch/pull/87507 Approved by: https://github.com/albanD

Author

soulitzer

Committer

pytorchmergebot

Parents

926827b8

Files6

test
- test_autograd.py
torch/csrc
- Module.cpp
- autograd
  - engine.cpp
  - function.h
  - graph_task.h
- distributed/autograd/engine
  - dist_engine.cpp

pytorch adb76ef5 - Expose API for backward execution order (#87507)

pytorch
adb76ef5 - Expose API for backward execution order (#87507)