Merge commit '6d20adb7043719912b7ac51e8b3e90722c6140c0'

soedinglab · Dec 2, 2024 · 59016d2 · 59016d2
2 parents 9c102e1 + 6d20adb
commit 59016d2
Showing 1 changed file with 23 additions and 21 deletions.
diff --git a/lib/libmarv/src/cudasw4.cuh b/lib/libmarv/src/cudasw4.cuh
@@ -2884,27 +2884,29 @@ namespace cudasw4{
                 cudaSetDevice(deviceIds[gpu]); CUERR;
                 auto& ws = *workingSets[gpu];
 
-                if(!batchPlansDstInfoVec[gpu][0].isUploaded){
-                    //all batches for cached db are now resident in gpu memory. update the flags
-                    if(ws.getNumBatchesInCachedDB() > 0){
-                        markCachedDBBatchesAsUploaded(gpu);
-
-                        // current offsets in cached db store the offsets for each batch, i.e. for each batch the offsets will start again at 0
-                        // compute prefix sum to obtain the single-batch offsets
-
-                        cudaMemsetAsync(ws.d_cacheddb->getOffsetData(), 0, sizeof(size_t), ws.workStreamForTempUsage); CUERR;
-
-                        auto d_paddedLengths = thrust::make_transform_iterator(
-                            ws.d_cacheddb->getLengthData(),
-                            RoundToNextMultiple<size_t, 4>{}
-                        );
-
-                        thrust::inclusive_scan(
-                            thrust::cuda::par_nosync(thrust_async_allocator<char>(ws.workStreamForTempUsage)).on(ws.workStreamForTempUsage),
-                            d_paddedLengths,
-                            d_paddedLengths + ws.getNumSequencesInCachedDB(),
-                            ws.d_cacheddb->getOffsetData() + 1
-                        );
+                if(batchPlansDstInfoVec[gpu].size() > 0){
+                    if(!batchPlansDstInfoVec[gpu][0].isUploaded){
+                        //all batches for cached db are now resident in gpu memory. update the flags
+                        if(ws.getNumBatchesInCachedDB() > 0){
+                            markCachedDBBatchesAsUploaded(gpu);
+
+                            // current offsets in cached db store the offsets for each batch, i.e. for each batch the offsets will start again at 0
+                            // compute prefix sum to obtain the single-batch offsets
+
+                            cudaMemsetAsync(ws.d_cacheddb->getOffsetData(), 0, sizeof(size_t), ws.workStreamForTempUsage); CUERR;
+
+                            auto d_paddedLengths = thrust::make_transform_iterator(
+                                ws.d_cacheddb->getLengthData(),
+                                RoundToNextMultiple<size_t, 4>{}
+                            );
+
+                            thrust::inclusive_scan(
+                                thrust::cuda::par_nosync(thrust_async_allocator<char>(ws.workStreamForTempUsage)).on(ws.workStreamForTempUsage),
+                                d_paddedLengths,
+                                d_paddedLengths + ws.getNumSequencesInCachedDB(),
+                                ws.d_cacheddb->getOffsetData() + 1
+                            );
+                        }
                     }
                 }
             }