add perror to example

NVIDIA · Jan 12, 2025 · cac6443 · cac6443
1 parent 865ef42
commit cac6443
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cudax/examples/file.cu b/cudax/examples/file.cu
@@ -119,7 +119,12 @@ try
   printf("waiting for the stream to finish\n");
   stream.wait();
 
-  if (numBytesWritten != numBytes)
+  if (numBytes == -1)
+  {
+    perror("Error writing file: ");
+    exit(EXIT_FAILURE);
+  }
+  else if (numBytesWritten != numBytes)
   {
     fprintf(stderr, "Error: wrote %ld bytes, expected %ld\n", numBytesWritten, numBytes);
     exit(EXIT_FAILURE);