Skip to content
Snippets Groups Projects

Add autodiff

Closed
Stephan Seitzrequested to merge
seitz/pystencils:autodiff into master
1 open thread

Draft for minimal integration of automatic differentiation. Tensorflow and Torch backend were removed (apart from AutoDiffOp.create_tensorlfow())

Only tests without LBM or tf/torch dependencies have been added. This implies that also numeric gradient checking is missing (depends on either tf or torch). Should we really move the backends into separate modules?

Apart from adding auto-differentiation functionality I added two more changes. Would be happy to AutoDiffOp.create_tensorflow() after feedback.

Edited by Martin Bauer

Merge request reports

Pipeline #16132 passed

Pipeline passed for 0bcd0156 on seitz:autodiff

Closed by Stephan SeitzStephan Seitz Jul 11, 2019 (Jul 11, 2019 10:12am UTC)

Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
265 270 elif isinstance(cpu_vectorize_info, dict):
266 271 vectorize(ast, **cpu_vectorize_info)
267 272 return ast
273
274
275 def make_python_function(kernel_function_node, target='cpu', argument_dict=None):
  • Martin Bauer unmarked as a Work In Progress

    unmarked as a Work In Progress

  • Please register or sign in to reply
    Loading