From 5fe080fd6742c6ca6930c4e9a915d1ac4600c15b Mon Sep 17 00:00:00 2001
From: "Arandjelovic, Marko" <Marko.Arandjelovic@amd.com>
Date: Wed, 16 Apr 2025 10:35:48 +0200
Subject: [PATCH] SWDEV-523137 - function ptrs should match across all devices
 (#171)

---
 hipamd/src/hip_module.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/hipamd/src/hip_module.cpp b/hipamd/src/hip_module.cpp
index ebc1bd09a0..b773f81dfc 100644
--- a/hipamd/src/hip_module.cpp
+++ b/hipamd/src/hip_module.cpp
@@ -877,6 +877,11 @@ hipError_t ihipLaunchCooperativeKernelMultiDevice(hipLaunchParams* launchParamsL
       return hipErrorInvalidDeviceFunction;
     }
 
+    // functions should match across all devices
+    if (i > 0 && launch.func != launchParamsList[i - 1].func) {
+      return hipErrorInvalidValue;
+    }
+
     functionLaunchParamsList[i].function = func;
     functionLaunchParamsList[i].gridDimX = launch.gridDim.x;
     functionLaunchParamsList[i].gridDimY = launch.gridDim.y;