Add CustomSympyPrinter._print_Sum
1 unresolved thread
1 unresolved thread
This makes sympy.Sum printable as instantaniously invoked lambda (Attention: C++-only, works in CUDA)
Merge request reports
Activity
385 385 return res 386 386 387 def _print_Sum(self, expr): 388 template = jinja2.Template( 389 """[&]() { 390 {{dtype}} sum = ({{dtype}}) 0; 391 for ( {{iterator_dtype}} {{var}} = {{start}}; {{condition}}; {{var}} += {{increment}} ) { 392 sum += {{expr}}; 393 } 394 return sum; 395 }()""") 396 var = expr.limits[0][0] 397 start = expr.limits[0][1] 398 end = expr.limits[0][2] 399 code = template.render( 400 dtype='float',
Please register or sign in to reply