首页 > 后端开发 > Golang > 正文

使用runtime.Pinner将二维数组从Go传递到C

PHPz
发布: 2024-02-12 17:51:05
转载
570 人浏览过

使用runtime.Pinner将二维数组从Go传递到C

问题内容

我试图将二维数组从 Go 传递到某个 C 函数 void foo(in **float, out *double)。因为我想要这个 C 函数的包装器,所以我希望 Go 函数具有像 func FooWrapper([][]float32) []float64 这样的定义。最简单但效率不高的实现是通过下面列出的 C 来分配所有内存:

<code>func FooWrapper(values [][]float32) []float64 {
    totalObj := len(values)
    totalParams := len(values[0])
    results := make([]float64, totalObj)

    ptrArrLength := uintptr(totalObj) * cPointerSize
    paramArrLength := uintptr(totalParams) * cFloatSize

    ptr := C.malloc(C.size_t(ptrArrLength + paramArrLength*uintptr(totalObj)))
    defer C.free(ptr)

    ptrSlice := unsafe.Slice((**C.float)(ptr), totalObj)
    for i, obj := range values {
        paramArrPtr := (*C.float)(unsafe.Add(ptr, ptrArrLength+uintptr(i)*paramArrLength))
        ptrSlice[i] = paramArrPtr

        paramSlice := unsafe.Slice(paramArrPtr, totalParams)
        for j, param := range obj {
            paramSlice[j] = (C.float)(param)
        }
    }

    C.foo((**C.float)(ptr), (*C.double)(&results[0]))

    return results
}
</code>
登录后复制

这样的实施安全吗?我可以传递result数据的指针吗?据我所知,这个指针将被固定,因为它传递给了 C 函数。

但是我想分配更少的内存,只是重用Go内存,我已经了解了 runtime.Pinner ,它使指针固定直到 runtime.Pinner.Unpin() 调用。我尝试使用 pinner 编写另一个实现:

<code>func FooWrapper(values [][]float32) []float64 {
    length := len(values)
    results := make([]float64, length)

    pinner := runtime.Pinner{}
    defer pinner.Unpin()

    arr := (**C.float)(C.malloc(C.size_t(uintptr(length) * cPointerSize)))
    defer C.free(unsafe.Pointer(arr))
    slice := unsafe.Slice(arr, length)
    for i, v := range values {
        pinner.Pin(&v[0])
        slice[i] = (*C.float)(&v[0])
    }

    C.foo(arr, (*C.double)(&results[0]))

    return results
}
</code>
登录后复制

但是,不幸的是,这段代码不起作用

runtime: pointer 0xc016ecbfc0 to unused region of span span.base()=0xc016eca000 span.limit=0xc016ecbfa0 span.state=1
fatal error: found bad pointer in Go heap (incorrect use of unsafe or cgo?)
登录后复制

我使用runtime.Pinner是否错误(据我所知,我可以固定切片数据)?或者这段代码还有另一个错误。除了将所有数据分配和复制到 C 内存之外,是否有一些将 3d(4d 等)数组传递给 C 函数的实现?

解决方法

我找到了问题的答案。使用 malloc 是危险的,因为当我尝试将 Go 指针写入 C 内存时,GC 可以将未初始化的内存识别为“坏”内存,因此如果我将 malloc 更改为 calloc 就不会出现问题。当然这不是最好的解决方案,最好在 Go 中分配内存,因为它更快、更安全。

所有讨论此处一个>.

以上是使用runtime.Pinner将二维数组从Go传递到C的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:stackoverflow.com
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板