60a342edc8b501802135df44869353cc8604d838,gpytorch/lazy/kronecker_product_lazy_variable.py,KroneckerProductLazyVariable,_matmul_closure_factory,#KroneckerProductLazyVariable#,31

Before Change

            def closure(mat2):
                return sym_kronecker_product_toeplitz_matmul(columns, mat2)

        elif len(args) == 3:
            columns, W_lefts, W_rights = args

            def closure(mat2):
                return kp_interpolated_toeplitz_matmul(columns, mat2, W_lefts, W_rights, None)

        elif len(args) == 4:
            columns, W_lefts, W_rights, added_diag = args

            def closure(mat2):
                return kp_interpolated_toeplitz_matmul(columns, mat2, W_lefts, W_rights, added_diag)

        else:
            raise AttributeError("Invalid number of arguments")

        return closure

    def _derivative_quadratic_form_factory(self, *args):

After Change

        i = 0
        for lazy_var in self.lazy_vars:
            len_repr = len(lazy_var.representation())
            sub_matmul_closure = lazy_var._matmul_closure_factory(*args[i:i + len_repr])
            sub_matmul_closures.append(sub_matmul_closure)
            i = i + len_repr

Instances