llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/ATen/Transforms/CMakeLists.txt

19 lines
316 B
CMake
Raw Normal View History

NFC: Re-organize ATen directory structure and fix warnings. * Still some more work to do on the Transforms tree to bring it in line with the others (will do that as I add things).
2020-10-23 05:13:26 +08:00
add_npcomp_conversion_library(NPCOMPATenPasses
NFC: Transition ATen passes to tablegen registration.
2020-10-23 08:12:01 +08:00
Passes.cpp
NFC: Re-organize ATen directory structure and fix warnings. * Still some more work to do on the Transforms tree to bring it in line with the others (will do that as I add things).
2020-10-23 05:13:26 +08:00
ATenLayerNamePass.cpp
ATenLoweringPass.cpp
ATenOpReport.cpp
ATenToStd.cpp
LivenessReport.cpp
Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. * 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.
2020-10-23 14:31:34 +08:00
RecognizeKernelsPass.cpp
NFC: Re-organize ATen directory structure and fix warnings. * Still some more work to do on the Transforms tree to bring it in line with the others (will do that as I add things).
2020-10-23 05:13:26 +08:00
ReturnEliminationPass.cpp
DEPENDS
MLIRATenToStdIncGen
NFC: Transition ATen passes to tablegen registration.
2020-10-23 08:12:01 +08:00
NPCOMPATenPassIncGen
LINK_LIBS PUBLIC
MLIRIR
MLIRPass
NFC: Re-organize ATen directory structure and fix warnings. * Still some more work to do on the Transforms tree to bring it in line with the others (will do that as I add things).
2020-10-23 05:13:26 +08:00
)