pytorch/test/cpp/jit
Ailing Zhang 6ab2e7442d Autograd using torchscript (#14604)
Summary:
This PR enables autodiff to use the forward/backward graph compiled from python code, instead of using symbolic gradients(modifying the original graph directly).

We put the map in a separate .h file for now to wait for the native_functions.yaml and derivatives.yaml merge. This should ideally go into native_functions.yaml eventually.

This PR should be enough to unblock us for now, we can start writing gradients for aten functions in python.

Differential Revision: D13494635

Pulled By: ailzhang

fbshipit-source-id: f8d51a15243ac46afd09d930c573ccdfcd9fdaaf
2018-12-18 19:10:57 -08:00
..
CMakeLists.txt Move JIT tests to gtest (#12030) 2018-10-06 23:09:44 -07:00
gtest.cpp Reuse KernelSpec for FusionGroups with equivalent graphs (#14541) 2018-12-13 07:54:35 -08:00
no-gtest.cpp migrate subgraph slicing to use moveBefore/moveAfter (#13862) 2018-11-14 17:33:36 -08:00
tests.h Autograd using torchscript (#14604) 2018-12-18 19:10:57 -08:00