experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Go: allow larger C array backed slices on 64 bit machines

Browse Source
This commit is contained in:
Siddhartha Bagaria 2019-08-14 17:39:18 -07:00
parent a92016d3fe
commit c3cbf5c041
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/go/graph.go
View File

@ -94,7 +94,12 @@ func (g *Graph) WriteTo(w io.Writer) (int64, error) {
// A []byte slice backed by C memory.
// See: https://github.com/golang/go/wiki/cgo#turning-c-arrays-into-go-slices
length := int(buf.length)
slice := (*[1 << 30]byte)(unsafe.Pointer(buf.data))[:length:length]
var slice []byte
if unsafe.Sizeof(unsafe.Pointer(nil)) == 8 {
slice = (*[1<<50 - 1]byte)(unsafe.Pointer(buf.data))[:length:length]
} else {
slice = (*[1 << 30]byte)(unsafe.Pointer(buf.data))[:length:length]
}
n, err := w.Write(slice)
return int64(n), err
}

7
tensorflow/go/tensor.go
View File

@ -207,7 +207,12 @@ func tensorData(c *C.TF_Tensor) []byte {
return nil
}
length := int(C.TF_TensorByteSize(c))
slice := (*[1 << 30]byte)(unsafe.Pointer(cbytes))[:length:length]
var slice []byte
if unsafe.Sizeof(unsafe.Pointer(nil)) == 8 {
slice = (*[1<<50 - 1]byte)(unsafe.Pointer(cbytes))[:length:length]
} else {
slice = (*[1 << 30]byte)(unsafe.Pointer(cbytes))[:length:length]
}
return slice
}