Vectorize write barrier (rust-lang#823)

wsmoses · web-flow · commit 1bd0e8276fb0 · 2022-09-05T06:46:16.000Z
diff --git a/enzyme/Enzyme/AdjointGenerator.h b/enzyme/Enzyme/AdjointGenerator.h
@@ -9851,7 +9851,10 @@ class AdjointGenerator
             {
               if (!gutils->isConstantValue(arg)) {
                 Value *ptrshadow = gutils->invertPointerM(arg, BuilderZ);
-                iargs.push_back(ptrshadow);
+                applyChainRule(
+                    BuilderZ,
+                    [&](Value *ptrshadow) { iargs.push_back(ptrshadow); },
+                    ptrshadow);
               }
             }
             if (iargs.size()) {

Original file line number	Diff line number	Diff line change
`@@ -9851,7 +9851,10 @@ class AdjointGenerator`
`9851`	`9851`	`{`
`9852`	`9852`	`if (!gutils->isConstantValue(arg)) {`
`9853`	`9853`	`Value *ptrshadow = gutils->invertPointerM(arg, BuilderZ);`
`9854`		`- iargs.push_back(ptrshadow);`
	`9854`	`+ applyChainRule(`
	`9855`	`+ BuilderZ,`
	`9856`	`+ [&](Value *ptrshadow) { iargs.push_back(ptrshadow); },`
	`9857`	`+ ptrshadow);`
`9855`	`9858`	`}`
`9856`	`9859`	`}`
`9857`	`9860`	`if (iargs.size()) {`