Update interface loaders

0x12CC · 0x12CC · commit 8e7b98d60db9 · 2024-01-24T10:43:51.000-08:00
Signed-off-by: Michael Aziz &lt;michael.aziz@intel.com&gt;
diff --git a/source/adapters/cuda/ur_interface_loader.cpp b/source/adapters/cuda/ur_interface_loader.cpp
@@ -399,7 +399,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetEnqueueExpProcAddrTable(
     return result;
   }
 
-  pDdiTable->pfnCooperativeKernelLaunchExp = nullptr;
+  pDdiTable->pfnCooperativeKernelLaunchExp =
+      urEnqueueCooperativeKernelLaunchExp;
 
   return UR_RESULT_SUCCESS;
 }
@@ -411,7 +412,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
     return result;
   }
 
-  pDdiTable->pfnSuggestMaxCooperativeGroupCountExp = nullptr;
+  pDdiTable->pfnSuggestMaxCooperativeGroupCountExp =
+      urKernelSuggestMaxCooperativeGroupCountExp;
 
   return UR_RESULT_SUCCESS;
 }
diff --git a/source/adapters/hip/ur_interface_loader.cpp b/source/adapters/hip/ur_interface_loader.cpp
@@ -369,7 +369,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetEnqueueExpProcAddrTable(
     return result;
   }
 
-  pDdiTable->pfnCooperativeKernelLaunchExp = nullptr;
+  pDdiTable->pfnCooperativeKernelLaunchExp =
+      urEnqueueCooperativeKernelLaunchExp;
 
   return UR_RESULT_SUCCESS;
 }
@@ -381,7 +382,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
     return result;
   }
 
-  pDdiTable->pfnSuggestMaxCooperativeGroupCountExp = nullptr;
+  pDdiTable->pfnSuggestMaxCooperativeGroupCountExp =
+      urKernelSuggestMaxCooperativeGroupCountExp;
 
   return UR_RESULT_SUCCESS;
 }
diff --git a/source/adapters/level_zero/ur_interface_loader.cpp b/source/adapters/level_zero/ur_interface_loader.cpp
@@ -446,7 +446,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetEnqueueExpProcAddrTable(
     return result;
   }
 
-  pDdiTable->pfnCooperativeKernelLaunchExp = nullptr;
+  pDdiTable->pfnCooperativeKernelLaunchExp =
+      urEnqueueCooperativeKernelLaunchExp;
 
   return UR_RESULT_SUCCESS;
 }
@@ -458,7 +459,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
     return result;
   }
 
-  pDdiTable->pfnSuggestMaxCooperativeGroupCountExp = nullptr;
+  pDdiTable->pfnSuggestMaxCooperativeGroupCountExp =
+      urKernelSuggestMaxCooperativeGroupCountExp;
 
   return UR_RESULT_SUCCESS;
 }
diff --git a/source/adapters/opencl/ur_interface_loader.cpp b/source/adapters/opencl/ur_interface_loader.cpp
@@ -390,7 +390,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetEnqueueExpProcAddrTable(
     return result;
   }
 
-  pDdiTable->pfnCooperativeKernelLaunchExp = nullptr;
+  pDdiTable->pfnCooperativeKernelLaunchExp =
+      urEnqueueCooperativeKernelLaunchExp;
 
   return UR_RESULT_SUCCESS;
 }
@@ -402,7 +403,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
     return result;
   }
 
-  pDdiTable->pfnSuggestMaxCooperativeGroupCountExp = nullptr;
+  pDdiTable->pfnSuggestMaxCooperativeGroupCountExp =
+      urKernelSuggestMaxCooperativeGroupCountExp;
 
   return UR_RESULT_SUCCESS;
 }

Original file line number	Diff line number	Diff line change
`@@ -399,7 +399,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetEnqueueExpProcAddrTable(`
`399`	`399`	`return result;`
`400`	`400`	`}`
`401`	`401`
`402`		`- pDdiTable->pfnCooperativeKernelLaunchExp = nullptr;`
	`402`	`+ pDdiTable->pfnCooperativeKernelLaunchExp =`
	`403`	`+ urEnqueueCooperativeKernelLaunchExp;`
`403`	`404`
`404`	`405`	`return UR_RESULT_SUCCESS;`
`405`	`406`	`}`
`@@ -411,7 +412,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(`
`411`	`412`	`return result;`
`412`	`413`	`}`
`413`	`414`
`414`		`- pDdiTable->pfnSuggestMaxCooperativeGroupCountExp = nullptr;`
	`415`	`+ pDdiTable->pfnSuggestMaxCooperativeGroupCountExp =`
	`416`	`+ urKernelSuggestMaxCooperativeGroupCountExp;`
`415`	`417`
`416`	`418`	`return UR_RESULT_SUCCESS;`
`417`	`419`	`}`
Original file line number	Diff line number	Diff line change
`@@ -369,7 +369,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetEnqueueExpProcAddrTable(`
`369`	`369`	`return result;`
`370`	`370`	`}`
`371`	`371`
`372`		`- pDdiTable->pfnCooperativeKernelLaunchExp = nullptr;`
	`372`	`+ pDdiTable->pfnCooperativeKernelLaunchExp =`
	`373`	`+ urEnqueueCooperativeKernelLaunchExp;`
`373`	`374`
`374`	`375`	`return UR_RESULT_SUCCESS;`
`375`	`376`	`}`
`@@ -381,7 +382,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(`
`381`	`382`	`return result;`
`382`	`383`	`}`
`383`	`384`
`384`		`- pDdiTable->pfnSuggestMaxCooperativeGroupCountExp = nullptr;`
	`385`	`+ pDdiTable->pfnSuggestMaxCooperativeGroupCountExp =`
	`386`	`+ urKernelSuggestMaxCooperativeGroupCountExp;`
`385`	`387`
`386`	`388`	`return UR_RESULT_SUCCESS;`
`387`	`389`	`}`
Original file line number	Diff line number	Diff line change
`@@ -446,7 +446,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetEnqueueExpProcAddrTable(`
`446`	`446`	`return result;`
`447`	`447`	`}`
`448`	`448`
`449`		`- pDdiTable->pfnCooperativeKernelLaunchExp = nullptr;`
	`449`	`+ pDdiTable->pfnCooperativeKernelLaunchExp =`
	`450`	`+ urEnqueueCooperativeKernelLaunchExp;`
`450`	`451`
`451`	`452`	`return UR_RESULT_SUCCESS;`
`452`	`453`	`}`
`@@ -458,7 +459,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(`
`458`	`459`	`return result;`
`459`	`460`	`}`
`460`	`461`
`461`		`- pDdiTable->pfnSuggestMaxCooperativeGroupCountExp = nullptr;`
	`462`	`+ pDdiTable->pfnSuggestMaxCooperativeGroupCountExp =`
	`463`	`+ urKernelSuggestMaxCooperativeGroupCountExp;`
`462`	`464`
`463`	`465`	`return UR_RESULT_SUCCESS;`
`464`	`466`	`}`
Original file line number	Diff line number	Diff line change
`@@ -390,7 +390,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetEnqueueExpProcAddrTable(`
`390`	`390`	`return result;`
`391`	`391`	`}`
`392`	`392`
`393`		`- pDdiTable->pfnCooperativeKernelLaunchExp = nullptr;`
	`393`	`+ pDdiTable->pfnCooperativeKernelLaunchExp =`
	`394`	`+ urEnqueueCooperativeKernelLaunchExp;`
`394`	`395`
`395`	`396`	`return UR_RESULT_SUCCESS;`
`396`	`397`	`}`
`@@ -402,7 +403,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(`
`402`	`403`	`return result;`
`403`	`404`	`}`
`404`	`405`
`405`		`- pDdiTable->pfnSuggestMaxCooperativeGroupCountExp = nullptr;`
	`406`	`+ pDdiTable->pfnSuggestMaxCooperativeGroupCountExp =`
	`407`	`+ urKernelSuggestMaxCooperativeGroupCountExp;`
`406`	`408`
`407`	`409`	`return UR_RESULT_SUCCESS;`
`408`	`410`	`}`