Clone printf and fprintf calls (#1147)

* Clone printf and fprintf calls * Add FIXME and print loop index
vgvassilev · Nov 22, 2024 · b5e0d12 · b5e0d12
1 parent 284c33e
commit b5e0d12
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/Differentiator/ReverseModeVisitor.cpp b/lib/Differentiator/ReverseModeVisitor.cpp
@@ -1459,6 +1459,10 @@ Expr* getArraySizeExpr(const ArrayType* AT, ASTContext& context,
       return StmtDiff(Clone(CE));
     }
 
+    // FIXME: Revisit this when variadic functions are supported.
+    if (FD->getNameAsString() == "printf" || FD->getNameAsString() == "fprintf")
+      return StmtDiff(Clone(CE));
+
     Expr* CUDAExecConfig = nullptr;
     if (const auto* KCE = dyn_cast<CUDAKernelCallExpr>(CE))
       CUDAExecConfig = Clone(KCE->getConfig());

diff --git a/test/CUDA/GradientKernels.cu b/test/CUDA/GradientKernels.cu
@@ -436,6 +436,7 @@ double fn_memory(double *out, double *in) {
   cudaMemcpy(out_host, out, 10 * sizeof(double), cudaMemcpyDeviceToHost);
   double res = 0;
   for (int i=0; i < 10; ++i) {
+    printf("Writing result of out[%d]\n", i);
     res += out_host[i];
   }
   free(out_host);
@@ -469,6 +470,7 @@ double fn_memory(double *out, double *in) {
 //CHECK-NEXT:                break;
 //CHECK-NEXT:        }
 //CHECK-NEXT:        _t0++;
+//CHECK-NEXT:        printf("Writing result of out[%d]\n", i);
 //CHECK-NEXT:        clad::push(_t1, res);
 //CHECK-NEXT:        res += out_host[i];
 //CHECK-NEXT:    }