Skip to content

Commit 12cf51d

Browse files
fix div
1 parent 8fbeda5 commit 12cf51d

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

thorin/pass/rw/auto_diff.cpp

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2605,8 +2605,6 @@ const Def* AutoDiffer::j_wrap_rop(ROp op, const Def* a, const Def* b) {
26052605
auto adiff = world_.tuple(vars_without_mem_cont(world_,middle));
26062606
auto bdiff = world_.tuple(vars_without_mem_cont(world_,end));
26072607

2608-
// auto [smem, sum] = vec_add(world_, end->mem_var(), adiff, bdiff);
2609-
// end->set_body(world_.app(pb->ret_var(), flat_tuple({ smem, sum})));
26102608
auto sum_pb=vec_add(world_,adiff,bdiff,pb->ret_var());
26112609
end->set_body(world_.app(sum_pb, end->mem_var()));
26122610
pullbacks_[dst] = pb;
@@ -2623,11 +2621,8 @@ const Def* AutoDiffer::j_wrap_rop(ROp op, const Def* a, const Def* b) {
26232621
auto za=world_.op(ROp::mul, (nat_t)0, pb->var(1), a);
26242622
auto bsq=world_.op(ROp::mul, (nat_t)0, b, b);
26252623
middle->set_body(world_.app(bpb, {middle->mem_var(), world_.op_rminus((nat_t)0, world_.op(ROp::div, (nat_t)0, za, bsq)), end}));
2626-
auto adiff = middle->var(1);
2627-
auto bdiff = end->var(1);
2628-
2629-
// auto [smem, sum] = vec_add(world_, end->mem_var(), adiff, bdiff);
2630-
// end->set_body(world_.app(pb->ret_var(), { smem, sum}));
2624+
auto adiff = world_.tuple(vars_without_mem_cont(world_,middle));
2625+
auto bdiff = world_.tuple(vars_without_mem_cont(world_,end));
26312626
auto sum_pb=vec_add(world_,adiff,bdiff,pb->ret_var());
26322627
end->set_body(world_.app(sum_pb, end->mem_var()));
26332628
pullbacks_[dst] = pb;

0 commit comments

Comments
 (0)