162306a36Sopenharmony_ci/*
262306a36Sopenharmony_ci * Copyright 2018 Red Hat Inc.
362306a36Sopenharmony_ci *
462306a36Sopenharmony_ci * Permission is hereby granted, free of charge, to any person obtaining a
562306a36Sopenharmony_ci * copy of this software and associated documentation files (the "Software"),
662306a36Sopenharmony_ci * to deal in the Software without restriction, including without limitation
762306a36Sopenharmony_ci * the rights to use, copy, modify, merge, publish, distribute, sublicense,
862306a36Sopenharmony_ci * and/or sell copies of the Software, and to permit persons to whom the
962306a36Sopenharmony_ci * Software is furnished to do so, subject to the following conditions:
1062306a36Sopenharmony_ci *
1162306a36Sopenharmony_ci * The above copyright notice and this permission notice shall be included in
1262306a36Sopenharmony_ci * all copies or substantial portions of the Software.
1362306a36Sopenharmony_ci *
1462306a36Sopenharmony_ci * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
1562306a36Sopenharmony_ci * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
1662306a36Sopenharmony_ci * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
1762306a36Sopenharmony_ci * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
1862306a36Sopenharmony_ci * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
1962306a36Sopenharmony_ci * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
2062306a36Sopenharmony_ci * OTHER DEALINGS IN THE SOFTWARE.
2162306a36Sopenharmony_ci */
2262306a36Sopenharmony_ci#include <nvif/fifo.h>
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_cistatic int
2562306a36Sopenharmony_cinvif_fifo_runlists(struct nvif_device *device)
2662306a36Sopenharmony_ci{
2762306a36Sopenharmony_ci	struct nvif_object *object = &device->object;
2862306a36Sopenharmony_ci	struct {
2962306a36Sopenharmony_ci		struct nv_device_info_v1 m;
3062306a36Sopenharmony_ci		struct {
3162306a36Sopenharmony_ci			struct nv_device_info_v1_data runlists;
3262306a36Sopenharmony_ci			struct nv_device_info_v1_data runlist[64];
3362306a36Sopenharmony_ci		} v;
3462306a36Sopenharmony_ci	} *a;
3562306a36Sopenharmony_ci	int ret, i;
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_ci	if (device->runlist)
3862306a36Sopenharmony_ci		return 0;
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci	if (!(a = kmalloc(sizeof(*a), GFP_KERNEL)))
4162306a36Sopenharmony_ci		return -ENOMEM;
4262306a36Sopenharmony_ci	a->m.version = 1;
4362306a36Sopenharmony_ci	a->m.count = sizeof(a->v) / sizeof(a->v.runlists);
4462306a36Sopenharmony_ci	a->v.runlists.mthd = NV_DEVICE_HOST_RUNLISTS;
4562306a36Sopenharmony_ci	for (i = 0; i < ARRAY_SIZE(a->v.runlist); i++) {
4662306a36Sopenharmony_ci		a->v.runlist[i].mthd = NV_DEVICE_HOST_RUNLIST_ENGINES;
4762306a36Sopenharmony_ci		a->v.runlist[i].data = i;
4862306a36Sopenharmony_ci	}
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	ret = nvif_object_mthd(object, NV_DEVICE_V0_INFO, a, sizeof(*a));
5162306a36Sopenharmony_ci	if (ret)
5262306a36Sopenharmony_ci		goto done;
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_ci	device->runlists = fls64(a->v.runlists.data);
5562306a36Sopenharmony_ci	device->runlist = kcalloc(device->runlists, sizeof(*device->runlist),
5662306a36Sopenharmony_ci				  GFP_KERNEL);
5762306a36Sopenharmony_ci	if (!device->runlist) {
5862306a36Sopenharmony_ci		ret = -ENOMEM;
5962306a36Sopenharmony_ci		goto done;
6062306a36Sopenharmony_ci	}
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci	for (i = 0; i < device->runlists; i++) {
6362306a36Sopenharmony_ci		if (a->v.runlist[i].mthd != NV_DEVICE_INFO_INVALID)
6462306a36Sopenharmony_ci			device->runlist[i].engines = a->v.runlist[i].data;
6562306a36Sopenharmony_ci	}
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_cidone:
6862306a36Sopenharmony_ci	kfree(a);
6962306a36Sopenharmony_ci	return ret;
7062306a36Sopenharmony_ci}
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_ciu64
7362306a36Sopenharmony_cinvif_fifo_runlist(struct nvif_device *device, u64 engine)
7462306a36Sopenharmony_ci{
7562306a36Sopenharmony_ci	u64 runm = 0;
7662306a36Sopenharmony_ci	int ret, i;
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_ci	if ((ret = nvif_fifo_runlists(device)))
7962306a36Sopenharmony_ci		return runm;
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	for (i = 0; i < device->runlists; i++) {
8262306a36Sopenharmony_ci		if (device->runlist[i].engines & engine)
8362306a36Sopenharmony_ci			runm |= BIT_ULL(i);
8462306a36Sopenharmony_ci	}
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci	return runm;
8762306a36Sopenharmony_ci}
88