workaround for nGraph python runtime tensor write data corruption (#1087)

0deea239 · Adam Straw · Robert Kimball · d783eece · 0deea239
Commit 0deea239 authored Jun 06, 2018 by Adam Straw Committed by Robert Kimball Jun 06, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

runtime.py python/ngraph/runtime.py +3 -1

No files found.
--- a/python/ngraph/runtime.py
+++ b/python/ngraph/runtime.py
@@ -120,7 +120,9 @@ class Computation:

        buffer_size = Computation._get_buffer_size(
            tensor_view.element_type, tensor_view.element_count)
-        tensor_view.write(util.numpy_to_c(np.ascontiguousarray(value)), 0, buffer_size)
+
+        nparray = np.ascontiguousarray(value)
+        tensor_view.write(util.numpy_to_c(nparray), 0, buffer_size)

    @staticmethod
    def _read_tensor_view_to_ndarray(tensor_view, output):