@@ -2605,8 +2605,6 @@ const Def* AutoDiffer::j_wrap_rop(ROp op, const Def* a, const Def* b) {
2605
2605
auto adiff = world_.tuple (vars_without_mem_cont (world_,middle));
2606
2606
auto bdiff = world_.tuple (vars_without_mem_cont (world_,end));
2607
2607
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})));
2610
2608
auto sum_pb=vec_add (world_,adiff,bdiff,pb->ret_var ());
2611
2609
end->set_body (world_.app (sum_pb, end->mem_var ()));
2612
2610
pullbacks_[dst] = pb;
@@ -2623,11 +2621,8 @@ const Def* AutoDiffer::j_wrap_rop(ROp op, const Def* a, const Def* b) {
2623
2621
auto za=world_.op (ROp::mul, (nat_t )0 , pb->var (1 ), a);
2624
2622
auto bsq=world_.op (ROp::mul, (nat_t )0 , b, b);
2625
2623
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));
2631
2626
auto sum_pb=vec_add (world_,adiff,bdiff,pb->ret_var ());
2632
2627
end->set_body (world_.app (sum_pb, end->mem_var ()));
2633
2628
pullbacks_[dst] = pb;
0 commit comments