fix cuda

vgvassilev · Nov 25, 2024 · de08917 · de08917
1 parent 3acb66a
commit de08917
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/Differentiator/ReverseModeVisitor.cpp b/lib/Differentiator/ReverseModeVisitor.cpp
@@ -1462,10 +1462,10 @@ Expr* getArraySizeExpr(const ArrayType* AT, ASTContext& context,
       return foundExpr;
     }
     // Not found, construct the adjoint and register it.
-    VarDecl* VDDiff =
-        BuildVarDecl(VD->getType(), CreateUniqueIdentifier(nameDiff_str),
-                     m_DerivativeFnScope->getParent()->getParent(), DC,
-                     getZeroInit(VD->getType()));
+    QualType TyDiff = getNonConstType(VD->getType(), m_Context, m_Sema);
+    VarDecl* VDDiff = BuildVarDecl(
+        TyDiff, CreateUniqueIdentifier(nameDiff_str),
+        m_DerivativeFnScope->getParent()->getParent(), DC, getZeroInit(TyDiff));
 
     DC->addDecl(VDDiff);
     DC->makeDeclVisibleInContext(VDDiff);