Disable compliler stick/unstick on zOS (#2888)

Signed-off-by: Alexandre Eichenberger <[email protected]>
onnx · Jul 25, 2024 · 603a8a2 · 603a8a2
1 parent e5b73a7
commit 603a8a2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/Accelerators/NNPA/Compiler/NNPACompilerUtils.cpp b/src/Accelerators/NNPA/Compiler/NNPACompilerUtils.cpp
@@ -52,6 +52,11 @@ void configurePassesNNPA() {
   // TODO: remove this if zDNN adds support for saturation.
   if (nnpaEnableSaturation)
     nnpaEnableCompilerStickUnstick = true;
+  // Currently nnpaEnableCompilerStickUnstick not supported on zOS.
+  // TODO enable on zOS
+  if (mtriple == "s390x-ibm-zos") {
+    nnpaEnableCompilerStickUnstick = false;
+  }
 }
 
 void addONNXToZHighPasses(mlir::PassManager &pm) {