--- /dev/null
+/*
+ * Copyright 2018 Red Hat Inc.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
+ * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ */
+#include <nvif/fifo.h>
+
+static int
+nvif_fifo_runlists(struct nvif_device *device)
+{
+       struct nvif_object *object = &device->object;
+       struct {
+               struct nv_device_info_v1 m;
+               struct {
+                       struct nv_device_info_v1_data runlists;
+                       struct nv_device_info_v1_data runlist[64];
+               } v;
+       } *a;
+       int ret, i;
+
+       if (device->runlist)
+               return 0;
+
+       if (!(a = kmalloc(sizeof(*a), GFP_KERNEL)))
+               return -ENOMEM;
+       a->m.version = 1;
+       a->m.count = sizeof(a->v) / sizeof(a->v.runlists);
+       a->v.runlists.mthd = NV_DEVICE_FIFO_RUNLISTS;
+       for (i = 0; i < ARRAY_SIZE(a->v.runlist); i++)
+               a->v.runlist[i].mthd = NV_DEVICE_FIFO_RUNLIST_ENGINES(i);
+
+       ret = nvif_object_mthd(object, NV_DEVICE_V0_INFO, a, sizeof(*a));
+       if (ret)
+               goto done;
+
+       device->runlists = fls64(a->v.runlists.data);
+       device->runlist = kzalloc(sizeof(*device->runlist) *
+                                 device->runlists, GFP_KERNEL);
+       if (!device->runlist) {
+               ret = -ENOMEM;
+               goto done;
+       }
+
+       for (i = 0; i < device->runlists; i++) {
+               if (a->v.runlists.data & BIT_ULL(i))
+                       device->runlist[i].engines = a->v.runlist[i].data;
+       }
+
+done:
+       kfree(a);
+       return ret;
+}
+
+u64
+nvif_fifo_runlist(struct nvif_device *device, u64 engine)
+{
+       struct nvif_object *object = &device->object;
+       struct {
+               struct nv_device_info_v1 m;
+               struct {
+                       struct nv_device_info_v1_data engine;
+               } v;
+       } a = {
+               .m.version = 1,
+               .m.count = sizeof(a.v) / sizeof(a.v.engine),
+               .v.engine.mthd = engine,
+       };
+       u64 runm = 0;
+       int ret, i;
+
+       if ((ret = nvif_fifo_runlists(device)))
+               return runm;
+
+       ret = nvif_object_mthd(object, NV_DEVICE_V0_INFO, &a, sizeof(a));
+       if (ret == 0) {
+               for (i = 0; i < device->runlists; i++) {
+                       if (device->runlist[i].engines & a.v.engine.data)
+                               runm |= BIT_ULL(i);
+               }
+       }
+
+       return runm;
+}
 
 #include <engine/sw.h>
 
 #include <nvif/class.h>
+#include <nvif/cl0080.h>
 
 struct gk104_fifo_engine_status {
        bool busy;
        nvkm_mask(device, 0x002140, 0x10000000, 0x10000000);
 }
 
+static int
+gk104_fifo_info(struct nvkm_fifo *base, u64 mthd, u64 *data)
+{
+       struct gk104_fifo *fifo = gk104_fifo(base);
+       switch (mthd) {
+       case NV_DEVICE_FIFO_RUNLISTS:
+               *data = (1ULL << fifo->runlist_nr) - 1;
+               return 0;
+       case NV_DEVICE_FIFO_RUNLIST_ENGINES(0)...
+            NV_DEVICE_FIFO_RUNLIST_ENGINES(63): {
+               int runl = mthd - NV_DEVICE_FIFO_RUNLIST_ENGINES(0), engn;
+               if (runl < fifo->runlist_nr) {
+                       unsigned long engm = fifo->runlist[runl].engm;
+                       struct nvkm_engine *engine;
+                       *data = 0;
+                       for_each_set_bit(engn, &engm, fifo->engine_nr) {
+                               if ((engine = fifo->engine[engn].engine))
+                                       *data |= BIT_ULL(engine->subdev.index);
+                       }
+                       return 0;
+               }
+       }
+               return -EINVAL;
+       default:
+               return -EINVAL;
+       }
+}
+
 static int
 gk104_fifo_oneinit(struct nvkm_fifo *base)
 {
 gk104_fifo_ = {
        .dtor = gk104_fifo_dtor,
        .oneinit = gk104_fifo_oneinit,
+       .info = gk104_fifo_info,
        .init = gk104_fifo_init,
        .fini = gk104_fifo_fini,
        .intr = gk104_fifo_intr,