P4 to Git Change 1719466 by wchau@wchau_OCL_boltzmann on 2018/12/12 15:42:59

SWDEV-169078 - Also copy  private_segment_size/group_segment_size to runtime handle for COMgr support

Affected files ...

... //depot/stg/opencl/drivers/opencl/runtime/device/pal/palkernel.cpp#75 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/rocm/rockernel.cpp#46 edit

rocclr/runtime/device/rocm/rockernel.cpp

@@ -85,9 +85,15 @@ bool LightningKernel::init() {
                                                  HSA_EXECUTABLE_SYMBOL_INFO_VARIABLE_ADDRESS,
                                                  &variable_address);
     }
+
     if (hsaStatus == HSA_STATUS_SUCCESS) {
+      const struct RuntimeHandle runtime_handle = {
+        kernelCodeHandle_,
+        workitemPrivateSegmentByteSize(),
+        WorkgroupGroupSegmentByteSize()
+      };
       hsaStatus = hsa_memory_copy(reinterpret_cast<void*>(variable_address),
-                                  &kernelCodeHandle_, variable_size);
+                                  &runtime_handle, variable_size);
     }
 
     if (hsaStatus != HSA_STATUS_SUCCESS) {