e25c3820aaa1ebb9c0bf7731467e24a2a2592a54,sru/cuda_functional.py,SRU_Compute_GPU,forward,#SRU_Compute_GPU#Any#Any#Any#Any#Any#Any#,488

Before Change

        else:
            x_ptr = 0

        stream, fwd_func, bifwd_func, _, _ = self.get_functions()
        FUNC = fwd_func if not self.bidirectional else bifwd_func
        FUNC(args=[
            u.contiguous().data_ptr(),

After Change

        init_ = x.new(ncols).zero_() if init is None else init
        size = (length, batch, d * bidir) if x.dim() == 3 else (batch, d * bidir)
        c = x.new(*size).zero_()
        h = x.new(*size).zero_()

        scale_x = self.scale_x
        if skip_type > 0 and k_ == 3:
            x_ = x.contiguous() * scale_x if scale_x is not None else x.contiguous()

Instances