Line data    Source code

       1             : /*
       2             :  * Copyright 2017 Valve Corporation
       3             :  *
       4             :  * Permission is hereby granted, free of charge, to any person obtaining a
       5             :  * copy of this software and associated documentation files (the "Software"),
       6             :  * to deal in the Software without restriction, including without limitation
       7             :  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
       8             :  * and/or sell copies of the Software, and to permit persons to whom the
       9             :  * Software is furnished to do so, subject to the following conditions:
      10             :  *
      11             :  * The above copyright notice and this permission notice shall be included in
      12             :  * all copies or substantial portions of the Software.
      13             :  *
      14             :  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
      15             :  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
      16             :  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
      17             :  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
      18             :  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
      19             :  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
      20             :  * OTHER DEALINGS IN THE SOFTWARE.
      21             :  *
      22             :  * Authors: Andres Rodriguez <andresx7@gmail.com>
      23             :  */
      24             : 
      25             : #include <linux/fdtable.h>
      26             : #include <linux/file.h>
      27             : #include <linux/pid.h>
      28             : 
      29             : #include <drm/amdgpu_drm.h>
      30             : 
      31             : #include "amdgpu.h"
      32             : #include "amdgpu_sched.h"
      33             : #include "amdgpu_vm.h"
      34             : 
      35           0 : static int amdgpu_sched_process_priority_override(struct amdgpu_device *adev,
      36             :                                                   int fd,
      37             :                                                   int32_t priority)
      38             : {
      39           0 :         struct fd f = fdget(fd);
      40             :         struct amdgpu_fpriv *fpriv;
      41             :         struct amdgpu_ctx *ctx;
      42             :         uint32_t id;
      43             :         int r;
      44             : 
      45           0 :         if (!f.file)
      46             :                 return -EINVAL;
      47             : 
      48           0 :         r = amdgpu_file_to_fpriv(f.file, &fpriv);
      49           0 :         if (r) {
      50           0 :                 fdput(f);
      51             :                 return r;
      52             :         }
      53             : 
      54           0 :         idr_for_each_entry(&fpriv->ctx_mgr.ctx_handles, ctx, id)
      55           0 :                 amdgpu_ctx_priority_override(ctx, priority);
      56             : 
      57           0 :         fdput(f);
      58             :         return 0;
      59             : }
      60             : 
      61           0 : static int amdgpu_sched_context_priority_override(struct amdgpu_device *adev,
      62             :                                                   int fd,
      63             :                                                   unsigned ctx_id,
      64             :                                                   int32_t priority)
      65             : {
      66           0 :         struct fd f = fdget(fd);
      67             :         struct amdgpu_fpriv *fpriv;
      68             :         struct amdgpu_ctx *ctx;
      69             :         int r;
      70             : 
      71           0 :         if (!f.file)
      72             :                 return -EINVAL;
      73             : 
      74           0 :         r = amdgpu_file_to_fpriv(f.file, &fpriv);
      75           0 :         if (r) {
      76           0 :                 fdput(f);
      77             :                 return r;
      78             :         }
      79             : 
      80           0 :         ctx = amdgpu_ctx_get(fpriv, ctx_id);
      81             : 
      82           0 :         if (!ctx) {
      83           0 :                 fdput(f);
      84             :                 return -EINVAL;
      85             :         }
      86             : 
      87           0 :         amdgpu_ctx_priority_override(ctx, priority);
      88           0 :         amdgpu_ctx_put(ctx);
      89           0 :         fdput(f);
      90             : 
      91             :         return 0;
      92             : }
      93             : 
      94           0 : int amdgpu_sched_ioctl(struct drm_device *dev, void *data,
      95             :                        struct drm_file *filp)
      96             : {
      97           0 :         union drm_amdgpu_sched *args = data;
      98           0 :         struct amdgpu_device *adev = drm_to_adev(dev);
      99             :         int r;
     100             : 
     101             :         /* First check the op, then the op's argument.
     102             :          */
     103           0 :         switch (args->in.op) {
     104             :         case AMDGPU_SCHED_OP_PROCESS_PRIORITY_OVERRIDE:
     105             :         case AMDGPU_SCHED_OP_CONTEXT_PRIORITY_OVERRIDE:
     106             :                 break;
     107             :         default:
     108           0 :                 DRM_ERROR("Invalid sched op specified: %d\n", args->in.op);
     109           0 :                 return -EINVAL;
     110             :         }
     111             : 
     112           0 :         if (!amdgpu_ctx_priority_is_valid(args->in.priority)) {
     113           0 :                 WARN(1, "Invalid context priority %d\n", args->in.priority);
     114           0 :                 return -EINVAL;
     115             :         }
     116             : 
     117           0 :         switch (args->in.op) {
     118             :         case AMDGPU_SCHED_OP_PROCESS_PRIORITY_OVERRIDE:
     119           0 :                 r = amdgpu_sched_process_priority_override(adev,
     120           0 :                                                            args->in.fd,
     121             :                                                            args->in.priority);
     122           0 :                 break;
     123             :         case AMDGPU_SCHED_OP_CONTEXT_PRIORITY_OVERRIDE:
     124           0 :                 r = amdgpu_sched_context_priority_override(adev,
     125           0 :                                                            args->in.fd,
     126             :                                                            args->in.ctx_id,
     127             :                                                            args->in.priority);
     128           0 :                 break;
     129             :         default:
     130             :                 /* Impossible.
     131             :                  */
     132             :                 r = -EINVAL;
     133             :                 break;
     134             :         }
     135             : 
     136             :         return r;
     137             : }