torch-mlir/frontends/pytorch/test
Stella Laurenzo 0c73c535d6 Capture backward conv and copy_ kernels.
* This is sufficient to capture the forward and backward pass and gradients of a convolutional model with an nllloss.
* As with the forward conv, the backward conv is a special case wrapped in an enigma on the PyTorch side. There aren't many like it, so special casing is just what we do.
* When I traced this, I found that the copy_ op is not yet boxing compatible so I had to map it manually. If there are many more like this, I'll probably do something a bit more clever to reduce duplication.
* This exposes new signature patterns that will need to be handled by the ATen lowering. Will take care of that next: It will be nice to have an e2e of a non-trivial case with full gradients.
* Fixes #97.
2020-10-30 22:59:26 -07:00
..
acap_export Capture backward conv and copy_ kernels. 2020-10-30 22:59:26 -07:00
acap_regression Update test configuration to import mlir from LLVM install location. 2020-10-12 15:25:07 -07:00
c10_dispatch Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 2020-10-26 20:31:45 -07:00
CMakeLists.txt More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00
extension_coexistence.py Update test configuration to import mlir from LLVM install location. 2020-10-12 15:25:07 -07:00
lit.cfg.py Port prior acap export tests to new dispatcher based versions. 2020-10-13 16:37:46 -07:00
lit.site.cfg.py.in Move tests around so we can have dedicated tests for the c10 dispatcher. 2020-09-24 18:28:06 -07:00