torch-mlir/include/npcomp/Dialect/Torch/IR
Stella Laurenzo 58adb6bd8e Work around various PyTorch issues in support of convolution.
* Enables the conv2d fwd test and ResA (which are both small).
* Deletes resnet18 and vgg, which both run but generate output that crashes FileCheck and lit (or at least makes them take an eternity).
2020-10-21 12:44:31 -07:00
..
CMakeLists.txt Add boilerplate for Torch dialect. 2020-09-28 15:26:17 -07:00
TorchBase.td Work around various PyTorch issues in support of convolution. 2020-10-21 12:44:31 -07:00
TorchDialect.h Add boilerplate for Torch dialect. 2020-09-28 15:26:17 -07:00
TorchOps.h Add boilerplate for Torch dialect. 2020-09-28 15:26:17 -07:00
TorchOps.td Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00