cbackend.py 20 KB
Newer Older
Martin Bauer's avatar
Martin Bauer committed
1
import sympy as sp
Martin Bauer's avatar
Martin Bauer committed
2
3
from collections import namedtuple
from sympy.core import S
4
from typing import Set
5
from sympy.printing.ccode import C89CodePrinter
6
7
8

from pystencils.fast_approximation import fast_division, fast_sqrt, fast_inv_sqrt

Martin Bauer's avatar
Martin Bauer committed
9
10
try:
    from sympy.printing.ccode import C99CodePrinter as CCodePrinter
Martin Bauer's avatar
Martin Bauer committed
11
12
except ImportError:
    from sympy.printing.ccode import CCodePrinter  # for sympy versions < 1.1
Martin Bauer's avatar
Martin Bauer committed
13

Martin Bauer's avatar
Martin Bauer committed
14
from pystencils.integer_functions import bitwise_xor, bit_shift_right, bit_shift_left, bitwise_and, \
Martin Bauer's avatar
Martin Bauer committed
15
    bitwise_or, modulo_ceil
16
from pystencils.astnodes import Node, KernelFunction
Martin Bauer's avatar
Martin Bauer committed
17
from pystencils.data_types import create_type, PointerType, get_type_of_expression, VectorType, cast_func, \
18
    vector_memory_access, reinterpret_cast_func
19

20
__all__ = ['generate_c', 'CustomCodeNode', 'PrintNode', 'get_headers', 'CustomSympyPrinter']
21

Martin Bauer's avatar
Martin Bauer committed
22

23
def generate_c(ast_node: Node, signature_only: bool = False, dialect='c') -> str:
Martin Bauer's avatar
Martin Bauer committed
24
25
26
27
28
29
30
31
32
    """Prints an abstract syntax tree node as C or CUDA code.

    This function does not need to distinguish between C, C++ or CUDA code, it just prints 'C-like' code as encoded
    in the abstract syntax tree (AST). The AST is built differently for C or CUDA by calling different create_kernel
    functions.

    Args:
        ast_node:
        signature_only:
33
        dialect: 'c' or 'cuda'
Martin Bauer's avatar
Martin Bauer committed
34
35
    Returns:
        C-like code for the ast node and its descendants
Martin Bauer's avatar
Martin Bauer committed
36
    """
37
    printer = CBackend(signature_only=signature_only,
38
39
                       vector_instruction_set=ast_node.instruction_set,
                       dialect=dialect)
Martin Bauer's avatar
Martin Bauer committed
40
    return printer(ast_node)
41
42


Martin Bauer's avatar
Martin Bauer committed
43
44
def get_headers(ast_node: Node) -> Set[str]:
    """Return a set of header files, necessary to compile the printed C-like code."""
45
46
    headers = set()

Martin Bauer's avatar
Martin Bauer committed
47
48
49
    if isinstance(ast_node, KernelFunction) and ast_node.instruction_set:
        headers.update(ast_node.instruction_set['headers'])

Martin Bauer's avatar
Martin Bauer committed
50
51
52
    if hasattr(ast_node, 'headers'):
        headers.update(ast_node.headers)
    for a in ast_node.args:
53
        if isinstance(a, Node):
Martin Bauer's avatar
Martin Bauer committed
54
            headers.update(get_headers(a))
55
56

    return headers
57
58


59
60
61
# --------------------------------------- Backend Specific Nodes -------------------------------------------------------


62
class CustomCodeNode(Node):
Martin Bauer's avatar
Martin Bauer committed
63
    def __init__(self, code, symbols_read, symbols_defined, parent=None):
64
        super(CustomCodeNode, self).__init__(parent=parent)
65
        self._code = "\n" + code
Martin Bauer's avatar
Martin Bauer committed
66
67
        self._symbolsRead = set(symbols_read)
        self._symbolsDefined = set(symbols_defined)
68
        self.headers = []
69

70
    def get_code(self, dialect, vector_instruction_set):
71
72
73
74
75
76
77
        return self._code

    @property
    def args(self):
        return []

    @property
Martin Bauer's avatar
Martin Bauer committed
78
    def symbols_defined(self):
79
80
81
        return self._symbolsDefined

    @property
Martin Bauer's avatar
Martin Bauer committed
82
83
    def undefined_symbols(self):
        return self.symbols_defined - self._symbolsRead
84
85


86
class PrintNode(CustomCodeNode):
Martin Bauer's avatar
Martin Bauer committed
87
88
89
90
    # noinspection SpellCheckingInspection
    def __init__(self, symbol_to_print):
        code = '\nstd::cout << "%s  =  " << %s << std::endl; \n' % (symbol_to_print.name, symbol_to_print.name)
        super(PrintNode, self).__init__(code, symbols_read=[symbol_to_print], symbols_defined=set())
91
        self.headers.append("<iostream>")
92
93
94
95


# ------------------------------------------- Printer ------------------------------------------------------------------

96

Martin Bauer's avatar
Martin Bauer committed
97
98
# noinspection PyPep8Naming
class CBackend:
99

100
    def __init__(self, sympy_printer=None,
101
                 signature_only=False, vector_instruction_set=None, dialect='c'):
Martin Bauer's avatar
Martin Bauer committed
102
103
        if sympy_printer is None:
            if vector_instruction_set is not None:
104
                self.sympy_printer = VectorizedCustomSympyPrinter(vector_instruction_set, dialect)
105
            else:
106
                self.sympy_printer = CustomSympyPrinter(dialect)
107
        else:
Martin Bauer's avatar
Martin Bauer committed
108
            self.sympy_printer = sympy_printer
109

110
        self._vector_instruction_set = vector_instruction_set
111
        self._indent = "   "
112
        self._dialect = dialect
Martin Bauer's avatar
Martin Bauer committed
113
        self._signatureOnly = signature_only
114
115

    def __call__(self, node):
Martin Bauer's avatar
Martin Bauer committed
116
        prev_is = VectorType.instruction_set
117
        VectorType.instruction_set = self._vector_instruction_set
118
        result = str(self._print(node))
Martin Bauer's avatar
Martin Bauer committed
119
        VectorType.instruction_set = prev_is
120
        return result
121
122
123

    def _print(self, node):
        for cls in type(node).__mro__:
Martin Bauer's avatar
Martin Bauer committed
124
125
126
127
            method_name = "_print_" + cls.__name__
            if hasattr(self, method_name):
                return getattr(self, method_name)(node)
        raise NotImplementedError("CBackend does not support node of type " + str(type(node)))
128
129

    def _print_KernelFunction(self, node):
130
        function_arguments = ["%s %s" % (str(s.symbol.dtype), s.symbol.name) for s in node.get_parameters()]
Martin Bauer's avatar
Martin Bauer committed
131
        func_declaration = "FUNC_PREFIX void %s(%s)" % (node.function_name, ", ".join(function_arguments))
132
        if self._signatureOnly:
Martin Bauer's avatar
Martin Bauer committed
133
            return func_declaration
134

135
        body = self._print(node.body)
Martin Bauer's avatar
Martin Bauer committed
136
        return func_declaration + "\n" + body
137
138

    def _print_Block(self, node):
Martin Bauer's avatar
Martin Bauer committed
139
140
        block_contents = "\n".join([self._print(child) for child in node.args])
        return "{\n%s\n}" % (self._indent + self._indent.join(block_contents.splitlines(True)))
141
142

    def _print_PragmaBlock(self, node):
Martin Bauer's avatar
Martin Bauer committed
143
        return "%s\n%s" % (node.pragma_line, self._print_Block(node))
144
145

    def _print_LoopOverCoordinate(self, node):
Martin Bauer's avatar
Martin Bauer committed
146
        counter_symbol = node.loop_counter_name
Martin Bauer's avatar
Martin Bauer committed
147
148
149
150
        start = "int %s = %s" % (counter_symbol, self.sympy_printer.doprint(node.start))
        condition = "%s < %s" % (counter_symbol, self.sympy_printer.doprint(node.stop))
        update = "%s += %s" % (counter_symbol, self.sympy_printer.doprint(node.step),)
        loop_str = "for (%s; %s; %s)" % (start, condition, update)
151

Martin Bauer's avatar
Martin Bauer committed
152
        prefix = "\n".join(node.prefix_lines)
153
154
        if prefix:
            prefix += "\n"
Martin Bauer's avatar
Martin Bauer committed
155
        return "%s%s\n%s" % (prefix, loop_str, self._print(node.body))
156
157

    def _print_SympyAssignment(self, node):
Martin Bauer's avatar
Martin Bauer committed
158
159
        if node.is_declaration:
            data_type = "const " + str(node.lhs.dtype) + " " if node.is_const else str(node.lhs.dtype) + " "
160
161
            return "%s%s = %s;" % (data_type, self.sympy_printer.doprint(node.lhs),
                                   self.sympy_printer.doprint(node.rhs))
162
        else:
Martin Bauer's avatar
Martin Bauer committed
163
            lhs_type = get_type_of_expression(node.lhs)
Martin Bauer's avatar
Martin Bauer committed
164
165
166
167
168
169
            if type(lhs_type) is VectorType and isinstance(node.lhs, cast_func):
                arg, data_type, aligned, nontemporal = node.lhs.args
                instr = 'storeU'
                if aligned:
                    instr = 'stream' if nontemporal else 'storeA'

170
171
172
173
174
175
                rhs_type = get_type_of_expression(node.rhs)
                if type(rhs_type) is not VectorType:
                    rhs = cast_func(node.rhs, VectorType(rhs_type))
                else:
                    rhs = node.rhs

176
177
                return self._vector_instruction_set[instr].format("&" + self.sympy_printer.doprint(node.lhs.args[0]),
                                                                  self.sympy_printer.doprint(rhs)) + ';'
178
            else:
Martin Bauer's avatar
Martin Bauer committed
179
                return "%s = %s;" % (self.sympy_printer.doprint(node.lhs), self.sympy_printer.doprint(node.rhs))
180
181

    def _print_TemporaryMemoryAllocation(self, node):
182
        align = 64
Martin Bauer's avatar
Martin Bauer committed
183
184
185
186
187
188
        np_dtype = node.symbol.dtype.base_type.numpy_dtype
        required_size = np_dtype.itemsize * node.size + align
        size = modulo_ceil(required_size, align)
        code = "{dtype} {name}=({dtype})aligned_alloc({align}, {size}) + {offset};"
        return code.format(dtype=node.symbol.dtype,
                           name=self.sympy_printer.doprint(node.symbol.name),
189
                           size=self.sympy_printer.doprint(size),
Martin Bauer's avatar
Martin Bauer committed
190
191
                           offset=int(node.offset(align)),
                           align=align)
192
193

    def _print_TemporaryMemoryFree(self, node):
194
        align = 64
Martin Bauer's avatar
Martin Bauer committed
195
        return "free(%s - %d);" % (self.sympy_printer.doprint(node.symbol.name), node.offset(align))
196

197
198
    def _print_CustomCodeNode(self, node):
        return node.get_code(self._dialect, self._vector_instruction_set)
199

200
    def _print_Conditional(self, node):
Martin Bauer's avatar
Martin Bauer committed
201
202
        condition_expr = self.sympy_printer.doprint(node.condition_expr)
        true_block = self._print_Block(node.true_block)
Martin Bauer's avatar
Martin Bauer committed
203
        result = "if (%s)\n%s " % (condition_expr, true_block)
Martin Bauer's avatar
Martin Bauer committed
204
205
        if node.false_block:
            false_block = self._print_Block(node.false_block)
Martin Bauer's avatar
Martin Bauer committed
206
            result += "else " + false_block
207
208
        return result

209
210
211
212

# ------------------------------------------ Helper function & classes -------------------------------------------------


Martin Bauer's avatar
Martin Bauer committed
213
# noinspection PyPep8Naming
214
class CustomSympyPrinter(CCodePrinter):
Martin Bauer's avatar
Martin Bauer committed
215

216
    def __init__(self, dialect):
Martin Bauer's avatar
Martin Bauer committed
217
        super(CustomSympyPrinter, self).__init__()
218
        self._float_type = create_type("float32")
219
        self._dialect = dialect
220
221
222
223
        if 'Min' in self.known_functions:
            del self.known_functions['Min']
        if 'Max' in self.known_functions:
            del self.known_functions['Max']
Martin Bauer's avatar
Martin Bauer committed
224

225
226
227
    def _print_Pow(self, expr):
        """Don't use std::pow function, for small integer exponents, write as multiplication"""
        if expr.exp.is_integer and expr.exp.is_number and 0 < expr.exp < 8:
228
            return "(" + self._print(sp.Mul(*[expr.base] * expr.exp, evaluate=False)) + ")"
229
230
        elif expr.exp.is_integer and expr.exp.is_number and - 8 < expr.exp < 0:
            return "1 / ({})".format(self._print(sp.Mul(*[expr.base] * (-expr.exp), evaluate=False)))
231
232
233
234
235
        else:
            return super(CustomSympyPrinter, self)._print_Pow(expr)

    def _print_Rational(self, expr):
        """Evaluate all rationals i.e. print 0.25 instead of 1.0/4.0"""
Martin Bauer's avatar
Martin Bauer committed
236
237
        res = str(expr.evalf().num)
        return res
238
239
240
241
242
243
244
245

    def _print_Equality(self, expr):
        """Equality operator is not printable in default printer"""
        return '((' + self._print(expr.lhs) + ") == (" + self._print(expr.rhs) + '))'

    def _print_Piecewise(self, expr):
        """Print piecewise in one line (remove newlines)"""
        result = super(CustomSympyPrinter, self)._print_Piecewise(expr)
Martin Bauer's avatar
Martin Bauer committed
246
247
        return result.replace("\n", "")

248
    def _print_Function(self, expr):
249
        infix_functions = {
Martin Bauer's avatar
Martin Bauer committed
250
251
252
253
254
            bitwise_xor: '^',
            bit_shift_right: '>>',
            bit_shift_left: '<<',
            bitwise_or: '|',
            bitwise_and: '&',
Martin Bauer's avatar
Martin Bauer committed
255
        }
Martin Bauer's avatar
Martin Bauer committed
256
257
        if hasattr(expr, 'to_c'):
            return expr.to_c(self._print)
258
259
260
261
        if isinstance(expr, reinterpret_cast_func):
            arg, data_type = expr.args
            return "*((%s)(& %s))" % (PointerType(data_type, restrict=False), self._print(arg))
        elif isinstance(expr, cast_func):
Martin Bauer's avatar
Martin Bauer committed
262
            arg, data_type = expr.args
263
264
265
            if isinstance(arg, sp.Number):
                return self._typed_number(arg, data_type)
            else:
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
                return "((%s)(%s))" % (data_type, self._print(arg))
        elif isinstance(expr, fast_division):
            if self._dialect == "cuda":
                return "__fdividef(%s, %s)" % tuple(self._print(a) for a in expr.args)
            else:
                return "({})".format(self._print(expr.args[0] / expr.args[1]))
        elif isinstance(expr, fast_sqrt):
            if self._dialect == "cuda":
                return "__fsqrt_rn(%s)" % tuple(self._print(a) for a in expr.args)
            else:
                return "({})".format(self._print(sp.sqrt(expr.args[0])))
        elif isinstance(expr, fast_inv_sqrt):
            if self._dialect == "cuda":
                return "__frsqrt_rn(%s)" % tuple(self._print(a) for a in expr.args)
            else:
                return "({})".format(self._print(1 / sp.sqrt(expr.args[0])))
282
283
        elif expr.func in infix_functions:
            return "(%s %s %s)" % (self._print(expr.args[0]), infix_functions[expr.func], self._print(expr.args[1]))
284
        else:
285
            return super(CustomSympyPrinter, self)._print_Function(expr)
Martin Bauer's avatar
Martin Bauer committed
286

287
288
    def _typed_number(self, number, dtype):
        res = self._print(number)
289
        if dtype.is_float():
290
291
292
293
294
295
296
297
            if dtype == self._float_type:
                if '.' not in res:
                    res += ".0f"
                else:
                    res += "f"
            return res
        else:
            return res
298

299
300
301
    _print_Max = C89CodePrinter._print_Max
    _print_Min = C89CodePrinter._print_Min

302

Martin Bauer's avatar
Martin Bauer committed
303
# noinspection PyPep8Naming
304
305
306
class VectorizedCustomSympyPrinter(CustomSympyPrinter):
    SummandInfo = namedtuple("SummandInfo", ['sign', 'term'])

307
308
    def __init__(self, instruction_set, dialect):
        super(VectorizedCustomSympyPrinter, self).__init__(dialect=dialect)
Martin Bauer's avatar
Martin Bauer committed
309
        self.instruction_set = instruction_set
310

Martin Bauer's avatar
Martin Bauer committed
311
312
313
314
    def _scalarFallback(self, func_name, expr, *args, **kwargs):
        expr_type = get_type_of_expression(expr)
        if type(expr_type) is not VectorType:
            return getattr(super(VectorizedCustomSympyPrinter, self), func_name)(expr, *args, **kwargs)
315
        else:
Martin Bauer's avatar
Martin Bauer committed
316
            assert self.instruction_set['width'] == expr_type.width
317
318
            return None

319
    def _print_Function(self, expr):
320
        if isinstance(expr, vector_memory_access):
Martin Bauer's avatar
Martin Bauer committed
321
322
323
            arg, data_type, aligned, _ = expr.args
            instruction = self.instruction_set['loadA'] if aligned else self.instruction_set['loadU']
            return instruction.format("& " + self._print(arg))
324
        elif isinstance(expr, cast_func):
Martin Bauer's avatar
Martin Bauer committed
325
326
            arg, data_type = expr.args
            if type(data_type) is VectorType:
Martin Bauer's avatar
Martin Bauer committed
327
                return self.instruction_set['makeVec'].format(self._print(arg))
328
        elif expr.func == fast_division:
329
330
331
            result = self._scalarFallback('_print_Function', expr)
            if not result:
                return self.instruction_set['/'].format(self._print(expr.args[0]), self._print(expr.args[1]))
332
333
334
        elif expr.func == fast_sqrt:
            return "({})".format(self._print(sp.sqrt(expr.args[0])))
        elif expr.func == fast_inv_sqrt:
335
336
337
338
339
340
            result = self._scalarFallback('_print_Function', expr)
            if not result:
                if self.instruction_set['rsqrt']:
                    return self.instruction_set['rsqrt'].format(self._print(expr.args[0]))
                else:
                    return "({})".format(self._print(1 / sp.sqrt(expr.args[0])))
341
342
        return super(VectorizedCustomSympyPrinter, self)._print_Function(expr)

343
344
345
346
347
    def _print_And(self, expr):
        result = self._scalarFallback('_print_And', expr)
        if result:
            return result

Martin Bauer's avatar
Martin Bauer committed
348
349
350
351
        arg_strings = [self._print(a) for a in expr.args]
        assert len(arg_strings) > 0
        result = arg_strings[0]
        for item in arg_strings[1:]:
Martin Bauer's avatar
Martin Bauer committed
352
            result = self.instruction_set['&'].format(result, item)
353
354
355
356
357
358
359
        return result

    def _print_Or(self, expr):
        result = self._scalarFallback('_print_Or', expr)
        if result:
            return result

Martin Bauer's avatar
Martin Bauer committed
360
361
362
363
        arg_strings = [self._print(a) for a in expr.args]
        assert len(arg_strings) > 0
        result = arg_strings[0]
        for item in arg_strings[1:]:
Martin Bauer's avatar
Martin Bauer committed
364
            result = self.instruction_set['|'].format(result, item)
365
366
        return result

367
    def _print_Add(self, expr, order=None):
368
369
370
        result = self._scalarFallback('_print_Add', expr)
        if result:
            return result
371
372
373
374

        summands = []
        for term in expr.args:
            if term.func == sp.Mul:
Martin Bauer's avatar
Martin Bauer committed
375
                sign, t = self._print_Mul(term, inside_add=True)
376
377
378
379
380
381
382
383
384
385
386
387
388
            else:
                t = self._print(term)
                sign = 1
            summands.append(self.SummandInfo(sign, t))
        # Use positive terms first
        summands.sort(key=lambda e: e.sign, reverse=True)
        # if no positive term exists, prepend a zero
        if summands[0].sign == -1:
            summands.insert(0, self.SummandInfo(1, "0"))

        assert len(summands) >= 2
        processed = summands[0].term
        for summand in summands[1:]:
Martin Bauer's avatar
Martin Bauer committed
389
            func = self.instruction_set['-'] if summand.sign == -1 else self.instruction_set['+']
390
391
392
            processed = func.format(processed, summand.term)
        return processed

393
    def _print_Pow(self, expr):
394
395
396
        result = self._scalarFallback('_print_Pow', expr)
        if result:
            return result
397

398
399
        one = self.instruction_set['makeVec'].format(1.0)

400
401
        if expr.exp.is_integer and expr.exp.is_number and 0 < expr.exp < 8:
            return "(" + self._print(sp.Mul(*[expr.base] * expr.exp, evaluate=False)) + ")"
402
403
404
405
406
        elif expr.exp == -1:
            one = self.instruction_set['makeVec'].format(1.0)
            return self.instruction_set['/'].format(one, self._print(expr.base))
        elif expr.exp == 0.5:
            return self.instruction_set['sqrt'].format(self._print(expr.base))
407
408
409
        elif expr.exp == -0.5:
            root = self.instruction_set['sqrt'].format(self._print(expr.base))
            return self.instruction_set['/'].format(one, root)
410
411
412
        elif expr.exp.is_integer and expr.exp.is_number and - 8 < expr.exp < 0:
            return self.instruction_set['/'].format(one,
                                                    self._print(sp.Mul(*[expr.base] * (-expr.exp), evaluate=False)))
413
        else:
414
            raise ValueError("Generic exponential not supported: " + str(expr))
415

Martin Bauer's avatar
Martin Bauer committed
416
417
418
419
    def _print_Mul(self, expr, inside_add=False):
        # noinspection PyProtectedMember
        from sympy.core.mul import _keep_coeff

420
421
422
        result = self._scalarFallback('_print_Mul', expr)
        if result:
            return result
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450

        c, e = expr.as_coeff_Mul()
        if c < 0:
            expr = _keep_coeff(-c, e)
            sign = -1
        else:
            sign = 1

        a = []  # items in the numerator
        b = []  # items that are in the denominator (if any)

        # Gather args for numerator/denominator
        for item in expr.as_ordered_factors():
            if item.is_commutative and item.is_Pow and item.exp.is_Rational and item.exp.is_negative:
                if item.exp != -1:
                    b.append(sp.Pow(item.base, -item.exp, evaluate=False))
                else:
                    b.append(sp.Pow(item.base, -item.exp))
            else:
                a.append(item)

        a = a or [S.One]

        a_str = [self._print(x) for x in a]
        b_str = [self._print(x) for x in b]

        result = a_str[0]
        for item in a_str[1:]:
Martin Bauer's avatar
Martin Bauer committed
451
            result = self.instruction_set['*'].format(result, item)
452
453
454
455

        if len(b) > 0:
            denominator_str = b_str[0]
            for item in b_str[1:]:
Martin Bauer's avatar
Martin Bauer committed
456
457
                denominator_str = self.instruction_set['*'].format(denominator_str, item)
            result = self.instruction_set['/'].format(result, denominator_str)
458

Martin Bauer's avatar
Martin Bauer committed
459
        if inside_add:
460
461
462
            return sign, result
        else:
            if sign < 0:
Martin Bauer's avatar
Martin Bauer committed
463
                return self.instruction_set['*'].format(self._print(S.NegativeOne), result)
464
465
466
            else:
                return result

467
    def _print_Relational(self, expr):
468
469
470
        result = self._scalarFallback('_print_Relational', expr)
        if result:
            return result
Martin Bauer's avatar
Martin Bauer committed
471
        return self.instruction_set[expr.rel_op].format(self._print(expr.lhs), self._print(expr.rhs))
472
473

    def _print_Equality(self, expr):
474
475
476
        result = self._scalarFallback('_print_Equality', expr)
        if result:
            return result
Martin Bauer's avatar
Martin Bauer committed
477
        return self.instruction_set['=='].format(self._print(expr.lhs), self._print(expr.rhs))
478
479

    def _print_Piecewise(self, expr):
480
481
482
        result = self._scalarFallback('_print_Piecewise', expr)
        if result:
            return result
483

Martin Bauer's avatar
Martin Bauer committed
484
        if expr.args[-1].cond.args[0] is not sp.sympify(True):
485
486
487
488
489
490
491
492
493
            # We need the last conditional to be a True, otherwise the resulting
            # function may not return a result.
            raise ValueError("All Piecewise expressions must contain an "
                             "(expr, True) statement to be used as a default "
                             "condition. Without one, the generated "
                             "expression may not evaluate to anything under "
                             "some condition.")

        result = self._print(expr.args[-1][0])
Martin Bauer's avatar
Martin Bauer committed
494
        for true_expr, condition in reversed(expr.args[:-1]):
Martin Bauer's avatar
Martin Bauer committed
495
            # noinspection SpellCheckingInspection
Martin Bauer's avatar
Martin Bauer committed
496
            result = self.instruction_set['blendv'].format(result, self._print(true_expr), self._print(condition))
497
        return result