Skip to content

Commit dccf3e5

Browse files
Add wgpuBufferRelease for CopyData
1 parent e98c3b2 commit dccf3e5

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

gpu.hpp

+6
Original file line numberDiff line numberDiff line change
@@ -1065,6 +1065,9 @@ inline void toCPU(Context &ctx, Tensor &tensor, void *data, size_t bufferSize) {
10651065
check(op.commandBuffer, "Create command buffer", __FILE__, __LINE__);
10661066
}
10671067
toCPU(ctx, tensor, data, bufferSize, op);
1068+
if (op.readbackBuffer) {
1069+
wgpuBufferRelease(op.readbackBuffer);
1070+
}
10681071
}
10691072

10701073
/**
@@ -1131,6 +1134,9 @@ inline void toCPU(Context &ctx, WGPUBuffer buffer, void *data,
11311134
},
11321135
&callbackData);
11331136
wait(ctx, op.future);
1137+
if (op.readbackBuffer) {
1138+
wgpuBufferRelease(op.readbackBuffer);
1139+
}
11341140
}
11351141

11361142

0 commit comments

Comments
 (0)