We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e7c744e commit 9cf7f44Copy full SHA for 9cf7f44
src/rules/parallelrules.jl
@@ -466,9 +466,6 @@ end
466
467
if mode == API.DEM_ReverseModeGradient
468
v = lookup_value(gutils, v, B)
469
- if !(pv isa Nothing)
470
- pv = lookup_value(gutils, pv, B)
471
- end
472
end
473
474
else
@@ -528,8 +525,8 @@ end
528
525
529
526
530
527
if value_type(dv) != spllty
+ pv = dv
531
if width == 1
532
- pv = dv
533
dv = load!(fwdbuilder, spllty, dv)
534
if dv2 !== nothing
535
dv = recombine_value!(fwdbuilder, dv, dv2)
@@ -745,6 +742,7 @@ end
745
742
746
743
for (pv, val, pllty) in copies
747
744
ld = load!(B, pllty, val)
+ pv = lookup_value(gutils, pv, B)
748
store!(B, ld, pv)
749
750
return nothing
0 commit comments