Clone printf and fprintf calls

vgvassilev · Nov 20, 2024 · 0e87928 · 0e87928
1 parent 8d916fe
commit 0e87928
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/lib/Differentiator/ReverseModeVisitor.cpp b/lib/Differentiator/ReverseModeVisitor.cpp
@@ -1456,6 +1456,9 @@ Expr* getArraySizeExpr(const ArrayType* AT, ASTContext& context,
       return StmtDiff(Clone(CE));
     }
 
+    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
@@ -504,6 +504,7 @@ double fn_memory(double *out, double *in) {
 //CHECK-NEXT:}
 
 void launch_add_kernel_4(int *out, int *in, const int N) {
+  printf("Launching add_kernel_4 for size: %d\n", N);
   int *in_dev = nullptr;
   cudaMalloc(&in_dev, N * sizeof(int));
   cudaMemcpy(in_dev, in, N * sizeof(int), cudaMemcpyHostToDevice);
@@ -520,6 +521,7 @@ void launch_add_kernel_4(int *out, int *in, const int N) {
 
 // CHECK: void launch_add_kernel_4_grad_0_1(int *out, int *in, const int N, int *_d_out, int *_d_in) {
 //CHECK-NEXT:    int _d_N = 0;
+//CHECK-NEXT:    printf("Launching add_kernel_4 for size: %d\n", N);
 //CHECK-NEXT:    int *_d_in_dev = nullptr;
 //CHECK-NEXT:    int *in_dev = nullptr;
 //CHECK-NEXT:    cudaMalloc(&_d_in_dev, N * sizeof(int));