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 }