Fix vector atomic gep (#899)

EnzymeAD · Oct 13, 2022 · 4d7f77a · 4d7f77a
1 parent 1efa420
commit 4d7f77a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/enzyme/Enzyme/GradientUtils.h b/enzyme/Enzyme/GradientUtils.h
@@ -2273,7 +2273,7 @@ class DiffeGradientUtils final : public GradientUtils {
                 ConstantInt::get(Type::getInt64Ty(vt->getContext()), 0),
                 ConstantInt::get(Type::getInt32Ty(vt->getContext()), i)};
 #if LLVM_VERSION_MAJOR > 7
-            auto vptr = BuilderM.CreateGEP(vt->getElementType(), ptr, Idxs);
+            auto vptr = BuilderM.CreateGEP(addingType, ptr, Idxs);
 #else
             auto vptr = BuilderM.CreateGEP(ptr, Idxs);
 #endif