print sample output

ROCm · Dec 4, 2023 · 259d0ae · 259d0ae
1 parent 9b4a479
commit 259d0ae
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 9 deletions.
diff --git a/samples/01_contraction/simple_bilinear_contraction.hpp b/samples/01_contraction/simple_bilinear_contraction.hpp
@@ -154,7 +154,7 @@ int bilinearContractionSample()
     /*******************
    * Initialize data
    *******************/
-    int initMethod = 0; // TODO read value from commandline
+    int initMethod = 1; // TODO read value from commandline
     for(int64_t i = 0; i < elementsA; i++)
     {
         if(initMethod == 0)
@@ -284,14 +284,9 @@ int bilinearContractionSample()
                                                0 /* stream */));
 
 #if !NDEBUG
-    bool printElements = false;
+    bool printElements = 1;
     bool storeElements = false;
 
-    if(printElements || storeElements)
-    {
-        CHECK_HIP_ERROR(hipMemcpy(C, C_d, sizeC, hipMemcpyDeviceToHost));
-    }
-
     if(printElements)
     {
         if(elementsA < MAX_ELEMENTS_PRINT_COUNT)
@@ -314,6 +309,15 @@ int bilinearContractionSample()
             hiptensorPrintArrayElements(std::cout, C, elementsC);
             std::cout << std::endl;
         }
+
+        CHECK_HIP_ERROR(hipMemcpy(C, C_d, sizeC, hipMemcpyDeviceToHost));
+
+        if(elementsC < MAX_ELEMENTS_PRINT_COUNT)
+        {
+            std::cout << "Tensor D elements:\n";
+            hiptensorPrintArrayElements(std::cout, C, elementsC);
+            std::cout << std::endl;
+        }
     }
 
     if(storeElements)
@@ -327,6 +331,12 @@ int bilinearContractionSample()
         hiptensorPrintElementsToFile(tensorB, B, elementsB, ", ");
         tensorB.close();
 
+        tensorC.open("tensor_C.txt");
+        hiptensorPrintElementsToFile(tensorC, C, elementsC, ", ");
+        tensorC.close();
+
+        CHECK_HIP_ERROR(hipMemcpy(C, C_d, sizeC, hipMemcpyDeviceToHost));
+
         tensorC.open("tensor_C_scale_contraction_results.txt");
         hiptensorPrintElementsToFile(tensorC, C, elementsC, ", ");
         tensorC.close();

diff --git a/samples/01_contraction/simple_scale_contraction.hpp b/samples/01_contraction/simple_scale_contraction.hpp
@@ -151,7 +151,7 @@ int scaleContractionSample()
     /*******************
    * Initialize data
    *******************/
-    int initMethod = 0; // TODO read the value from command line
+    int initMethod = 1; // TODO read the value from command line
     for(int64_t i = 0; i < elementsA; i++)
     {
         if(initMethod == 0)
@@ -274,7 +274,7 @@ int scaleContractionSample()
                                                0 /* stream */));
 
 #if !NDEBUG
-    bool printElements = false;
+    bool printElements = 1;
     bool storeElements = false;
 
     if(printElements || storeElements)