The version number shall increase in the GPU data copy that is access…

…ed in WRITE mode, even if it is already on the GPU.
ICLDisco · Mar 12, 2024 · 41e8e9b · 41e8e9b
1 parent fbfa824
commit 41e8e9b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/parsec/mca/device/device_gpu.c b/parsec/mca/device/device_gpu.c
@@ -1331,6 +1331,10 @@ parsec_device_data_stage_in( parsec_device_gpu_module_t* gpu_device,
 
         parsec_data_end_transfer_ownership_to_copy(original, gpu_device->super.device_index, (uint8_t)type);
 
+        if( (PARSEC_FLOW_ACCESS_WRITE & type) && (gpu_task->task_type != PARSEC_GPU_TASK_TYPE_PREFETCH) ) {
+            gpu_elem->version = candidate->version + 1;
+		}
+
         PARSEC_DEBUG_VERBOSE(10, parsec_gpu_output_stream,
                              "GPU[%s]:\t\tNO Move for data copy %p [ref_count %d, key %x]",
                              gpu_device->super.name,