zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-12 19:11:19 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Command buffer reports errors (#479)

Browse Source 
* command buffer reports errors

* typo

* simplify
This commit is contained in:
Awni Hannun 2024-01-17 11:53:30 -08:00 committed by GitHub
parent 4a9012cba0
commit 275db7221a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -42,6 +42,15 @@ MTL::CommandBuffer* increment_command_buffer(Stream s) {
return command_buffer;
}
inline void check_error(MTL::CommandBuffer* cbuf) {
if (cbuf->status() == MTL::CommandBufferStatusError) {
std::ostringstream msg;
msg << "[METAL] Command buffer execution failed: "
<< cbuf->error()->localizedDescription()->utf8String();
throw std::runtime_error(msg.str());
}
}
std::function<void()> make_task(
array& arr,
std::vector<std::shared_future<void>> deps,
@ -59,7 +68,7 @@ std::function<void()> make_task(
metal::device(s.device).end_encoding(s.index);
scheduler::notify_new_task(s);
command_buffer->addCompletedHandler(
[s, arr, p = std::move(p)](MTL::CommandBuffer*) mutable {
[s, arr, p = std::move(p)](MTL::CommandBuffer* cbuf) mutable {
if (!arr.is_tracer()) {
arr.detach();
for (auto s : arr.siblings()) {
@ -68,14 +77,16 @@ std::function<void()> make_task(
}
p->set_value();
scheduler::notify_task_completion(s);
check_error(cbuf);
});
metal::device(s.device).commit_command_buffer(s.index);
} else {
command_buffer->addCompletedHandler(
[s, arr](MTL::CommandBuffer*) mutable {
[s, arr](MTL::CommandBuffer* cbuf) mutable {
if (!arr.is_tracer()) {
arr.detach();
}
check_error(cbuf);
});
}
};