/* Fill in IOLINK subtype.
         * TODO: Fill-in other fields of iolink subtype
         */
-       if (kdev->adev->gmc.xgmi.connected_to_cpu) {
+       if (kdev->adev->gmc.xgmi.connected_to_cpu ||
+           (KFD_GC_VERSION(kdev) == IP_VERSION(9, 4, 3) &&
+            kdev->adev->smuio.funcs->get_pkg_type(kdev->adev) ==
+            AMDGPU_PKG_TYPE_APU)) {
                bool ext_cpu = KFD_GC_VERSION(kdev) != IP_VERSION(9, 4, 3);
                int mem_bw = 819200, weight = ext_cpu ? KFD_CRAT_XGMI_WEIGHT :
                                                        KFD_CRAT_INTRA_SOCKET_WEIGHT;
 
                        continue;
 
                /* Include the CPU peer in GPU hive if connected over xGMI. */
-               if (!peer_dev->gpu && !peer_dev->node_props.hive_id &&
-                               dev->node_props.hive_id &&
-                               dev->gpu->adev->gmc.xgmi.connected_to_cpu)
+               if (!peer_dev->gpu &&
+                   link->iolink_type == CRAT_IOLINK_TYPE_XGMI)
                        peer_dev->node_props.hive_id = dev->node_props.hive_id;
 
                list_for_each_entry(inbound_link, &peer_dev->io_link_props,