dmlc--dgl
VoVAllen
dd65ee211e
* change ci image * fix * force bash * fix * fix python version * fix * fix * fix * update gpu * cuda * jenkins * fix build sh * fix * Revert "fix" This reverts commit 6b091914b3ef6b4300fa662ee375aa4b27944f5c. * try fix * fix * Revert "fix" This reverts commit e42c3035fa4974c6b197aaba0748f7b45113ddaa. * try fix * fix * fix * fix * fix * fix * fix * fix tests * try fix * fix * fix * fix * fix * fix * fix * fix ctx problem * fix many tests * fix typo * add backend * move to pytorch folder * fix? * fix ci * try skip * try false * try? * try? * Revert to 7d9a52f * fix various * fix lint * Revert Jenkinsfile * revert env * revert env * address comment * remove file
96 行
2.6 KiB
Python
96 行
2.6 KiB
Python
import dgl
|
|
import dgl.ndarray as nd
|
|
from dgl.utils import toindex
|
|
import numpy as np
|
|
import backend as F
|
|
import unittest
|
|
|
|
@unittest.skipIf(dgl.backend.backend_name == "tensorflow", reason="TF doesn't support inplace update")
|
|
def test_dlpack():
|
|
# test dlpack conversion.
|
|
def nd2th():
|
|
ans = np.array([[1., 1., 1., 1.],
|
|
[0., 0., 0., 0.],
|
|
[0., 0., 0., 0.]])
|
|
x = nd.array(np.zeros((3, 4), dtype=np.float32))
|
|
dl = x.to_dlpack()
|
|
y = F.zerocopy_from_dlpack(dl)
|
|
y[0] = 1
|
|
print(x)
|
|
print(y)
|
|
assert np.allclose(x.asnumpy(), ans)
|
|
|
|
def th2nd():
|
|
ans = np.array([[1., 1., 1., 1.],
|
|
[0., 0., 0., 0.],
|
|
[0., 0., 0., 0.]])
|
|
x = F.zeros((3, 4))
|
|
dl = F.zerocopy_to_dlpack(x)
|
|
y = nd.from_dlpack(dl)
|
|
x[0] = 1
|
|
print(x)
|
|
print(y)
|
|
assert np.allclose(y.asnumpy(), ans)
|
|
|
|
def th2nd_incontiguous():
|
|
x = F.astype(F.tensor([[0, 1], [2, 3]]), F.int64)
|
|
ans = np.array([0, 2])
|
|
y = x[:2, 0]
|
|
# Uncomment this line and comment the one below to observe error
|
|
#dl = dlpack.to_dlpack(y)
|
|
dl = F.zerocopy_to_dlpack(y)
|
|
z = nd.from_dlpack(dl)
|
|
print(x)
|
|
print(z)
|
|
assert np.allclose(z.asnumpy(), ans)
|
|
|
|
nd2th()
|
|
th2nd()
|
|
th2nd_incontiguous()
|
|
|
|
def test_index():
|
|
ans = np.ones((10,), dtype=np.int64) * 10
|
|
# from np data
|
|
data = np.ones((10,), dtype=np.int64) * 10
|
|
idx = toindex(data)
|
|
y1 = idx.tonumpy()
|
|
y2 = F.asnumpy(idx.tousertensor())
|
|
y3 = idx.todgltensor().asnumpy()
|
|
assert np.allclose(ans, y1)
|
|
assert np.allclose(ans, y2)
|
|
assert np.allclose(ans, y3)
|
|
|
|
# from list
|
|
data = [10] * 10
|
|
idx = toindex(data)
|
|
y1 = idx.tonumpy()
|
|
y2 = F.asnumpy(idx.tousertensor())
|
|
y3 = idx.todgltensor().asnumpy()
|
|
assert np.allclose(ans, y1)
|
|
assert np.allclose(ans, y2)
|
|
assert np.allclose(ans, y3)
|
|
|
|
# from dl tensor
|
|
data = F.ones((10,), dtype=F.int64) * 10
|
|
idx = toindex(data)
|
|
y1 = idx.tonumpy()
|
|
y2 = F.asnumpy(idx.tousertensor())
|
|
y3 = idx.todgltensor().asnumpy()
|
|
assert np.allclose(ans, y1)
|
|
assert np.allclose(ans, y2)
|
|
assert np.allclose(ans, y3)
|
|
|
|
# from dgl.NDArray
|
|
data = dgl.ndarray.array(np.ones((10,), dtype=np.int64) * 10)
|
|
idx = toindex(data)
|
|
y1 = idx.tonumpy()
|
|
y2 = F.asnumpy(idx.tousertensor())
|
|
y3 = idx.todgltensor().asnumpy()
|
|
assert np.allclose(ans, y1)
|
|
assert np.allclose(ans, y2)
|
|
assert np.allclose(ans, y3)
|
|
|
|
if __name__ == '__main__':
|
|
test_dlpack()
|
|
test_index()
|