mirror of
https://github.com/pseXperiments/icicle.git
synced 2026-01-08 23:17:54 -05:00
2b07513310
## Describe the changes This PR adds the capability to pin host memory in golang bindings allowing data transfers to be quicker. Memory can be pinned once for multiple devices by passing the flag `cuda_runtime.CudaHostRegisterPortable` or `cuda_runtime.CudaHostAllocPortable` depending on how pinned memory is called
67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
package cuda_runtime
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"unsafe"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMalloc(t *testing.T) {
|
|
_, err := Malloc(20)
|
|
assert.Equal(t, CudaSuccess, err, "Unable to allocate device memory due to %d", err)
|
|
}
|
|
|
|
func TestMallocAsync(t *testing.T) {
|
|
stream, _ := CreateStream()
|
|
_, err := MallocAsync(20, stream)
|
|
assert.Equal(t, CudaSuccess, err, "Unable to allocate device memory due to %d", err)
|
|
}
|
|
|
|
func TestFree(t *testing.T) {
|
|
mem, err := Malloc(20)
|
|
assert.Equal(t, CudaSuccess, err, "Unable to allocate device memory due to %d", err)
|
|
|
|
err = Free(mem)
|
|
assert.Equal(t, CudaSuccess, err, "Unable to free device memory due to %v", err)
|
|
}
|
|
|
|
func TestCopyFromToHost(t *testing.T) {
|
|
someInts := make([]int32, 1)
|
|
someInts[0] = 34
|
|
numBytes := uint(8)
|
|
deviceMem, _ := Malloc(numBytes)
|
|
deviceMem, err := CopyToDevice(deviceMem, unsafe.Pointer(&someInts[0]), numBytes)
|
|
assert.Equal(t, CudaSuccess, err, "Couldn't copy to device due to %v", err)
|
|
|
|
someInts2 := make([]int32, 1)
|
|
_, err = CopyFromDevice(unsafe.Pointer(&someInts2[0]), deviceMem, numBytes)
|
|
assert.Equal(t, CudaSuccess, err, "Couldn't copy to device due to %v", err)
|
|
assert.Equal(t, someInts, someInts2, "Elements of host slices do not match. Copying from/to host failed")
|
|
}
|
|
|
|
func TestRegisterUnregisterPinned(t *testing.T) {
|
|
data := []int{1, 2, 3, 4, 5, 7, 8, 9}
|
|
dataUnsafe := unsafe.Pointer(&data[0])
|
|
_, err := RegisterPinned(dataUnsafe, int(unsafe.Sizeof(data[0])*9), CudaHostRegisterDefault)
|
|
assert.Equal(t, CudaSuccess, err)
|
|
_, err = RegisterPinned(dataUnsafe, int(unsafe.Sizeof(data[0])*9), CudaHostRegisterDefault)
|
|
assert.Equal(t, CudaErrorHostMemoryAlreadyRegistered, err)
|
|
|
|
err = FreeRegisteredPinned(dataUnsafe)
|
|
assert.Equal(t, CudaSuccess, err)
|
|
err = FreeRegisteredPinned(dataUnsafe)
|
|
assert.Equal(t, CudaErrorHostMemoryNotRegistered, err)
|
|
}
|
|
|
|
func TestAllocFreePinned(t *testing.T) {
|
|
pinnedMemPointer, err := AllocPinned(int(unsafe.Sizeof(1)*9), CudaHostAllocDefault)
|
|
assert.Equal(t, CudaSuccess, err)
|
|
|
|
err = FreeAllocPinned(pinnedMemPointer)
|
|
assert.Equal(t, CudaSuccess, err)
|
|
err = FreeAllocPinned(pinnedMemPointer)
|
|
assert.Equal(t, CudaErrorInvalidValue, err)
|
|
}
|