diff --git a/cuda-stream.cu b/cuda-stream.cu index 9b06efd..d5eaa31 100644 --- 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) diff --git a/ocl-stream.cpp b/ocl-stream.cpp index a2f6f2f..e64548a 100644 --- 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) {