func TestCpy(t *testing.T) {
N0, N1, N2 := 2, 4, 32
N := N0 * N1 * N2
mesh := data.NewMesh(N0, N1, N2, 1, 1, 1)
h1 := make([]float32, N)
for i := range h1 {
h1[i] = float32(i)
}
hs := data.SliceFromList([][]float32{h1}, mesh)
d := NewSlice(1, mesh)
data.Copy(d, hs)
d2 := NewSlice(1, mesh)
data.Copy(d2, d)
h2 := data.NewSlice(1, mesh)
data.Copy(h2, d2)
res := h2.Host()[0]
for i := range res {
if res[i] != h1[i] {
t.Fail()
}
}
}
func toGPU(list []float32) *data.Slice {
mesh := data.NewMesh(1, 1, len(list), 1, 1, 1)
h := data.SliceFromList([][]float32{list}, mesh)
d := NewSlice(1, mesh)
data.Copy(d, h)
return d
}