2d21747abdbf1968a9d565a4090d5d6297ade71a,encoding/functions/encoding.py,_scaledL2,forward,#Any#Any#Any#Any#,86

Before Change

    def forward(ctx, X, C, S):
        B, N, _ = X.size()
        K = C.size(0)
        with torch.cuda.device_of(X):
            SL = X.new(B, N, K)
        if isinstance(X, torch.cuda.FloatTensor):
            with torch.cuda.device_of(X):
                encoding_lib.Encoding_Float_scaledl2_forward(SL, X, C, S)
        elif isinstance(X, torch.cuda.DoubleTensor):
            with torch.cuda.device_of(X):
                encoding_lib.Encoding_Double_scaledl2_forward(SL, X, C, S)
        else:
            raise RuntimeError("Unimplemented data type!")
        ctx.save_for_backward(X, C, S, SL)
        return SL

    @staticmethod

After Change

class _scaledL2(Function):
    @staticmethod
    def forward(ctx, X, C, S):
        if X.is_cuda:
            SL = lib.gpu.scaled_l2_forward(X, C, S)
        else:
            raise NotImplemented
        ctx.save_for_backward(X, C, S, SL)
        return SL

    @staticmethod

Instances