Add NoMemoryEffect for ops with value semantics

2024-10-13 15:06:32 -05:00 · 2024-10-13 15:06:32 -05:00 · 896bcd3cf7
parent b176939808
commit 896bcd3cf7
2 changed files with 530 additions and 0 deletions
--- a/include/torch-mlir/Dialect/Torch/IR/GeneratedTorchOps.td
+++ b/include/torch-mlir/Dialect/Torch/IR/GeneratedTorchOps.td
--- a/projects/pt1/python/torch_mlir/jit_ir_importer/build_tools/torch_ods_gen.py
+++ b/projects/pt1/python/torch_mlir/jit_ir_importer/build_tools/torch_ods_gen.py
@ -255,6 +255,7 @@ def emit_op(
    traits += ["AllowsTypeRefinement"]
    if operator.has_value_semantics():
        traits += ["HasValueSemantics"]
+        traits += ["NoMemoryEffect"]
    if operator.is_readonly():
        traits += ["ReadOnly"]