$ git clone https://gitee.com/mindspore/graphlearning.git 正克隆到 'graphlearning'... remote: Enumerating objects: 1275, done. remote: Counting objects: 100% (221/221), done. remote: Compressing objects: 100% (152/152), done. remote: Total 1275 (delta 116), reused 127 (delta 68), pack-reused 1054 接收对象中: 100% (1275/1275), 1.41 MiB | 316.00 KiB/s, 完成. 处理 delta 中: 100% (715/715), 完成. $ cd graphlearning/ $ ll 总用量 112 drwxrwxr-x 12 dechin dechin 4096 11月 9 17:19 ./ drwxrwxr-x 10 dechin dechin 4096 11月 9 17:19 ../ -rwxrwxr-x 1 dechin dechin 1429 11月 9 17:19 build.sh* drwxrwxr-x 2 dechin dechin 4096 11月 9 17:19 examples/ -rwxrwxr-x 1 dechin dechin 3148 11月 9 17:19 FAQ_CN.md* -rwxrwxr-x 1 dechin dechin 4148 11月 9 17:19 faq.md* drwxrwxr-x 8 dechin dechin 4096 11月 9 17:19 .git/ -rwxrwxr-x 1 dechin dechin 1844 11月 9 17:19 .gitignore* drwxrwxr-x 2 dechin dechin 4096 11月 9 17:19 images/ drwxrwxr-x 3 dechin dechin 4096 11月 9 17:19 .jenkins/ -rw-rw-r-- 1 dechin dechin 11357 11月 9 17:19 LICENSE drwxrwxr-x 11 dechin dechin 4096 11月 9 17:19 mindspore_gl/ drwxrwxr-x 11 dechin dechin 4096 11月 9 17:19 model_zoo/ -rwxrwxr-x 1 dechin dechin 52 11月 9 17:19 OWNERS* -rwxrwxr-x 1 dechin dechin 3648 11月 9 17:19 README_CN.md* -rwxrwxr-x 1 dechin dechin 4570 11月 9 17:19 README.md* drwxrwxr-x 4 dechin dechin 4096 11月 9 17:19 recommendation/ -rwxrwxr-x 1 dechin dechin 922 11月 9 17:19 RELEASE.md* -rwxrwxr-x 1 dechin dechin 108 11月 9 17:19 requirements.txt* drwxrwxr-x 2 dechin dechin 4096 11月 9 17:19 scripts/ -rwxrwxr-x 1 dechin dechin 4164 11月 9 17:19 setup.py* drwxrwxr-x 5 dechin dechin 4096 11月 9 17:19 tests/ drwxrwxr-x 5 dechin dechin 4096 11月 9 17:19 tools/

$ bash build.sh mkdir: 已创建目录 '/home/dechin/projects/mindspore/graphlearning/output' Collecting Cython>=0.29.24 Downloading Cython-0.29.32-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.manylinux_2_24_x86_64.whl (2.0 MB) |████████████████████████████████| 2.0 MB 823 kB/s Collecting ast-decompiler>=0.6.0 Downloading ast_decompiler-0.7.0-py3-none-any.whl (13 kB) Collecting astpretty>=2.1.0 Downloading astpretty-3.0.0-py2.py3-none-any.whl (4.9 kB) Collecting scikit-learn>=0.24.2 Downloading scikit_learn-1.1.3-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (30.8 MB) |████████████████████████████████| 30.8 MB 2.6 MB/s Requirement already satisfied: numpy>=1.21.2 in /home/dechin/anaconda3/envs/mindspore16/lib/python3.9/site-packages (from -r /home/dechin/projects/mindspore/graphlearning/requirements.txt (line 5)) (1.23.2) Collecting networkx>=2.6.3 Downloading networkx-2.8.8-py3-none-any.whl (2.0 MB) |████████████████████████████████| 2.0 MB 4.6 MB/s Requirement already satisfied: scipy>=1.3.2 in /home/dechin/anaconda3/envs/mindspore16/lib/python3.9/site-packages (from scikit-learn>=0.24.2->-r /home/dechin/projects/mindspore/graphlearning/requirements.txt (line 4)) (1.5.3) Collecting threadpoolctl>=2.0.0 Downloading threadpoolctl-3.1.0-py3-none-any.whl (14 kB) Collecting joblib>=1.0.0 Downloading joblib-1.2.0-py3-none-any.whl (297 kB) |████████████████████████████████| 297 kB 2.2 MB/s Installing collected packages: threadpoolctl, joblib, scikit-learn, networkx, Cython, astpretty, ast-decompiler Successfully installed Cython-0.29.32 ast-decompiler-0.7.0 astpretty-3.0.0 joblib-1.2.0 networkx-2.8.8 scikit-learn-1.1.3 threadpoolctl-3.1.0 running bdist_wheel running build running build_py creating build creating build/lib.linux-x86_64-3.9 ... removing build/bdist.linux-x86_64/wheel mindspore_gl_gpu-0.1-cp39-cp39-linux_x86_64.whl ------Successfully created mindspore_gl package------

$ python3 -m pip install ./output/mindspore_gl_gpu-0.1-cp39-cp39-linux_x86_64.whl Processing ./output/mindspore_gl_gpu-0.1-cp39-cp39-linux_x86_64.whl Requirement already satisfied: Cython in /home/dechin/.local/lib/python3.9/site-packages (from mindspore-gl-gpu==0.1) (0.29.32) Requirement already satisfied: astpretty in /home/dechin/.local/lib/python3.9/site-packages (from mindspore-gl-gpu==0.1) (3.0.0) Requirement already satisfied: ast-decompiler>=0.3.2 in /home/dechin/.local/lib/python3.9/site-packages (from mindspore-gl-gpu==0.1) (0.7.0) Requirement already satisfied: scikit-learn>=0.24.2 in /home/dechin/.local/lib/python3.9/site-packages (from mindspore-gl-gpu==0.1) (1.1.3) Requirement already satisfied: threadpoolctl>=2.0.0 in /home/dechin/.local/lib/python3.9/site-packages (from scikit-learn>=0.24.2->mindspore-gl-gpu==0.1) (3.1.0) Requirement already satisfied: joblib>=1.0.0 in /home/dechin/.local/lib/python3.9/site-packages (from scikit-learn>=0.24.2->mindspore-gl-gpu==0.1) (1.2.0) Requirement already satisfied: scipy>=1.3.2 in /home/dechin/anaconda3/envs/mindspore16/lib/python3.9/site-packages (from scikit-learn>=0.24.2->mindspore-gl-gpu==0.1) (1.5.3) Requirement already satisfied: numpy>=1.17.3 in /home/dechin/anaconda3/envs/mindspore16/lib/python3.9/site-packages (from scikit-learn>=0.24.2->mindspore-gl-gpu==0.1) (1.23.2) Installing collected packages: mindspore-gl-gpu Successfully installed mindspore-gl-gpu-0.1

$ python3 -c 'import mindspore_gl' [WARNING] ME(3662914:140594637309120,MainProcess):2022-11-09-17:22:29.348.03 [mindspore/run_check/_check_version.py:189] Cuda ['10.1', '11.1'] version(need by mindspore-gpu) is not found, please confirm that the path of cuda is set to the env LD_LIBRARY_PATH, please refer to the installation guidelines: https://www.mindspore.cn/install [WARNING] ME(3662914:140594637309120,MainProcess):2022-11-09-17:22:29.350.73 [mindspore/run_check/_check_version.py:189] Cuda ['10.1', '11.1'] version(need by mindspore-gpu) is not found, please confirm that the path of cuda is set to the env LD_LIBRARY_PATH, please refer to the installation guidelines: https://www.mindspore.cn/install [WARNING] ME(3662914:140594637309120,MainProcess):2022-11-09-17:22:29.351.54 [mindspore/run_check/_check_version.py:189] Cuda ['10.1', '11.1'] version(need by mindspore-gpu) is not found, please confirm that the path of cuda is set to the env LD_LIBRARY_PATH, please refer to the installation guidelines: https://www.mindspore.cn/install [WARNING] ME(3662914:140594637309120,MainProcess):2022-11-09-17:22:29.352.40 [mindspore/run_check/_check_version.py:189] Cuda ['10.1', '11.1'] version(need by mindspore-gpu) is not found, please confirm that the path of cuda is set to the env LD_LIBRARY_PATH, please refer to the installation guidelines: https://www.mindspore.cn/install [WARNING] ME(3662914:140594637309120,MainProcess):2022-11-09-17:22:29.352.94 [mindspore/run_check/_check_version.py:189] Cuda ['10.1', '11.1'] version(need by mindspore-gpu) is not found, please confirm that the path of cuda is set to the env LD_LIBRARY_PATH, please refer to the installation guidelines: https://www.mindspore.cn/install [WARNING] ME(3662914:140594637309120,MainProcess):2022-11-09-17:22:29.353.43 [mindspore/run_check/_check_version.py:189] Cuda ['10.1', '11.1'] version(need by mindspore-gpu) is not found, please confirm that the path of cuda is set to the env LD_LIBRARY_PATH, please refer to the installation guidelines: https://www.mindspore.cn/install [WARNING] ME(3662914:140594637309120,MainProcess):2022-11-09-17:22:29.353.91 [mindspore/run_check/_check_version.py:189] Cuda ['10.1', '11.1'] version(need by mindspore-gpu) is not found, please confirm that the path of cuda is set to the env LD_LIBRARY_PATH, please refer to the installation guidelines: https://www.mindspore.cn/install

In [1]: import mindspore as ms

In [2]: from mindspore_gl import Graph, GraphField

In [3]: from mindspore_gl.nn import GNNCell

In [4]: n_nodes = 3

In [5]: n_edges = 3

In [6]: src_idx = ms.Tensor([0, 1, 2], ms.int32)

In [7]: dst_idx = ms.Tensor([1, 2, 0], ms.int32)

In [8]: graph_field = GraphField(src_idx, dst_idx, n_nodes, n_edges)

In [9]: node_feat = ms.Tensor([[1], [2], [3]], ms.float32)

In [10]: class TestSetVertexAttr(GNNCell): ...: def construct(self, x, y, g: Graph): ...: g.set_src_attr({"hs": x}) ...: g.set_dst_attr({"hd": y}) ...: return [v.hd for v in g.dst_vertex] * [u.hs for u in g.src_vertex] ...:

In [11]: ret = TestSetVertexAttr()(node_feat[src_idx], node_feat[dst_idx], *graph_field.get_graph()).asnumpy().tolist() -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | def construct(self, x, y, g: Graph): 1 || 1 def construct( | | || self, | | || x, | | || y, | | || src_idx, | | || dst_idx, | | || n_nodes, | | || n_edges, | | || UNUSED_0=None, | | || UNUSED_1=None, | | || UNUSED_2=None | | || ): | | || 2 SCATTER_ADD = ms.ops.TensorScatterAdd() | | || 3 SCATTER_MAX = ms.ops.TensorScatterMax() | | || 4 SCATTER_MIN = ms.ops.TensorScatterMin() | | || 5 GATHER = ms.ops.Gather() | | || 6 ZEROS = ms.ops.Zeros() | | || 7 FILL = ms.ops.Fill() | | || 8 MASKED_FILL = ms.ops.MaskedFill() | | || 9 IS_INF = ms.ops.IsInf() | | || 10 SHAPE = ms.ops.Shape() | | || 11 RESHAPE = ms.ops.Reshape() | | || 12 scatter_src_idx = RESHAPE(src_idx, (SHAPE(src_idx)[0], 1)) | | || 13 scatter_dst_idx = RESHAPE(dst_idx, (SHAPE(dst_idx)[0], 1)) | | g.set_src_attr({'hs': x}) 2 || 14 hs, = [x] | | g.set_dst_attr({'hd': y}) 3 || 15 hd, = [y] | | return [v.hd for v in g.dst_vertex] * [u.hs for u in g.src_vertex] 4 || 16 return hd * hs | --------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

In [12]: print (ret) [[2.0], [6.0], [3.0]]

import mindspore as ms from mindspore import ops from mindspore_gl import Graph, GraphField from mindspore_gl.nn import GNNCell

n_nodes = 3 n_edges = 3

src_idx = ms.Tensor([0, 1, 2, 3, 4], ms.int32) dst_idx = ms.Tensor([1, 2, 0, 1, 2], ms.int32)

graph_field = GraphField(src_idx, dst_idx, n_nodes, n_edges) node_feat = ms.Tensor([[1], [2], [3], [4], [5]], ms.float32)

class GraphConvCell(GNNCell): def construct(self, x, y, g: Graph): g.set_src_attr({"hs": x}) g.set_dst_attr({"hd": y}) return [g.sum([u.hs for u in v.innbs]) for v in g.dst_vertex]

ret = GraphConvCell()(node_feat[src_idx], node_feat[dst_idx], *graph_field.get_graph()).asnumpy().tolist() print (ret)

$ python3 test_msgl_01.py -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | def construct(self, x, y, g: Graph): 1 || 1 def construct( | | || self, | | || x, | | || y, | | || src_idx, | | || dst_idx, | | || n_nodes, | | || n_edges, | | || UNUSED_0=None, | | || UNUSED_1=None, | | || UNUSED_2=None | | || ): | | || 2 SCATTER_ADD = ms.ops.TensorScatterAdd() | | || 3 SCATTER_MAX = ms.ops.TensorScatterMax() | | || 4 SCATTER_MIN = ms.ops.TensorScatterMin() | | || 5 GATHER = ms.ops.Gather() | | || 6 ZEROS = ms.ops.Zeros() | | || 7 FILL = ms.ops.Fill() | | || 8 MASKED_FILL = ms.ops.MaskedFill() | | || 9 IS_INF = ms.ops.IsInf() | | || 10 SHAPE = ms.ops.Shape() | | || 11 RESHAPE = ms.ops.Reshape() | | || 12 scatter_src_idx = RESHAPE(src_idx, (SHAPE(src_idx)[0], 1)) | | || 13 scatter_dst_idx = RESHAPE(dst_idx, (SHAPE(dst_idx)[0], 1)) | | g.set_src_attr({'hs': x}) 2 || 14 hs, = [x] | | g.set_dst_attr({'hd': y}) 3 || 15 hd, = [y] | | return [g.sum([u.hs for u in v.innbs]) for v in g.dst_vertex] 4 || 16 SCATTER_INPUT_SNAPSHOT1 = GATHER(hs, src_idx, 0) | | || 17 return SCATTER_ADD( | | || ZEROS( | | || (n_nodes,) + SHAPE(SCATTER_INPUT_SNAPSHOT1)[1:], | | || SCATTER_INPUT_SNAPSHOT1.dtype | | || ), | | || scatter_dst_idx, | | || SCATTER_INPUT_SNAPSHOT1 | | || ) | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- [[3.0], [5.0], [7.0]]