Skip to content
Snippets Groups Projects

Add CustomSympyPrinter._print_Sum

Closed Stephan Seitz requested to merge seitz/pystencils:print_sympy_sum into master
1 unresolved thread

This makes sympy.Sum printable as instantaniously invoked lambda (Attention: C++-only, works in CUDA)

Merge request reports

Loading
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
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',
  • closed

  • Please register or sign in to reply
    Loading