Searched refs:num_sub_streams (Results 1 – 1 of 1) sorted by relevance
1187 …const auto num_sub_streams = nv > batchsize ? std::min((nv + batchsize) / batchsize, batchsize) … in MDot_() local1213 PetscCall(PetscDeviceContextForkWithStreamType(dctx, stype, num_sub_streams, &sub)); in MDot_()1216 if (num_sub_streams) { in MDot_()1217 cur_ctx = sub[subidx++ % num_sub_streams]; in MDot_()1250 … PetscCall(PetscDeviceContextJoin(dctx, num_sub_streams, PETSC_DEVICE_CONTEXT_JOIN_DESTROY, &sub)); in MDot_()