1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
import numpy
import pygpu
import pytest
from .support import (gen_gpuarray, context)
def test_array_split():
xc, xg = gen_gpuarray((8,), 'float32', ctx=context)
rc = numpy.array_split(xc, 3)
rg = pygpu.array_split(xg, 3)
assert len(rc) == len(rg)
for pc, pg in zip(rc, rg):
numpy.testing.assert_allclose(pc, numpy.asarray(pg))
xc, xg = gen_gpuarray((8,), 'float32', ctx=context)
rc = numpy.array_split(xc, 3, axis=-1)
rg = pygpu.array_split(xg, 3, axis=-1)
assert len(rc) == len(rg)
for pc, pg in zip(rc, rg):
numpy.testing.assert_allclose(pc, numpy.asarray(pg))
def test_split():
for spl in (3, [3, 5, 6, 10]):
xsplit('', (9,), spl)
def test_xsplit():
if tuple(int(v) for v in numpy.version.version.split('.')[:2]) < (1, 11):
pytest.skip("Numpy version too old")
for l in ('h', 'v'):
for spl in (2, [3, 6]):
xsplit(l, (4, 4), spl)
xsplit(l, (2, 2, 2), 2)
for spl in (2, [3, 6]):
xsplit('d', (2, 2, 4), spl)
def xsplit(l, shp, spl):
xc, xg = gen_gpuarray(shp, 'float32', ctx=context)
n = l + 'split'
rc = getattr(numpy, n)(xc, spl)
rg = getattr(pygpu, n)(xg, spl)
assert len(rc) == len(rg)
for pc, pg in zip(rc, rg):
numpy.testing.assert_allclose(pc, numpy.asarray(pg))
def test_concatenate():
ac, ag = gen_gpuarray((2, 2), 'float32', ctx=context)
bc, bg = gen_gpuarray((1, 2), 'float32', ctx=context)
rc = numpy.concatenate((ac, bc), axis=0)
rg = pygpu.concatenate((ag, bg), axis=0)
numpy.testing.assert_allclose(rc, numpy.asarray(rg))
rc = numpy.concatenate((ac, bc.T), axis=1)
rg = pygpu.concatenate((ag, bg.T), axis=1)
numpy.testing.assert_allclose(rc, numpy.asarray(rg))
rc = numpy.concatenate((ac, bc.T), axis=-1)
rg = pygpu.concatenate((ag, bg.T), axis=-1)
numpy.testing.assert_allclose(rc, numpy.asarray(rg))
def test_hstack():
for shp in [(3,), (3, 1)]:
xstack('h', (shp, shp), (), context)
def test_vstack():
for shp in [(3,), (3, 1)]:
xstack('v', (shp, shp), (), context)
def test_dstack():
for shp in [(3,), (3, 1)]:
xstack('d', (shp, shp), (), context)
def xstack(l, shps, tup, ctx):
tupc = list(tup)
tupg = list(tup)
for shp in shps:
tc, tg = gen_gpuarray(shp, 'float32', ctx=context)
tupc.append(tc)
tupg.append(tg)
n = l + 'stack'
rc = getattr(numpy, n)(tuple(tupc))
rg = getattr(pygpu, n)(tuple(tupg), ctx)
numpy.testing.assert_allclose(rc, numpy.asarray(rg))
|
