86fe58ebb8a046215c2a558217a7a836b254a1a8,python/dgl/nn/pytorch/softmax.py,EdgeSoftmax,backward,#Any#Any#,60

Before Change

        ctx.backward_cache = None
        g.edata["out"] = out
        g.edata["grad_s"] = out * grad_out
        g.update_all(fn.copy_e("grad_s", "m"), fn.sum("m", "accum"))
        g.apply_edges(fn.e_mul_v("out", "accum", "out"))
        grad_score = g.edata["grad_s"] - g.edata["out"]
        return None, grad_score, None

After Change

            grad_score = sds - sds * sds_sum  // multiple expressions
            return grad_score.data
        
        n_nodes, n_edges, gidx = ctx.backward_cache
        out, = ctx.saved_tensors

        //g.edata["grad_s"] = out * grad_out

Instances