diff options
| author |  Andrei Nigmatulin <andrei.nigmatulin@gmail.com> | 2017-12-21 16:04:46 +0000 |
|---|---|---|
| committer |  Shanqing Cai <cais@google.com> | 2017-12-21 11:04:46 -0500 |
| commit | 9315a12cf78a0b109f89b08556a6ae63a50e1f30 (patch) | |
| tree | 935c4f2edb659eb494fad3214533ea3916ac4dfa /tensorflow/go/session_test.go | |
| parent | f950ea836de77483ca2eb0fce8e845188a76b3fc (diff) | |
golang: added Session.ListDevices method (#14385)
* golang: added Session.ListDevices method
Diffstat (limited to 'tensorflow/go/session_test.go')
| -rw-r--r-- | tensorflow/go/session_test.go | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tensorflow/go/session_test.go b/tensorflow/go/session_test.go index 73d78a8e57..05ace99a23 100644 --- a/tensorflow/go/session_test.go +++ b/tensorflow/go/session_test.go @@ -283,3 +283,19 @@ func TestSessionConfig(t *testing.T) { t.Fatalf("Got %v, want -1", output[0].Value()) } } + +func TestListDevices(t *testing.T) { + s, err := NewSession(NewGraph(), nil) + if err != nil { + t.Fatalf("NewSession(): %v", err) + } + + devices, err := s.ListDevices() + if err != nil { + t.Fatalf("ListDevices(): %v", err) + } + + if len(devices) == 0 { + t.Fatalf("no devices detected") + } +} |