[Mlir-commits] [mlir] 02eac66 - Improve invalid-ir-print-after-failure.mlir to show the effects of -mlir-print-assume-verified (NFC)

[Mehdi Amini]

Author: Mehdi Amini
Date: 2022-04-20T20:40:41Z
New Revision: 02eac667ed2dbf6eca4bc1195600eb0d3ecaff77

URL: https://github.com/llvm/llvm-project/commit/02eac667ed2dbf6eca4bc1195600eb0d3ecaff77
DIFF: https://github.com/llvm/llvm-project/commit/02eac667ed2dbf6eca4bc1195600eb0d3ecaff77.diff

LOG: Improve invalid-ir-print-after-failure.mlir to show the effects of -mlir-print-assume-verified (NFC)

Added: 
    

Modified: 
    mlir/test/Pass/invalid-ir-print-after-failure.mlir

Removed: 
    


################################################################################
diff  --git a/mlir/test/Pass/invalid-ir-print-after-failure.mlir b/mlir/test/Pass/invalid-ir-print-after-failure.mlir
index f205550cd3207..283a8810f5dc5 100644
--- a/mlir/test/Pass/invalid-ir-print-after-failure.mlir
+++ b/mlir/test/Pass/invalid-ir-print-after-failure.mlir
@@ -1,12 +1,16 @@
-// RUN: not mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=true signal-pass-failure=true})' -mlir-print-ir-after-failure 2>&1 | FileCheck %s --check-prefix=CHECK-INVALID
-// RUN: not mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=true signal-pass-failure=false})' -mlir-print-ir-after-failure 2>&1 | FileCheck %s --check-prefix=CHECK-INVALID
-// RUN: not mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=false signal-pass-failure=true})' -mlir-print-ir-after-failure 2>&1  | FileCheck %s --check-prefix=CHECK-VALID
-// RUN: mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=false signal-pass-failure=false})' -mlir-print-ir-after-failure 2>&1  | FileCheck %s --check-prefix=CHECK-VALID
+// RUN: not mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=true signal-pass-failure=true})' -mlir-print-ir-after-failure 2>&1 | FileCheck %s --check-prefix=CHECK-GENERIC
+// RUN: not mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=true signal-pass-failure=false})' -mlir-print-ir-after-failure 2>&1 | FileCheck %s --check-prefix=CHECK-GENERIC
+// RUN: not mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=false signal-pass-failure=true})' -mlir-print-ir-after-failure 2>&1  | FileCheck %s --check-prefix=CHECK-CUSTOM
+// RUN: mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=false signal-pass-failure=false})' -mlir-print-ir-after-failure 2>&1  | FileCheck %s --check-prefix=CHECK-CUSTOM
+
+// Check that `-mlir-print-assume-verified` will print custom even when the IR is invalid.
+// RUN: not mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=true signal-pass-failure=true})' -mlir-print-ir-after-failure 2>&1 -mlir-print-assume-verified | FileCheck %s --check-prefix=CHECK-CUSTOM
+// RUN: not mlir-opt %s -pass-pipeline='func.func(test-pass-create-invalid-ir{emit-invalid-ir=true signal-pass-failure=false})' -mlir-print-ir-after-failure 2>&1 -mlir-print-assume-verified | FileCheck %s --check-prefix=CHECK-CUSTOM
 
 // Test whether we print generically or not on pass failure, depending on whether there is invalid IR or not.
 
-// CHECK-VALID: func @TestCreateInvalidCallInPass
-// CHECK-INVALID: "func.func"
+// CHECK-CUSTOM: func @TestCreateInvalidCallInPass
+// CHECK-GENERIC: "func.func"
 func @TestCreateInvalidCallInPass() {
   return
 }