change the way of memory allocation

buensons · Dec 13, 2020 · 2ff3df7 · 2ff3df7
1 parent 69e5d09
commit 2ff3df7
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/TimeSeriesPredictor.cu b/src/TimeSeriesPredictor.cu
@@ -17,7 +17,7 @@ TimeSeriesPredictor::~TimeSeriesPredictor() {
     gpuErrchk(cudaFree(this->mapWeightsGpu));
     gpuErrchk(cudaFree(this->dataGpu));
     gpuErrchk(cudaFree(this->mapInputGpu));
-    free(this->fitnessHost);
+    delete [] this->fitnessHost;
 }
 
 auto TimeSeriesPredictor::prepareGpuMemory() -> void {
@@ -27,9 +27,7 @@ auto TimeSeriesPredictor::prepareGpuMemory() -> void {
     gpuErrchk(cudaMalloc((void**)&this->mapWeightsGpu, this->populationSize * this->numberOfNodes * this->numberOfNodes * sizeof(float)));
     gpuErrchk(cudaMalloc((void**)&this->mapInputGpu, this->populationSize * this->numberOfNodes * sizeof(float)));
 
-    if((this->fitnessHost = (float *)malloc(this->populationSize * sizeof(float))) == NULL) {
-        std::cerr << "malloc\n";
-    }
+    this->fitnessHost = new float[this->populationSize];
     gpuErrchk(cudaMemcpy(this->dataGpu, this->data.data(), this->data.size() * sizeof(float), cudaMemcpyHostToDevice));
 }