kernel: Booter-Unload success + SM order table (unload/reload cycle works)

- gsp_emu_poll_boot generic SEC2 kick path: clear SEC2_MAILBOX0=0
  (success) - Booter Unload during GPU shutdown read a stale 0xff,
  failing the unload and killing the subsequent re-init ('Booter failed
  with non-zero error code: 0xff' -> RmInitAdapter 0x40). A SEC2 kick
  after a completed boot is a GSP shutdown/reload: clear boot state.
  Driver unload/reload cycle (last close -> RmShutdown, next open ->
  RmInitAdapter) now works.
- STATIC_KGR_GET_GLOBAL_SM_ORDER(0x20800a22): TU104 topology 6 GPC x
  4 TPC x 2 SM = 48 SMs/24 TPCs on GR0. kgrStaticInit is init-critical
  (NV_CHECK_OK_OR_GOTO); zeros left numSm=0 and NVML spun on empty
  topology.

1 files changed, 54 insertions, 0 deletions

diff --git a/kernel/gsp_rm_emulation.c b/kernel/gsp_rm_emulation.c
index 665f073..8f8f9c2 100644
--- a/kernel/gsp_rm_emulation.c
+++ b/kernel/gsp_rm_emulation.c
@@ -1009,6 +1009,54 @@ static void gsp_rm_handle_ce_get_fault_method_buffer_size(
 	nv_vgpu_info(ctx->vgpu, "GSPRM: CE_GET_FAULT_METHOD_BUFFER_SIZE -> %u", val32);
 }
 
+/*
+ * STATIC_KGR_GET_GLOBAL_SM_ORDER (0x20800a22)
+ *
+ * kgrStaticInit copies globalSmOrder[grIdx] into the GR engine's static
+ * info - init-critical (NV_CHECK_OK_OR_GOTO, cannot fail). Zeros leave
+ * numSm=0 and the driver/nvidia-smi spin on empty topology. Report the
+ * TU104 layout: 6 GPCs x 4 TPCs x 2 SMs = 48 SMs on GR0, engines 1-7 empty.
+ * params: globalSmOrder[8] x (globalSmId[240]x12B + numSm(2) + numTpc(2)).
+ */
+static void gsp_rm_handle_kgr_get_global_sm_order(struct gsp_emu_context *ctx,
+						  u32 sequence,
+						  u32 *out_status,
+						  u8 *out_params,
+						  u32 *out_params_size)
+{
+	struct tu104_gr_global_sm_order *order =
+		(struct tu104_gr_global_sm_order *)out_params; /* engine 0 */
+	u32 idx = 0;
+	u32 gpc, tpc, sm;
+
+	memset(out_params, 0, TU104_GR_MAX_ENGINES *
+	       sizeof(struct tu104_gr_global_sm_order));
+
+	for (gpc = 0; gpc < TU104_NUM_GPC; gpc++) {
+		for (tpc = 0; tpc < TU104_TPC_PER_GPC; tpc++) {
+			for (sm = 0; sm < TU104_SM_PER_TPC; sm++) {
+				struct tu104_gr_sm_id *e = &order->globalSmId[idx++];
+
+				e->gpcId = gpc;
+				e->localTpcId = tpc;
+				e->localSmId = sm;
+				e->globalTpcId = gpc * TU104_TPC_PER_GPC + tpc;
+				e->virtualGpcId = gpc;
+				e->migratableTpcId = e->globalTpcId;
+			}
+		}
+	}
+	order->numSm = TU104_NUM_SM;
+	order->numTpc = TU104_NUM_TPC;
+
+	*out_params_size = TU104_GR_MAX_ENGINES *
+			   sizeof(struct tu104_gr_global_sm_order);
+	*out_status = 0;
+
+	nv_vgpu_info(ctx->vgpu, "GSPRM: KGR_GET_GLOBAL_SM_ORDER -> %u SMs, %u TPCs on GR0",
+		     TU104_NUM_SM, TU104_NUM_TPC);
+}
+
 static void gsp_rm_handle_rm_control(struct gsp_emu_context *ctx,
 				     const u8 *elem, u32 sequence)
 {
@@ -1097,6 +1145,12 @@ static void gsp_rm_handle_rm_control(struct gsp_emu_context *ctx,
 			reply + GSP_RM_CTRL_HDR_SIZE, &out_params_size);
 		break;
 
+	case TU104_CTRL_STATIC_KGR_GET_GLOBAL_SM_ORDER:
+		gsp_rm_handle_kgr_get_global_sm_order(ctx, sequence,
+			&out_status,
+			reply + GSP_RM_CTRL_HDR_SIZE, &out_params_size);
+		break;
+
 	default:
 		/* Unknown control command - return success with zeroed params.
 		 * Do NOT echo the request: pure queries pass uninitialized