src/array/cuda/rowwise_sampling_prob.cu · 72781efbcf2532254957090c889ec5b7b9563245 · OpenDAS / dgl

[Sampling] Enable sampling with edge masks on homogeneous graph (#4748) · 72781efb

Quan (Andy) Gan authored Oct 28, 2022

* sample neighbors with masks

* oops

* refactor again

* remove

* remove debug code

* rename macro

* address comments

* address comment

* address comments

* rename a lot of stuff

* oops

72781efb

rowwise_sampling_prob.cu 28.4 KB

Replace rowwise_sampling_prob.cu