[Mlir-commits] [mlir] 0d70bc9 - [mlir][msan][test] Disable jit tests

[Vitaly Buka]

Author: Vitaly Buka
Date: 2022-04-28T08:50:13-07:00
New Revision: 0d70bc990bc4b3c90bedd17b6d5b80e35d147524

URL: https://github.com/llvm/llvm-project/commit/0d70bc990bc4b3c90bedd17b6d5b80e35d147524
DIFF: https://github.com/llvm/llvm-project/commit/0d70bc990bc4b3c90bedd17b6d5b80e35d147524.diff

LOG: [mlir][msan][test] Disable jit tests

I am going to enable MLIR test on msan bot
https://lab.llvm.org/buildbot/#/builders/sanitizer-x86_64-linux-bootstrap-msan

Reviewed By: ftynse

Differential Revision: https://reviews.llvm.org/D124574

Added: 
    

Modified: 
    mlir/test/mlir-cpu-runner/lit.local.cfg
    mlir/unittests/ExecutionEngine/Invoke.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/test/mlir-cpu-runner/lit.local.cfg b/mlir/test/mlir-cpu-runner/lit.local.cfg
index 3a9b0ef14268a..7dc3525ae0aef 100644
--- a/mlir/test/mlir-cpu-runner/lit.local.cfg
+++ b/mlir/test/mlir-cpu-runner/lit.local.cfg
@@ -4,6 +4,10 @@ import sys
 if sys.platform == 'win32':
     config.unsupported = True
 
+# MSAN does not work with JIT.
+if 'msan' in config.available_features:
+  config.unsupported = True
+
 # Requires the native target to be configured in.
 if 'llvm_has_native_target' not in config.available_features:
     config.unsupported = True

diff  --git a/mlir/unittests/ExecutionEngine/Invoke.cpp b/mlir/unittests/ExecutionEngine/Invoke.cpp
index cc91254c474ed..febfbd139b6ea 100644
--- a/mlir/unittests/ExecutionEngine/Invoke.cpp
+++ b/mlir/unittests/ExecutionEngine/Invoke.cpp
@@ -232,7 +232,13 @@ static void memrefMultiply(::StridedMemRefType<float, 2> *memref,
     elt *= coefficient;
 }
 
-TEST(NativeMemRefJit, JITCallback) {
+// MSAN does not work with JIT.
+#if __has_feature(memory_sanitizer)
+#define MAYBE_JITCallback DISABLED_JITCallback
+#else
+#define MAYBE_JITCallback JITCallback
+#endif
+TEST(NativeMemRefJit, MAYBE_JITCallback) {
   constexpr int k = 2;
   constexpr int m = 2;
   int64_t shape[] = {k, m};