Print exceptions in eval_cpu/eval_gpu and abort (#1754)

This commit is contained in:
Cheng
2025-01-08 23:31:09 +09:00
committed by GitHub
parent d1766f2c70
commit b8f76f717a
4 changed files with 22 additions and 2 deletions

View File

@@ -2,6 +2,7 @@
#pragma once
#include <exception>
#include <variant>
#include "mlx/array.h"
@@ -53,6 +54,9 @@ struct PrintFormatter {
PrintFormatter& get_global_formatter();
/** Print the exception and then abort. */
void abort_with_exception(const std::exception& error);
/** Holds information about floating-point types. */
struct finfo {
explicit finfo(Dtype dtype);