Only get inBlockBuffer from data channel if we are not doing our own …

…reading
ubarsc · Feb 22, 2024 · 1582b63 · 1582b63
1 parent 7ab9509
commit 1582b63
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/rios/cmdline/rios_computeworker.py b/rios/cmdline/rios_computeworker.py
@@ -66,14 +66,17 @@ def riosComputeWorker(workerID, dataChan):
     controls = dataChan.workerCommonData.get('controls', None)
     allInfo = dataChan.workerCommonData.get('allInfo', None)
     workinggrid = dataChan.workerCommonData.get('workinggrid', None)
-    inBlockBuffer = dataChan.inBlockBuffer
     outBlockBuffer = dataChan.outBlockBuffer
+    if not controls.concurrency.computeWorkersRead:
+        inBlockBuffer = dataChan.inBlockBuffer
+    else:
+        inBlockBuffer = None
 
     blockList = dataChan.workerLocalData.get(workerID, None)
 
     rtn = applier.apply_singleCompute(userFunction, infiles, outfiles,
         otherArgs, controls, allInfo, workinggrid, blockList, outBlockBuffer,
-        inBlockBuffer)
+        inBlockBuffer, workerID)
 
     # Make a pickleable version of the timings
     timings = Timers(pairs=rtn.timings.pairs, withlock=False)