예제 #1
0
    def _generate_multiply(self, mult_op: saldag.Multiply):
        """ Generate code for Multiply operations. """

        store_code = ''
        if mult_op.is_leaf():
            store_code += self._generate_store(mult_op)

        op_cols = mult_op.operands
        operands = []
        scalar = 1

        for op_col in op_cols:
            if hasattr(op_col, 'name'):
                operands.append(mult_op.get_in_rel().name + '.' + op_col.name)
            else:
                scalar = op_col

        template = open(
            "{0}/{1}.tmpl".format(self.template_directory, 'multiply'),
            'r').read()

        data = {
            'OPERANDS': '*'.join(c for c in operands),
            'SCALAR': scalar,
            'TARGET': mult_op.target_col.name,
            'INREL': mult_op.get_in_rel().name,
            'OUTREL': mult_op.out_rel.name,
            'CACHE_VAR': cache_var(mult_op)
        }

        return pystache.render(template, data) + store_code
예제 #2
0
파일: python.py 프로젝트: yangzpag/conclave
 def _generate_multiply(self, mult_op: ccdag.Multiply):
     """ Generate code for Multiply operations. """
     operands = [self._col_or_scalar(col) for col in mult_op.operands]
     lambda_expr = "lambda row : " + " * ".join([op for op in operands])
     return "{}{} = arithmetic_project({}, {}, {})\n".format(
         self.space, mult_op.out_rel.name,
         mult_op.get_in_rel().name, mult_op.target_col.idx, lambda_expr)
예제 #3
0
 def _generate_multiply(self, mult_op: saldag.Multiply):
     """ Generate code for Multiply operations. """
     operands = [col.idx for col in mult_op.operands]
     lambda_expr = "lambda row : " + " * ".join(
         ["row[{}]".format(idx) for idx in operands])
     return "{}{} = arithmetic_project({}, {}, {})\n".format(
         self.space, mult_op.out_rel.name,
         mult_op.get_in_rel().name, mult_op.target_col.idx, lambda_expr)
예제 #4
0
    def _generate_multiply(self, multiply_op: saldag.Multiply):
        """ Generate code for Multiply operations. """

        operand_col_str = " * ".join(
            [str(col) for col in multiply_op.operands])
        return "MULTIPLY{} [{} -> {}] FROM ({}) AS {}\n".format(
            "MPC" if multiply_op.is_mpc else "", str(multiply_op.target_col),
            operand_col_str,
            multiply_op.get_in_rel().dbg_str(), multiply_op.out_rel.dbg_str())
예제 #5
0
    def _rewrite_multiply(self, node: saldag.Multiply):
        """ Push down collusion sets for a Multiply node. """

        out_rel_cols = node.out_rel.columns
        operands = node.operands
        target_col = node.target_col

        # Update target column collusion set
        target_col_out = out_rel_cols[target_col.idx]

        target_col_out.coll_sets |= utils.coll_sets_from_columns(operands)

        # The other columns weren't modified so the collusion sets
        # simply carry over
        for in_col, out_col in zip(node.get_in_rel().columns, out_rel_cols):
            if in_col != target_col:
                out_col.coll_sets |= copy.deepcopy(in_col.coll_sets)