Bubble up exceptions in EXLA.DeviceBuffer

Author: josevalim

exla/lib/exla/defn/buffers.ex

@@ -109,7 +109,13 @@ defmodule EXLA.Defn.Buffers do
     case data do
       %EXLA.Backend{buffer: %EXLA.DeviceBuffer{ref: ref} = buffer}
       when node(ref) != node() ->
-        binary = :erpc.call(node(ref), EXLA.DeviceBuffer, :read, [buffer])
+        try do
+          :erpc.call(node(ref), EXLA.DeviceBuffer, :read, [buffer])
+        catch
+          :error, {:exception, reason, stacktrace} ->
+            reraise Exception.normalize(:error, reason, stacktrace), stacktrace
+        end
+
         EXLA.BinaryBuffer.from_binary(binary, to_typespec(tensor))
 
       %EXLA.Backend{buffer: %EXLA.DeviceBuffer{} = buffer}