47a96974ab2d75fc20c76a6d6da20976ecad32bb,gpytorch/random_variables/random_variable.py,RandomVariable,cpu,#RandomVariable#,16

Before Change

        representation = self.representation()
        if not isinstance(representation, tuple) or isinstance(representation, list):
            representation = representation,
        return self.__class__(*(var.cpu() for var in representation))

    def cuda(self, device_id=None):
        representation = self.representation()
        if not isinstance(representation, tuple) or isinstance(representation, list):

After Change

                new_args.append(arg.cpu())
            else:
                new_args.append(arg)
        for name, val in self._kwargs.items():
            if hasattr(val, "cpu"):
                new_kwargs[name] = val.cpu()
            else:
                new_kwargs[name] = val
        return self.__class__(*new_args, **new_kwargs)

    def cuda(self, device_id=None):
        new_args = []

Instances