Free host arrays at end of program

2015-07-28 13:59:31 +01:00 · 2015-07-28 13:59:31 +01:00 · dcf734387e
commit dcf734387e
parent a9a087622d
2 changed files with 10 additions and 0 deletions
--- a/cuda-stream.cu
+++ b/cuda-stream.cu
@ -288,6 +288,11 @@ int main(int argc, char *argv[])
            << std::endl;
    }

+    // Free host vectors
+    free(h_a);
+    free(h_b);
+    free(h_c);
+
 }

 std::string getDeviceName(int device)
--- a/ocl-stream.cpp
+++ b/ocl-stream.cpp
@ -313,6 +313,11 @@ int main(int argc, char *argv[])
                << std::endl;
        }

+        // Free host vectors
+        free(h_a);
+        free(h_b);
+        free(h_c);
+
    }
    catch (cl::Error &e)
    {