unit test for bytes allocated

Author: alphaville

include/tensor.cuh

@@ -1,3 +1,6 @@
+#ifndef TENSOR_CUH
+#define TENSOR_CUH
+
 #include <random>
 #include <algorithm>
 #include <iterator>
@@ -13,8 +16,6 @@
 #include <cassert>
 #include <fstream>
 
-#ifndef TENSOR_CUH
-#define TENSOR_CUH
 
 /**
  * Define defaults

main.cu

@@ -8,13 +8,16 @@
 int main() {
     /* Write to binary file */
     auto r = DTensor<double>::createRandomTensor(3, 6, 4, -1, 1);
+    auto r2 = DTensor<double>::createRandomTensor(300, 600, 4, -1, 1);
     std::string fName = "tensor.bt"; // binary tensor file extension: .bt
     r.saveToFile(fName);
 
     /* Parse binary file */
     auto recov = DTensor<double>::parseFromFile(fName);
     auto err = r - recov;
-    std::cout << "max error : " << err.maxAbs();
-    std::cout << "Memory: " << Session::getInstance().totalAllocatedBytes() << " B" << std::endl;
+    std::cout << "max error : " << err.maxAbs() << std::endl;
+    std::cout << "Memory: " << std::setprecision(3)
+            << (float) Session::getInstance().totalAllocatedBytes() / 1e6
+    << " MB" << std::endl;
     return 0;
 }