项目文件夹

文件

T

文件历史

Xin Yao 5fcd7f290f [Feature] Enable UVA for GPU PinSAGE and RandomWalk (#3857 )

* enable uva for pinsage sampler

* unit test

* modify some checks on the python side

* remove legacy random walk code

* update unit test

* update unit test

* fix unit test

* adjust checks

* move some checks to c++

* move max_nodes check to cuda kernel

* fix ci for tf

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>

2022-04-11 20:01:23 +08:00

data

[Performance] Replace np.array with np.asarray (#1301 )

2020-03-03 20:45:41 +08:00

sampling

[Feature] Enable UVA for GPU PinSAGE and RandomWalk (#3857 )

2022-04-11 20:01:23 +08:00

__init__.py

[Feature][Performance][GPU] Introducing UnifiedTensor for efficient zero-copy host memory access from GPU (#3086 )

2021-07-17 00:06:20 +08:00

dis_kvstore.py

[Distributed] Distributed node embedding and sparse optimizer (#2733 )