File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -45,7 +45,8 @@ CUstream ur_queue_handle_t_::getNextComputeStream(uint32_t *StreamToken) {
45
45
// change NumComputeStreams after that
46
46
if (NumComputeStreams < ComputeStreams.size ()) {
47
47
UR_CHECK_ERROR (cuStreamCreateWithPriority (
48
- &ComputeStreams[NumComputeStreams++], Flags, Priority));
48
+ &ComputeStreams[NumComputeStreams], Flags, Priority));
49
+ ++NumComputeStreams;
49
50
}
50
51
}
51
52
Token = ComputeStreamIndex++;
@@ -110,7 +111,8 @@ CUstream ur_queue_handle_t_::getNextTransferStream() {
110
111
// change NumTransferStreams after that
111
112
if (NumTransferStreams < TransferStreams.size ()) {
112
113
UR_CHECK_ERROR (cuStreamCreateWithPriority (
113
- &TransferStreams[NumTransferStreams++], Flags, Priority));
114
+ &TransferStreams[NumTransferStreams], Flags, Priority));
115
+ ++NumTransferStreams;
114
116
}
115
117
}
116
118
uint32_t StreamI = TransferStreamIndex++ % TransferStreams.size ();
You can’t perform that action at this time.
0 commit comments