Fix: Go GC could destroy input tensors while SessionRun() was still running, see https://github.com/tensorflow/tensorflow/issues/13129

2017-09-20 15:33:28 +01:00 · 2017-09-20 15:33:28 +01:00 · 846454faae
commit 846454faae
parent 03619fab3f
1 changed files with 4 additions and 0 deletions
--- a/tensorflow/go/session.go
+++ b/tensorflow/go/session.go
@ -89,6 +89,10 @@ func (s *Session) Run(feeds map[Output]*Tensor, fetches []Output, targets []*Ope
 		ptrOutput(c.fetches), ptrTensor(c.fetchTensors), C.int(len(fetches)),
 		ptrOperation(c.targets), C.int(len(targets)),
 		nil, status.c)
+
+	// Make sure GC won't harvest input tensors until SessionRun() is finished
+	runtime.KeepAlive(feeds)
+
 	if err := status.Err(); err != nil {
 		return nil, err
 	}