#define GT212_DMA                                                    0x000085b5
 #define FERMI_DMA                                                    0x000090b5
+#define KEPLER_DMA_COPY_A                                            0x0000a0b5
+#define MAXWELL_DMA_COPY_A                                           0x0000b0b5
 
 #define FERMI_DECOMPRESS                                             0x000090b8
 
 
  * Authors: Ben Skeggs
  */
 #include <engine/ce.h>
+#include <engine/fifo.h>
 
-#include <core/engctx.h>
-
-/*******************************************************************************
- * Copy object classes
- ******************************************************************************/
-
-static struct nvkm_oclass
-gk104_ce_sclass[] = {
-       { 0xa0b5, &nvkm_object_ofuncs },
-       {},
-};
-
-/*******************************************************************************
- * PCE context
- ******************************************************************************/
-
-static struct nvkm_ofuncs
-gk104_ce_context_ofuncs = {
-       .ctor = _nvkm_engctx_ctor,
-       .dtor = _nvkm_engctx_dtor,
-       .init = _nvkm_engctx_init,
-       .fini = _nvkm_engctx_fini,
-       .rd32 = _nvkm_engctx_rd32,
-       .wr32 = _nvkm_engctx_wr32,
-};
-
-static struct nvkm_oclass
-gk104_ce_cclass = {
-       .handle = NV_ENGCTX(CE0, 0xc0),
-       .ofuncs = &gk104_ce_context_ofuncs,
-};
-
-/*******************************************************************************
- * PCE engine/subdev functions
- ******************************************************************************/
+#include <nvif/class.h>
 
 static void
 gk104_ce_intr(struct nvkm_subdev *subdev)
        }
 }
 
+static const struct nvkm_engine_func
+gk104_ce = {
+       .sclass = {
+               { -1, -1, KEPLER_DMA_COPY_A },
+               {}
+       }
+};
+
 static int
 gk104_ce0_ctor(struct nvkm_object *parent, struct nvkm_object *engine,
               struct nvkm_oclass *oclass, void *data, u32 size,
        if (ret)
                return ret;
 
+       ce->func = &gk104_ce;
        nv_subdev(ce)->unit = 0x00000040;
        nv_subdev(ce)->intr = gk104_ce_intr;
-       nv_engine(ce)->cclass = &gk104_ce_cclass;
-       nv_engine(ce)->sclass = gk104_ce_sclass;
        return 0;
 }
 
        if (ret)
                return ret;
 
+       ce->func = &gk104_ce;
        nv_subdev(ce)->unit = 0x00000080;
        nv_subdev(ce)->intr = gk104_ce_intr;
-       nv_engine(ce)->cclass = &gk104_ce_cclass;
-       nv_engine(ce)->sclass = gk104_ce_sclass;
        return 0;
 }
 
        if (ret)
                return ret;
 
+       ce->func = &gk104_ce;
        nv_subdev(ce)->unit = 0x00200000;
        nv_subdev(ce)->intr = gk104_ce_intr;
-       nv_engine(ce)->cclass = &gk104_ce_cclass;
-       nv_engine(ce)->sclass = gk104_ce_sclass;
        return 0;
 }
 
 
  * Authors: Ben Skeggs
  */
 #include <engine/ce.h>
+#include <engine/fifo.h>
 
-#include <core/engctx.h>
-
-/*******************************************************************************
- * Copy object classes
- ******************************************************************************/
-
-static struct nvkm_oclass
-gm204_ce_sclass[] = {
-       { 0xb0b5, &nvkm_object_ofuncs },
-       {},
-};
-
-/*******************************************************************************
- * PCE context
- ******************************************************************************/
-
-static struct nvkm_ofuncs
-gm204_ce_context_ofuncs = {
-       .ctor = _nvkm_engctx_ctor,
-       .dtor = _nvkm_engctx_dtor,
-       .init = _nvkm_engctx_init,
-       .fini = _nvkm_engctx_fini,
-       .rd32 = _nvkm_engctx_rd32,
-       .wr32 = _nvkm_engctx_wr32,
-};
-
-static struct nvkm_oclass
-gm204_ce_cclass = {
-       .handle = NV_ENGCTX(CE0, 0x24),
-       .ofuncs = &gm204_ce_context_ofuncs,
-};
-
-/*******************************************************************************
- * PCE engine/subdev functions
- ******************************************************************************/
+#include <nvif/class.h>
 
 static void
 gm204_ce_intr(struct nvkm_subdev *subdev)
        }
 }
 
+static const struct nvkm_engine_func
+gm204_ce = {
+       .sclass = {
+               { -1, -1, MAXWELL_DMA_COPY_A },
+               {}
+       }
+};
+
 static int
 gm204_ce0_ctor(struct nvkm_object *parent, struct nvkm_object *engine,
               struct nvkm_oclass *oclass, void *data, u32 size,
        if (ret)
                return ret;
 
+       ce->func = &gm204_ce;
        nv_subdev(ce)->unit = 0x00000040;
        nv_subdev(ce)->intr = gm204_ce_intr;
-       nv_engine(ce)->cclass = &gm204_ce_cclass;
-       nv_engine(ce)->sclass = gm204_ce_sclass;
        return 0;
 }
 
        if (ret)
                return ret;
 
+       ce->func = &gm204_ce;
        nv_subdev(ce)->unit = 0x00000080;
        nv_subdev(ce)->intr = gm204_ce_intr;
-       nv_engine(ce)->cclass = &gm204_ce_cclass;
-       nv_engine(ce)->sclass = gm204_ce_sclass;
        return 0;
 }
 
        if (ret)
                return ret;
 
+       ce->func = &gm204_ce;
        nv_subdev(ce)->unit = 0x00200000;
        nv_subdev(ce)->intr = gm204_ce_intr;
-       nv_engine(ce)->cclass = &gm204_ce_cclass;
-       nv_engine(ce)->sclass = gm204_ce_sclass;
        return 0;
 }