Skip to content

Commit bc61768

Browse files
fogsong233vgvassilev
authored andcommitted
fix use-after-free
1 parent fb4795d commit bc61768

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

test/CUDA/GradientKernels.cu

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1047,10 +1047,6 @@ int main(void) {
10471047
TEST_2_N(add_kernel_5, dim3(2, 1, 1), dim3(1), 0, false, "in, out", dummy_out, dummy_in, d_out, d_in, 5); // CHECK-EXEC: 5, 5, 5, 5, 5
10481048
TEST_2(add_kernel_6, dim3(1), dim3(5, 1, 1), 0, false, "a, b", dummy_out, dummy_in, d_out, d_in, 5); // CHECK-EXEC: 25, 0, 0, 0, 0
10491049

1050-
cudaFree(dummy_in);
1051-
cudaFree(dummy_out);
1052-
cudaFree(d_out);
1053-
cudaFree(d_in);
10541050

10551051
double *dummy_in_double, *dummy_out_double, *d_out_double, *d_in_double;
10561052
cudaMalloc(&dummy_in_double, 10 * sizeof(double));
@@ -1157,6 +1153,11 @@ int main(void) {
11571153
TEST(injective_reassignment, dim3(1), dim3(1), 0, false, n, d_n, 1); // CHECK-EXEC: 1
11581154
TEST(injective_reassignment_loop, dim3(1), dim3(1), 0, false, n, d_n, 1); // CHECK-EXEC: 1
11591155

1156+
cudaFree(dummy_in);
1157+
cudaFree(dummy_out);
1158+
cudaFree(d_out);
1159+
cudaFree(d_in);
1160+
11601161
cudaFree(n);
11611162
cudaFree(d_n);
11621163
free(res);

0 commit comments

Comments
 (0)