Quelle  engine.c   Sprache: C

/* SPDX-License-Identifier: MIT
 *
 * Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
 */
#include "engine.h"
#include "gpu.h"

#include <core/object.h>
#include <engine/fifo/chan.h>

struct nvkm_rm_engine {
 struct nvkm_engine engine;

 struct nvkm_engine_func func;
};

struct nvkm_rm_engine_obj {
 struct nvkm_object object;
 struct nvkm_gsp_object rm;
};

static void*
nvkm_rm_engine_obj_dtor(struct nvkm_object *object)
{
 struct nvkm_rm_engine_obj *obj = container_of(object, typeof(*obj), object);

 nvkm_gsp_rm_free(&obj->rm);
 return obj;
}

static const struct nvkm_object_func
nvkm_rm_engine_obj = {
 .dtor = nvkm_rm_engine_obj_dtor,
};

int
nvkm_rm_engine_obj_new(struct nvkm_gsp_object *chan, int chid, const struct nvkm_oclass *oclass,
         struct nvkm_object **pobject)
{
 struct nvkm_rm *rm = chan->client->gsp->rm;
 const int inst = oclass->engine->subdev.inst;
 const u32 class = oclass->base.oclass;
 const u32 handle = oclass->handle;
 struct nvkm_rm_engine_obj *obj;
 int ret;

 obj = kzalloc(sizeof(*obj), GFP_KERNEL);
 if (!obj)
  return -ENOMEM;

 switch (oclass->engine->subdev.type) {
 case NVKM_ENGINE_CE:
  ret = rm->api->ce->alloc(chan, handle, class, inst, &obj->rm);
  break;
 case NVKM_ENGINE_GR:
  ret = nvkm_gsp_rm_alloc(chan, handle, class, 0, &obj->rm);
  break;
 case NVKM_ENGINE_NVDEC:
  ret = rm->api->nvdec->alloc(chan, handle, class, inst, &obj->rm);
  break;
 case NVKM_ENGINE_NVENC:
  ret = rm->api->nvenc->alloc(chan, handle, class, inst, &obj->rm);
  break;
 case NVKM_ENGINE_NVJPG:
  ret = rm->api->nvjpg->alloc(chan, handle, class, inst, &obj->rm);
  break;
 case NVKM_ENGINE_OFA:
  ret = rm->api->ofa->alloc(chan, handle, class, inst, &obj->rm);
  break;
 default:
  ret = -EINVAL;
  WARN_ON(1);
  break;
 }

 if (ret) {
  kfree(obj);
  return ret;
 }

 nvkm_object_ctor(&nvkm_rm_engine_obj, oclass, &obj->object);
 *pobject = &obj->object;
 return 0;
}

static int
nvkm_rm_engine_obj_ctor(const struct nvkm_oclass *oclass, void *argv, u32 argc,
   struct nvkm_object **pobject)
{
 struct nvkm_chan *chan = nvkm_uchan_chan(oclass->parent);

 return nvkm_rm_engine_obj_new(&chan->rm.object, chan->id, oclass, pobject);
}

static void *
nvkm_rm_engine_dtor(struct nvkm_engine *engine)
{
 kfree(engine->func);
 return engine;
}

int
nvkm_rm_engine_ctor(void *(*dtor)(struct nvkm_engine *), struct nvkm_rm *rm,
      enum nvkm_subdev_type type, int inst,
      const u32 *class, int nclass, struct nvkm_engine *engine)
{
 struct nvkm_engine_func *func;

 func = kzalloc(struct_size(func, sclass, nclass + 1), GFP_KERNEL);
 if (!func)
  return -ENOMEM;

 func->dtor = dtor;

 for (int i = 0; i < nclass; i++) {
  func->sclass[i].oclass = class[i];
  func->sclass[i].minver = -1;
  func->sclass[i].maxver = 0;
  func->sclass[i].ctor = nvkm_rm_engine_obj_ctor;
 }

 nvkm_engine_ctor(func, rm->device, type, inst, true, engine);
 return 0;
}

static int
nvkm_rm_engine_new_(struct nvkm_rm *rm, enum nvkm_subdev_type type, int inst, u32 class,
      struct nvkm_engine **pengine)
{
 struct nvkm_engine *engine;
 int ret;

 engine = kzalloc(sizeof(*engine), GFP_KERNEL);
 if (!engine)
  return -ENOMEM;

 ret = nvkm_rm_engine_ctor(nvkm_rm_engine_dtor, rm, type, inst, &class, 1, engine);
 if (ret) {
  kfree(engine);
  return ret;
 }

 *pengine = engine;
 return 0;
}

int
nvkm_rm_engine_new(struct nvkm_rm *rm, enum nvkm_subdev_type type, int inst)
{
 const struct nvkm_rm_gpu *gpu = rm->gpu;
 struct nvkm_device *device = rm->device;

 switch (type) {
 case NVKM_ENGINE_CE:
  if (WARN_ON(inst >= ARRAY_SIZE(device->ce)))
   return -EINVAL;

  return nvkm_rm_engine_new_(rm, type, inst, gpu->ce.class, &device->ce[inst]);
 case NVKM_ENGINE_GR:
  if (inst != 0)
   return -ENODEV; /* MiG not supported, just ignore. */

  return nvkm_rm_gr_new(rm);
 case NVKM_ENGINE_NVDEC:
  if (WARN_ON(inst >= ARRAY_SIZE(device->nvdec)))
   return -EINVAL;

  return nvkm_rm_nvdec_new(rm, inst);
 case NVKM_ENGINE_NVENC:
  if (WARN_ON(inst >= ARRAY_SIZE(device->nvenc)))
   return -EINVAL;

  return nvkm_rm_nvenc_new(rm, inst);
 case NVKM_ENGINE_NVJPG:
  if (WARN_ON(inst >= ARRAY_SIZE(device->nvjpg)))
   return -EINVAL;

  return nvkm_rm_engine_new_(rm, type, inst, gpu->nvjpg.class, &device->nvjpg[inst]);
 case NVKM_ENGINE_OFA:
  if (WARN_ON(inst >= ARRAY_SIZE(device->ofa)))
   return -EINVAL;

  return nvkm_rm_engine_new_(rm, type, inst, gpu->ofa.class, &device->ofa[inst]);
 default:
  break;
 }

 return -ENODEV;
}