瀏覽代碼

starpufft/tests/testx.c: fix call to STARPUFFT(free) (followup to a4e0ec5aede2bb037ee6f05d321911764f097448)

Nathalie Furmento 4 年之前
父節點
當前提交
9753d59196
共有 1 個文件被更改,包括 4 次插入4 次删除
  1. 4 4
      starpufft/tests/testx.c

+ 4 - 4
starpufft/tests/testx.c

@@ -298,12 +298,12 @@ int main(int argc, char *argv[])
 #endif
 #endif
 
-	STARPUFFT(free)(in_orig);
-	STARPUFFT(free)(in);
-	STARPUFFT(free)(out);
+	STARPUFFT(free)(in_orig, size * sizeof(*in_orig));
+	STARPUFFT(free)(in, size * sizeof(*in));
+	STARPUFFT(free)(out, size * sizeof(*out));
 
 #ifdef STARPU_HAVE_FFTW
-	STARPUFFT(free)(out_fftw);
+	STARPUFFT(free)(out_fftw, size * sizeof(*out_fftw));
 #endif
 
 #ifdef STARPU_USE_CUDA