torch-mlir/lib/Dialect/ATen/IR/CMakeLists.txt

add_npcomp_dialect_library(NPCOMPATenDialect
  ATenDialect.cpp
  ATenDialectOpStats.cpp

  ADDITIONAL_HEADER_DIRS
  ${PROJECT_SOURCE_DIR}/include/npcomp/Dialect/ATen

  DEPENDS
  MLIRATenIncGen
  #MLIRATenEnumsIncGen
  MLIRATenOpInterfacesIncGen
  MLIRTorchOpInterfacesIncGen
  #MLIRATenToStdIncGen

  LINK_LIBS PUBLIC
  MLIRPass
  MLIRTransformUtils
)
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_dialect_library(NPCOMPATenDialect`
			`ATenDialect.cpp`
			`ATenDialectOpStats.cpp`

			`ADDITIONAL_HEADER_DIRS`
			`${PROJECT_SOURCE_DIR}/include/npcomp/Dialect/ATen`

			`DEPENDS`
			`MLIRATenIncGen`
			`#MLIRATenEnumsIncGen`
			`MLIRATenOpInterfacesIncGen`
Add initial TorchScript module importer It turns out that this was easiest to structure as a general IValue importer, since torch module are just one of the possible IValue's. We import the IValue object graph in a braindead fashion into basicpy ops and a new `torch.nn_module` op that is used to model the attributes/methods of a torch::jit::Module IValue. See `Torch/ops.mlir` for an example, and also check out the .py import tests in `frontends/pytorch/test/module_import`. As part of this change, a few housekeeping tasks: - extract some helpers from graph_importer.cpp - more helpers around the C API - misc touchups 2021-01-28 08:35:44 +08:00			`MLIRTorchOpInterfacesIncGen`
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			`#MLIRATenToStdIncGen`

			`LINK_LIBS PUBLIC`
			`MLIRPass`
			`MLIRTransformUtils`
			`)`