Also fixes the values for Turing, even though we don't use it yet.
Signed-off-by: Ben Skeggs <bskeggs@redhat.com>
        int (*enable)(struct nvkm_falcon *falcon);
        void (*disable)(struct nvkm_falcon *falcon);
 
+       struct {
+               u32 head;
+               u32 tail;
+               u32 stride;
+       } cmdq, msgq;
+
        struct nvkm_sclass sclass[];
 };
 
 
        .start = nvkm_falcon_v1_start,
        .enable = nvkm_falcon_v1_enable,
        .disable = nvkm_falcon_v1_disable,
+       .cmdq = { 0xa00, 0xa04, 8 },
+       .msgq = { 0xa30, 0xa34, 8 },
 };
 
 const struct nvkm_sec2_func
 
        .start = nvkm_falcon_v1_start,
        .enable = nvkm_falcon_v1_enable,
        .disable = nvkm_falcon_v1_disable,
+       .cmdq = { 0xc00, 0xc04, 8 },
+       .msgq = { 0xc80, 0xc84, 8 },
 };
 
 static const struct nvkm_sec2_func
 
 {
        struct nvkm_falcon *falcon = priv->falcon;
        const struct nvkm_subdev *subdev = falcon->owner;
+       const u32 tail_reg = falcon->func->msgq.tail;
        u32 tail;
-       u32 tail_reg;
        int ret;
 
-       /*
-        * Of course the message queue registers vary depending on the falcon
-        * used...
-        */
-       switch (falcon->owner->index) {
-       case NVKM_SUBDEV_PMU:
-               tail_reg = 0x4cc;
-               break;
-       case NVKM_ENGINE_SEC2:
-               tail_reg = 0xa34;
-               break;
-       default:
-               nvkm_error(subdev, "falcon %s unsupported for msgqueue!\n",
-                          nvkm_subdev_name[falcon->owner->index]);
-               return -EINVAL;
-       }
-
        /*
         * Read the message - queues are not initialized yet so we cannot rely
         * on msg_queue_read()
 
                u16 sw_managed_area_offset;
                u16 sw_managed_area_size;
        } *init = (void *)hdr;
+       const struct nvkm_falcon_func *func = _queue->falcon->func;
        const struct nvkm_subdev *subdev = _queue->falcon->owner;
        int i;
 
                queue->size = init->queue_info[i].size;
 
                if (i != MSGQUEUE_0137C63D_MESSAGE_QUEUE) {
-                       queue->head_reg = 0x4a0 + (queue->index * 4);
-                       queue->tail_reg = 0x4b0 + (queue->index * 4);
+                       queue->head_reg = func->cmdq.head + queue->index *
+                                         func->cmdq.stride;
+                       queue->tail_reg = func->cmdq.tail + queue->index *
+                                         func->cmdq.stride;
                } else {
-                       queue->head_reg = 0x4c8;
-                       queue->tail_reg = 0x4cc;
+                       queue->head_reg = func->msgq.head;
+                       queue->tail_reg = func->msgq.tail;
                }
 
                nvkm_debug(subdev,
 
                u16 sw_managed_area_offset;
                u16 sw_managed_area_size;
        } *init = (void *)hdr;
+       const struct nvkm_falcon_func *func = _queue->falcon->func;
        const struct nvkm_subdev *subdev = _queue->falcon->owner;
        int i;
 
                queue->size = init->queue_info[i].size;
 
                if (id == MSGQUEUE_0148CDEC_MESSAGE_QUEUE) {
-                       queue->head_reg = 0xa30 + (queue->index * 8);
-                       queue->tail_reg = 0xa34 + (queue->index * 8);
+                       queue->head_reg = func->msgq.head + queue->index *
+                                         func->msgq.stride;
+                       queue->tail_reg = func->msgq.tail + queue->index *
+                                         func->msgq.stride;
                } else {
-                       queue->head_reg = 0xa00 + (queue->index * 8);
-                       queue->tail_reg = 0xa04 + (queue->index * 8);
+                       queue->head_reg = func->cmdq.head + queue->index *
+                                         func->cmdq.stride;
+                       queue->tail_reg = func->cmdq.tail + queue->index *
+                                         func->cmdq.stride;
                }
 
                nvkm_debug(subdev,
 
        .start = nvkm_falcon_v1_start,
        .enable = nvkm_falcon_v1_enable,
        .disable = nvkm_falcon_v1_disable,
+       .cmdq = { 0x4a0, 0x4b0, 4 },
+       .msgq = { 0x4c8, 0x4cc, 0 },
 };
 
 static const struct nvkm_pmu_func