patch_dgl

class patch_dgl

Patches DGL so that attention layers (gatconv, dotgatconv, agnngatconv) use a different edge_softmax function that supports sar.core.GraphShardManager. Also modifies DGL’s RelGraphConv to add a decorator to its message function to tell SAR how to find the parameters used to create edge messages.