Clone kernel call instead of recreating it

vgvassilev · Nov 9, 2024 · b851139 · b851139
1 parent b00cd15
commit b851139
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/Differentiator/ReverseModeVisitor.cpp b/lib/Differentiator/ReverseModeVisitor.cpp
@@ -2303,6 +2303,9 @@ Expr* getArraySizeExpr(const ArrayType* AT, ASTContext& context,
       m_ExternalSource->ActBeforeFinalizingVisitCallExpr(
           CE, OverloadedDerivedFn, DerivedCallArgs, CallArgDx, asGrad);
 
+    if (isa<CUDAKernelCallExpr>(CE))
+      return StmtDiff(Clone(CE));
+
     Expr* call = nullptr;
 
     QualType returnType = FD->getReturnType();