test crash

sdasgup3 · Mar 16, 2024 · 3e8ead5 · 3e8ead5
1 parent dc4bc72
commit 3e8ead5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/examples/c++/stablehlo_add.cpp b/examples/c++/stablehlo_add.cpp
@@ -12,6 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
+#include "llvm/Support/PrettyStackTrace.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
 #include "mlir/Dialect/Quant/QuantOps.h"
 #include "mlir/IR/BuiltinOps.h"
@@ -20,6 +21,8 @@ limitations under the License.
 #include "stablehlo/dialect/StablehloOps.h"
 
 int main() {
+  llvm::EnablePrettyStackTrace();  // Enable pretty stack traces
+
   mlir::MLIRContext context;
 
   /** create module **/

diff --git a/stablehlo/tools/StablehloOptMain.cpp b/stablehlo/tools/StablehloOptMain.cpp
@@ -13,6 +13,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
 
+#include "llvm/Support/PrettyStackTrace.h"
 #include "mlir/Dialect/Tosa/IR/TosaOps.h"
 #include "mlir/Dialect/Tosa/Transforms/Passes.h"
 #include "mlir/InitAllDialects.h"
@@ -28,6 +29,7 @@ limitations under the License.
 #include "stablehlo/transforms/Passes.h"
 
 int main(int argc, char **argv) {
+  llvm::EnablePrettyStackTrace();  // Enable pretty stack traces
   mlir::registerAllPasses();
   mlir::hlo::registerAllTestPasses();
   mlir::stablehlo::registerPassPipelines();