wrap launchHostFunc call in device_task_fn into DeviceSafeCall

ValeevGroup · Aug 27, 2024 · 2e907ed · 2e907ed
1 parent 09bb258
commit 2e907ed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/TiledArray/device/device_task_fn.h b/src/TiledArray/device/device_task_fn.h
@@ -121,7 +121,8 @@ struct deviceTaskFn : public TaskInterface {
       } else {
         // TODO should we use device callback or device events??
         // insert device callback
-        TiledArray::device::launchHostFunc(*stream_, device_callback, task_);
+        DeviceSafeCall(TiledArray::device::launchHostFunc(
+            *stream_, device_callback, task_));
         // processed sync, clear state
         stream_ = {};
       }