Преглед на файлове

examples/openmp/vector_scal_omp.c: free memory and better deal with enodev error

Nathalie Furmento преди 11 години
родител
ревизия
a3905bb1e7
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  1. 3 0
      examples/openmp/vector_scal_omp.c

+ 3 - 0
examples/openmp/vector_scal_omp.c

@@ -124,9 +124,12 @@ int main(int argc, char **argv)
 	FPRINTF(stderr, "AFTER: First element is %f\n", vector[0]);
 	FPRINTF(stderr, "AFTER: Last element is %f\n", vector[NX-1]);
 
+	free(vector);
 	return 0;
 
 enodev:
+	starpu_data_unregister(vector_handle);
+	free(vector);
 	starpu_shutdown();
 	return 77;
 }