zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-22 02:58:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

6
mlx/backend/metal/metal.cpp
View File

@@ -47,7 +47,11 @@ std::function<void()> make_task(array arr, bool signal) {
}
debug_set_primitive_buffer_label(command_buffer, arr.primitive());
arr.primitive().eval_gpu(arr.inputs(), outputs);
try {
arr.primitive().eval_gpu(arr.inputs(), outputs);
} catch (const std::exception& error) {
abort_with_exception(error);
}
}
std::vector<std::shared_ptr<array::Data>> buffers;
for (auto& in : arr.inputs()) {