• kylasa's avatar
    code changes for bug fixes identified during mag_lsc dataset (#4187) · 3ccd973c
    kylasa authored
    * code changes for bug fixes identified during mag_lsc dataset
    
    1. Changed from call torch.Tensor() to torch.from_numpy() to address memory corruption issues when creating large tensors. Tricky thing is this works correctly for small tensors.
    2. Changed dgl.graph() function call to include 'num_nodes" argument to specifically mention all the nodes in  a graph partition.
    
    * Update convert_partition.py
    
    Moving the changes to the function "create_metadata_json" function to the "multiple-file-format" support, where this change is more appropriate. Since multiple machine testing was done with these code changes.
    
    * Addressing review comments.
    
    Removed space as suggested at the end of the line
    3ccd973c
convert_partition.py 10.9 KB