nv-tegra-mirror/linux-nvgpu
1
0
Fork 0
mirror of git://nv-tegra.nvidia.com/linux-nvgpu.git synced 2025-12-24 02:22:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

gpu: nvgpu: open TSG with the share token

Browse Source 
Implement OPEN_TSG ioctl with share tokens.

Bug 3677982
JIRA NVGPU-8681

Change-Id: If44aef863c932163df769acef5b3586f97aaecd3
Signed-off-by: Sagar Kamble <skamble@nvidia.com>
Reviewed-on: https://git-master.nvidia.com/r/c/linux-nvgpu/+/2792082
Reviewed-by: svc-mobile-coverity <svc-mobile-coverity@nvidia.com>
Reviewed-by: svc-mobile-cert <svc-mobile-cert@nvidia.com>
Reviewed-by: svcacv <svcacv@nvidia.com>
Reviewed-by: Scott Long <scottl@nvidia.com>
Reviewed-by: Vijayakumar Subbu <vsubbu@nvidia.com>
GVS: Gerrit_Virtual_Submit <buildbot_gerritrpt@nvidia.com>
This commit is contained in:
Sagar Kamble
2022-10-11 22:05:37 +05:30
committed by mobile promotions
parent 96f675595c
commit ce26e92de6
4 changed files with 128 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
drivers/gpu/nvgpu/os/linux/ioctl_tsg.c
View File

@@ -629,7 +629,9 @@ static int nvgpu_tsg_ioctl_revoke_share_token(struct gk20a *g,
#endif
int nvgpu_ioctl_tsg_open(struct gk20a *g, struct gk20a_ctrl_priv *ctrl_priv,
struct nvgpu_cdev *cdev, struct file *filp)
struct nvgpu_cdev *cdev, struct file *filp,
bool open_share, u64 source_device_instance_id,
u64 target_device_instance_id, u64 share_token)
{
struct tsg_private *priv;
struct nvgpu_tsg *tsg;
@@ -650,20 +652,37 @@ int nvgpu_ioctl_tsg_open(struct gk20a *g, struct gk20a_ctrl_priv *ctrl_priv,
goto free_ref;
}
err = gk20a_busy(g);
if (err) {
nvgpu_err(g, "failed to power on, %d", err);
goto free_mem;
if (!open_share) {
err = gk20a_busy(g);
if (err) {
nvgpu_err(g, "failed to power on, %d", err);
goto free_mem;
}
tsg = nvgpu_tsg_open(g, nvgpu_current_pid(g));
gk20a_idle(g);
if (tsg == NULL) {
err = -ENOMEM;
goto free_mem;
}
gk20a_sched_ctrl_tsg_added(g, tsg);
#ifndef CONFIG_NVGPU_TSG_SHARING
}
#else
} else {
tsg = nvgpu_gpu_open_tsg_with_share_token(g,
source_device_instance_id,
target_device_instance_id,
share_token);
if (tsg == NULL) {
nvgpu_err(g, "TSG open with token failed");
err = -EINVAL;
goto free_mem;
}
}
tsg = nvgpu_tsg_open(g, nvgpu_current_pid(g));
gk20a_idle(g);
if (!tsg) {
err = -ENOMEM;
goto free_mem;
}
#ifdef CONFIG_NVGPU_TSG_SHARING
if (ctrl_priv != NULL) {
err = nvgpu_tsg_add_ctrl_dev_inst_id(tsg, ctrl_priv);
if (err != 0) {
@@ -680,8 +699,6 @@ int nvgpu_ioctl_tsg_open(struct gk20a *g, struct gk20a_ctrl_priv *ctrl_priv,
priv->ctrl_priv = ctrl_priv;
filp->private_data = priv;
gk20a_sched_ctrl_tsg_added(g, tsg);
return 0;
free_mem:
@@ -708,7 +725,8 @@ int nvgpu_ioctl_tsg_dev_open(struct inode *inode, struct file *filp)
return ret;
}
ret = nvgpu_ioctl_tsg_open(g, NULL, cdev, filp);
ret = nvgpu_ioctl_tsg_open(g, NULL, cdev, filp, false,
0ULL, 0ULL, 0ULL);
gk20a_idle(g);
nvgpu_log_fn(g, "done");
@@ -727,13 +745,13 @@ void nvgpu_ioctl_tsg_release(struct nvgpu_ref *ref)
gk20a_sched_ctrl_tsg_removed(g, tsg);
nvgpu_tsg_release(ref);
nvgpu_put(g);
}
int nvgpu_ioctl_tsg_dev_release(struct inode *inode, struct file *filp)
{
struct tsg_private *priv = filp->private_data;
struct nvgpu_tsg *tsg;
struct gk20a *g;
#ifdef CONFIG_NVGPU_TSG_SHARING
u32 count;
int err;
@@ -745,15 +763,16 @@ int nvgpu_ioctl_tsg_dev_release(struct inode *inode, struct file *filp)
}
tsg = priv->tsg;
g = tsg->g;
#ifdef CONFIG_NVGPU_TSG_SHARING
nvgpu_mutex_acquire(&tsg->tsg_share_lock);
err = nvgpu_gpu_tsg_revoke_share_tokens(tsg->g,
err = nvgpu_gpu_tsg_revoke_share_tokens(g,
nvgpu_gpu_get_device_instance_id(priv->ctrl_priv),
tsg, &count);
if (err != 0) {
nvgpu_err(tsg->g, "revoke token(%llu) failed %d",
nvgpu_err(g, "revoke token(%llu) failed %d",
nvgpu_gpu_get_device_instance_id(priv->ctrl_priv),
err);
}
@@ -766,7 +785,10 @@ int nvgpu_ioctl_tsg_dev_release(struct inode *inode, struct file *filp)
#endif
nvgpu_ref_put(&tsg->refcount, nvgpu_ioctl_tsg_release);
nvgpu_kfree(tsg->g, priv);
nvgpu_kfree(g, priv);
nvgpu_put(g);
return 0;
}