mirror of https://github.com/llvm/torch-mlir
510f226df2
* Two op interfaces, one for querying instance metadata and one for getting static data needed to construct an op from a generic form. * For torch.generic_kernel ops, metadata is splatted in during capture from Torch (it comes from the op registry, which will work for either device capture or graph import). * Moved the 'add' out of the generated set so I can experiment on it. It implements the TorchBuildableKernelOpInterface interface which provides its metadata. * The ATenRecognizeKernelsPass pass generically lowers from a torch.generic_kernel to recognized ops that implement the TorchBuildableKernelOpInterface, handling the various types of transformations that we allow at this stage. |
||
|---|---|---|
| .. | ||
| aten_add.mlir | ||
| aten_addmm.mlir | ||
| aten_as_strided.mlir | ||
| aten_batchnorm.mlir | ||
| aten_conv2d.mlir | ||
| aten_conv2d_back.mlir | ||
| aten_maxpool2d.mlir | ||
| aten_relu.mlir | ||
| aten_resA.mlir | ||
| lenet_fwd.mlir | ||
| recognize_aten_kernels.mlir | ||