d30a69bf277effd9a013ddd81fa33661f2e31a01,python/dgl/runtime/spmv.py,,build_gidx_and_mapping_uv,#Any#Any#Any#,146
Before Change
forward, backward = gidx.get_csr_shuffle_order(0)
eid = eid.tousertensor()
nbits = gidx.bits_needed(0)
forward_map = utils.to_nbits_int(eid[forward.tousertensor()], nbits)
backward_map = utils.to_nbits_int(eid[backward.tousertensor()], nbits)
forward_map = F.zerocopy_to_dgl_ndarray(forward_map)
backward_map = F.zerocopy_to_dgl_ndarray(backward_map)
After Change
forward, backward = gidx.get_csr_shuffle_order(0)
eid = eid.tousertensor()
nbits = gidx.bits_needed(0)
forward_map = utils.to_nbits_int(F.gather_row(eid, forward.tousertensor()), nbits)
backward_map = utils.to_nbits_int(F.gather_row(eid, backward.tousertensor()), nbits)
forward_map = F.zerocopy_to_dgl_ndarray(forward_map)
backward_map = F.zerocopy_to_dgl_ndarray(backward_map)
In pattern: SUPERPATTERN
Frequency: 3
Non-data size: 4
Instances
Project Name: dmlc/dgl
Commit Name: d30a69bf277effd9a013ddd81fa33661f2e31a01
Time: 2019-12-19
Author: VoVAllen@users.noreply.github.com
File Name: python/dgl/runtime/spmv.py
Class Name:
Method Name: build_gidx_and_mapping_uv
Project Name: dmlc/dgl
Commit Name: d30a69bf277effd9a013ddd81fa33661f2e31a01
Time: 2019-12-19
Author: VoVAllen@users.noreply.github.com
File Name: python/dgl/nodeflow.py
Class Name: NodeFlow
Method Name: map_to_parent_eid
Project Name: dmlc/dgl
Commit Name: d30a69bf277effd9a013ddd81fa33661f2e31a01
Time: 2019-12-19
Author: VoVAllen@users.noreply.github.com
File Name: python/dgl/nodeflow.py
Class Name: NodeFlow
Method Name: map_to_parent_nid