mirror of https://github.com/llvm/torch-mlir
bb47b36eac
This PR adds an `AllowedInModuleInitializer` trait to keep track of ops that are permitted in the module initializer. We have a handful of such ops that are produced by the IValue importer, and so this change avoids maintaining a list of ops in `TorchOps.cpp` that could lead to spurious merge conflicts, and help us integrate torch-mlir in our downstream compiler better. Please let me know if you'd prefer a better name for the trait itself. Feedback is welcome! |
||
|---|---|---|
| .. | ||
| Torch | ||
| TorchConversion | ||
| CMakeLists.txt | ||