drivers/gpu/drm/amd/amdgpu/amdgpu_gfx.c

   1 /*
   2  * Copyright 2014 Advanced Micro Devices, Inc.
   3  * Copyright 2008 Red Hat Inc.
   4  * Copyright 2009 Jerome Glisse.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the "Software"),
   8  * to deal in the Software without restriction, including without limitation
   9  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  10  * and/or sell copies of the Software, and to permit persons to whom the
  11  * Software is furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  19  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
  20  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  21  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
  22  * OTHER DEALINGS IN THE SOFTWARE.
  23  *
  24  */
  25
  26 #include <linux/firmware.h>
  27 #include "amdgpu.h"
  28 #include "amdgpu_gfx.h"
  29 #include "amdgpu_rlc.h"
  30 #include "amdgpu_ras.h"
  31
  32 /* delay 0.1 second to enable gfx off feature */
  33 #define GFX_OFF_DELAY_ENABLE         msecs_to_jiffies(100)
  34
  35 #define GFX_OFF_NO_DELAY 0
  36
  37 /*
  38  * GPU GFX IP block helpers function.
  39  */
  40
  41 int amdgpu_gfx_mec_queue_to_bit(struct amdgpu_device *adev, int mec,
  42                                 int pipe, int queue)
  43 {
  44         int bit = 0;
  45
  46         bit += mec * adev->gfx.mec.num_pipe_per_mec
  47                 * adev->gfx.mec.num_queue_per_pipe;
  48         bit += pipe * adev->gfx.mec.num_queue_per_pipe;
  49         bit += queue;
  50
  51         return bit;
  52 }
  53
  54 void amdgpu_queue_mask_bit_to_mec_queue(struct amdgpu_device *adev, int bit,
  55                                  int *mec, int *pipe, int *queue)
  56 {
  57         *queue = bit % adev->gfx.mec.num_queue_per_pipe;
  58         *pipe = (bit / adev->gfx.mec.num_queue_per_pipe)
  59                 % adev->gfx.mec.num_pipe_per_mec;
  60         *mec = (bit / adev->gfx.mec.num_queue_per_pipe)
  61                / adev->gfx.mec.num_pipe_per_mec;
  62
  63 }
  64
  65 bool amdgpu_gfx_is_mec_queue_enabled(struct amdgpu_device *adev,
  66                                      int xcc_id, int mec, int pipe, int queue)
  67 {
  68         return test_bit(amdgpu_gfx_mec_queue_to_bit(adev, mec, pipe, queue),
  69                         adev->gfx.mec_bitmap[xcc_id].queue_bitmap);
  70 }
  71
  72 int amdgpu_gfx_me_queue_to_bit(struct amdgpu_device *adev,
  73                                int me, int pipe, int queue)
  74 {
  75         int bit = 0;
  76
  77         bit += me * adev->gfx.me.num_pipe_per_me
  78                 * adev->gfx.me.num_queue_per_pipe;
  79         bit += pipe * adev->gfx.me.num_queue_per_pipe;
  80         bit += queue;
  81
  82         return bit;
  83 }
  84
  85 void amdgpu_gfx_bit_to_me_queue(struct amdgpu_device *adev, int bit,
  86                                 int *me, int *pipe, int *queue)
  87 {
  88         *queue = bit % adev->gfx.me.num_queue_per_pipe;
  89         *pipe = (bit / adev->gfx.me.num_queue_per_pipe)
  90                 % adev->gfx.me.num_pipe_per_me;
  91         *me = (bit / adev->gfx.me.num_queue_per_pipe)
  92                 / adev->gfx.me.num_pipe_per_me;
  93 }
  94
  95 bool amdgpu_gfx_is_me_queue_enabled(struct amdgpu_device *adev,
  96                                     int me, int pipe, int queue)
  97 {
  98         return test_bit(amdgpu_gfx_me_queue_to_bit(adev, me, pipe, queue),
  99                         adev->gfx.me.queue_bitmap);
 100 }
 101
 102 /**
 103  * amdgpu_gfx_parse_disable_cu - Parse the disable_cu module parameter
 104  *
 105  * @mask: array in which the per-shader array disable masks will be stored
 106  * @max_se: number of SEs
 107  * @max_sh: number of SHs
 108  *
 109  * The bitmask of CUs to be disabled in the shader array determined by se and
 110  * sh is stored in mask[se * max_sh + sh].
 111  */
 112 void amdgpu_gfx_parse_disable_cu(unsigned *mask, unsigned max_se, unsigned max_sh)
 113 {
 114         unsigned se, sh, cu;
 115         const char *p;
 116
 117         memset(mask, 0, sizeof(*mask) * max_se * max_sh);
 118
 119         if (!amdgpu_disable_cu || !*amdgpu_disable_cu)
 120                 return;
 121
 122         p = amdgpu_disable_cu;
 123         for (;;) {
 124                 char *next;
 125                 int ret = sscanf(p, "%u.%u.%u", &se, &sh, &cu);
 126                 if (ret < 3) {
 127                         DRM_ERROR("amdgpu: could not parse disable_cu\n");
 128                         return;
 129                 }
 130
 131                 if (se < max_se && sh < max_sh && cu < 16) {
 132                         DRM_INFO("amdgpu: disabling CU %u.%u.%u\n", se, sh, cu);
 133                         mask[se * max_sh + sh] |= 1u << cu;
 134                 } else {
 135                         DRM_ERROR("amdgpu: disable_cu %u.%u.%u is out of range\n",
 136                                   se, sh, cu);
 137                 }
 138
 139                 next = strchr(p, ',');
 140                 if (!next)
 141                         break;
 142                 p = next + 1;
 143         }
 144 }
 145
 146 static bool amdgpu_gfx_is_graphics_multipipe_capable(struct amdgpu_device *adev)
 147 {
 148         return amdgpu_async_gfx_ring && adev->gfx.me.num_pipe_per_me > 1;
 149 }
 150
 151 static bool amdgpu_gfx_is_compute_multipipe_capable(struct amdgpu_device *adev)
 152 {
 153         if (amdgpu_compute_multipipe != -1) {
 154                 DRM_INFO("amdgpu: forcing compute pipe policy %d\n",
 155                          amdgpu_compute_multipipe);
 156                 return amdgpu_compute_multipipe == 1;
 157         }
 158
 159         if (adev->ip_versions[GC_HWIP][0] > IP_VERSION(9, 0, 0))
 160                 return true;
 161
 162         /* FIXME: spreading the queues across pipes causes perf regressions
 163          * on POLARIS11 compute workloads */
 164         if (adev->asic_type == CHIP_POLARIS11)
 165                 return false;
 166
 167         return adev->gfx.mec.num_mec > 1;
 168 }
 169
 170 bool amdgpu_gfx_is_high_priority_graphics_queue(struct amdgpu_device *adev,
 171                                                 struct amdgpu_ring *ring)
 172 {
 173         int queue = ring->queue;
 174         int pipe = ring->pipe;
 175
 176         /* Policy: use pipe1 queue0 as high priority graphics queue if we
 177          * have more than one gfx pipe.
 178          */
 179         if (amdgpu_gfx_is_graphics_multipipe_capable(adev) &&
 180             adev->gfx.num_gfx_rings > 1 && pipe == 1 && queue == 0) {
 181                 int me = ring->me;
 182                 int bit;
 183
 184                 bit = amdgpu_gfx_me_queue_to_bit(adev, me, pipe, queue);
 185                 if (ring == &adev->gfx.gfx_ring[bit])
 186                         return true;
 187         }
 188
 189         return false;
 190 }
 191
 192 bool amdgpu_gfx_is_high_priority_compute_queue(struct amdgpu_device *adev,
 193                                                struct amdgpu_ring *ring)
 194 {
 195         /* Policy: use 1st queue as high priority compute queue if we
 196          * have more than one compute queue.
 197          */
 198         if (adev->gfx.num_compute_rings > 1 &&
 199             ring == &adev->gfx.compute_ring[0])
 200                 return true;
 201
 202         return false;
 203 }
 204
 205 void amdgpu_gfx_compute_queue_acquire(struct amdgpu_device *adev)
 206 {
 207         int i, j, queue, pipe;
 208         bool multipipe_policy = amdgpu_gfx_is_compute_multipipe_capable(adev);
 209         int max_queues_per_mec = min(adev->gfx.mec.num_pipe_per_mec *
 210                                      adev->gfx.mec.num_queue_per_pipe,
 211                                      adev->gfx.num_compute_rings);
 212         int num_xcc = adev->gfx.xcc_mask ? NUM_XCC(adev->gfx.xcc_mask) : 1;
 213
 214         if (multipipe_policy) {
 215                 /* policy: make queues evenly cross all pipes on MEC1 only
 216                  * for multiple xcc, just use the original policy for simplicity */
 217                 for (j = 0; j < num_xcc; j++) {
 218                         for (i = 0; i < max_queues_per_mec; i++) {
 219                                 pipe = i % adev->gfx.mec.num_pipe_per_mec;
 220                                 queue = (i / adev->gfx.mec.num_pipe_per_mec) %
 221                                          adev->gfx.mec.num_queue_per_pipe;
 222
 223                                 set_bit(pipe * adev->gfx.mec.num_queue_per_pipe + queue,
 224                                         adev->gfx.mec_bitmap[j].queue_bitmap);
 225                         }
 226                 }
 227         } else {
 228                 /* policy: amdgpu owns all queues in the given pipe */
 229                 for (j = 0; j < num_xcc; j++) {
 230                         for (i = 0; i < max_queues_per_mec; ++i)
 231                                 set_bit(i, adev->gfx.mec_bitmap[j].queue_bitmap);
 232                 }
 233         }
 234
 235         for (j = 0; j < num_xcc; j++) {
 236                 dev_dbg(adev->dev, "mec queue bitmap weight=%d\n",
 237                         bitmap_weight(adev->gfx.mec_bitmap[j].queue_bitmap, AMDGPU_MAX_COMPUTE_QUEUES));
 238         }
 239 }
 240
 241 void amdgpu_gfx_graphics_queue_acquire(struct amdgpu_device *adev)
 242 {
 243         int i, queue, pipe;
 244         bool multipipe_policy = amdgpu_gfx_is_graphics_multipipe_capable(adev);
 245         int max_queues_per_me = adev->gfx.me.num_pipe_per_me *
 246                                         adev->gfx.me.num_queue_per_pipe;
 247
 248         if (multipipe_policy) {
 249                 /* policy: amdgpu owns the first queue per pipe at this stage
 250                  * will extend to mulitple queues per pipe later */
 251                 for (i = 0; i < max_queues_per_me; i++) {
 252                         pipe = i % adev->gfx.me.num_pipe_per_me;
 253                         queue = (i / adev->gfx.me.num_pipe_per_me) %
 254                                 adev->gfx.me.num_queue_per_pipe;
 255
 256                         set_bit(pipe * adev->gfx.me.num_queue_per_pipe + queue,
 257                                 adev->gfx.me.queue_bitmap);
 258                 }
 259         } else {
 260                 for (i = 0; i < max_queues_per_me; ++i)
 261                         set_bit(i, adev->gfx.me.queue_bitmap);
 262         }
 263
 264         /* update the number of active graphics rings */
 265         adev->gfx.num_gfx_rings =
 266                 bitmap_weight(adev->gfx.me.queue_bitmap, AMDGPU_MAX_GFX_QUEUES);
 267 }
 268
 269 static int amdgpu_gfx_kiq_acquire(struct amdgpu_device *adev,
 270                                   struct amdgpu_ring *ring, int xcc_id)
 271 {
 272         int queue_bit;
 273         int mec, pipe, queue;
 274
 275         queue_bit = adev->gfx.mec.num_mec
 276                     * adev->gfx.mec.num_pipe_per_mec
 277                     * adev->gfx.mec.num_queue_per_pipe;
 278
 279         while (--queue_bit >= 0) {
 280                 if (test_bit(queue_bit, adev->gfx.mec_bitmap[xcc_id].queue_bitmap))
 281                         continue;
 282
 283                 amdgpu_queue_mask_bit_to_mec_queue(adev, queue_bit, &mec, &pipe, &queue);
 284
 285                 /*
 286                  * 1. Using pipes 2/3 from MEC 2 seems cause problems.
 287                  * 2. It must use queue id 0, because CGPG_IDLE/SAVE/LOAD/RUN
 288                  * only can be issued on queue 0.
 289                  */
 290                 if ((mec == 1 && pipe > 1) || queue != 0)
 291                         continue;
 292
 293                 ring->me = mec + 1;
 294                 ring->pipe = pipe;
 295                 ring->queue = queue;
 296
 297                 return 0;
 298         }
 299
 300         dev_err(adev->dev, "Failed to find a queue for KIQ\n");
 301         return -EINVAL;
 302 }
 303
 304 int amdgpu_gfx_kiq_init_ring(struct amdgpu_device *adev,
 305                              struct amdgpu_ring *ring,
 306                              struct amdgpu_irq_src *irq, int xcc_id)
 307 {
 308         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 309         int r = 0;
 310
 311         spin_lock_init(&kiq->ring_lock);
 312
 313         ring->adev = NULL;
 314         ring->ring_obj = NULL;
 315         ring->use_doorbell = true;
 316         ring->xcc_id = xcc_id;
 317         ring->vm_hub = AMDGPU_GFXHUB(xcc_id);
 318         ring->doorbell_index = (adev->doorbell_index.kiq + xcc_id) << 1;
 319
 320         r = amdgpu_gfx_kiq_acquire(adev, ring, xcc_id);
 321         if (r)
 322                 return r;
 323
 324         ring->eop_gpu_addr = kiq->eop_gpu_addr;
 325         ring->no_scheduler = true;
 326         sprintf(ring->name, "kiq_%d.%d.%d.%d", xcc_id, ring->me, ring->pipe, ring->queue);
 327         r = amdgpu_ring_init(adev, ring, 1024, irq, AMDGPU_CP_KIQ_IRQ_DRIVER0,
 328                              AMDGPU_RING_PRIO_DEFAULT, NULL);
 329         if (r)
 330                 dev_warn(adev->dev, "(%d) failed to init kiq ring\n", r);
 331
 332         return r;
 333 }
 334
 335 void amdgpu_gfx_kiq_free_ring(struct amdgpu_ring *ring)
 336 {
 337         amdgpu_ring_fini(ring);
 338 }
 339
 340 void amdgpu_gfx_kiq_fini(struct amdgpu_device *adev, int xcc_id)
 341 {
 342         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 343
 344         amdgpu_bo_free_kernel(&kiq->eop_obj, &kiq->eop_gpu_addr, NULL);
 345 }
 346
 347 int amdgpu_gfx_kiq_init(struct amdgpu_device *adev,
 348                         unsigned hpd_size, int xcc_id)
 349 {
 350         int r;
 351         u32 *hpd;
 352         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 353
 354         r = amdgpu_bo_create_kernel(adev, hpd_size, PAGE_SIZE,
 355                                     AMDGPU_GEM_DOMAIN_GTT, &kiq->eop_obj,
 356                                     &kiq->eop_gpu_addr, (void **)&hpd);
 357         if (r) {
 358                 dev_warn(adev->dev, "failed to create KIQ bo (%d).\n", r);
 359                 return r;
 360         }
 361
 362         memset(hpd, 0, hpd_size);
 363
 364         r = amdgpu_bo_reserve(kiq->eop_obj, true);
 365         if (unlikely(r != 0))
 366                 dev_warn(adev->dev, "(%d) reserve kiq eop bo failed\n", r);
 367         amdgpu_bo_kunmap(kiq->eop_obj);
 368         amdgpu_bo_unreserve(kiq->eop_obj);
 369
 370         return 0;
 371 }
 372
 373 /* create MQD for each compute/gfx queue */
 374 int amdgpu_gfx_mqd_sw_init(struct amdgpu_device *adev,
 375                            unsigned mqd_size, int xcc_id)
 376 {
 377         int r, i, j;
 378         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 379         struct amdgpu_ring *ring = &kiq->ring;
 380         u32 domain = AMDGPU_GEM_DOMAIN_GTT;
 381
 382         /* Only enable on gfx10 and 11 for now to avoid changing behavior on older chips */
 383         if (adev->ip_versions[GC_HWIP][0] >= IP_VERSION(10, 0, 0))
 384                 domain |= AMDGPU_GEM_DOMAIN_VRAM;
 385
 386         /* create MQD for KIQ */
 387         if (!adev->enable_mes_kiq && !ring->mqd_obj) {
 388                 /* originaly the KIQ MQD is put in GTT domain, but for SRIOV VRAM domain is a must
 389                  * otherwise hypervisor trigger SAVE_VF fail after driver unloaded which mean MQD
 390                  * deallocated and gart_unbind, to strict diverage we decide to use VRAM domain for
 391                  * KIQ MQD no matter SRIOV or Bare-metal
 392                  */
 393                 r = amdgpu_bo_create_kernel(adev, mqd_size, PAGE_SIZE,
 394                                             AMDGPU_GEM_DOMAIN_VRAM |
 395                                             AMDGPU_GEM_DOMAIN_GTT,
 396                                             &ring->mqd_obj,
 397                                             &ring->mqd_gpu_addr,
 398                                             &ring->mqd_ptr);
 399                 if (r) {
 400                         dev_warn(adev->dev, "failed to create ring mqd ob (%d)", r);
 401                         return r;
 402                 }
 403
 404                 /* prepare MQD backup */
 405                 kiq->mqd_backup = kmalloc(mqd_size, GFP_KERNEL);
 406                 if (!kiq->mqd_backup)
 407                                 dev_warn(adev->dev, "no memory to create MQD backup for ring %s\n", ring->name);
 408         }
 409
 410         if (adev->asic_type >= CHIP_NAVI10 && amdgpu_async_gfx_ring) {
 411                 /* create MQD for each KGQ */
 412                 for (i = 0; i < adev->gfx.num_gfx_rings; i++) {
 413                         ring = &adev->gfx.gfx_ring[i];
 414                         if (!ring->mqd_obj) {
 415                                 r = amdgpu_bo_create_kernel(adev, mqd_size, PAGE_SIZE,
 416                                                             domain, &ring->mqd_obj,
 417                                                             &ring->mqd_gpu_addr, &ring->mqd_ptr);
 418                                 if (r) {
 419                                         dev_warn(adev->dev, "failed to create ring mqd bo (%d)", r);
 420                                         return r;
 421                                 }
 422
 423                                 ring->mqd_size = mqd_size;
 424                                 /* prepare MQD backup */
 425                                 adev->gfx.me.mqd_backup[i] = kmalloc(mqd_size, GFP_KERNEL);
 426                                 if (!adev->gfx.me.mqd_backup[i])
 427                                         dev_warn(adev->dev, "no memory to create MQD backup for ring %s\n", ring->name);
 428                         }
 429                 }
 430         }
 431
 432         /* create MQD for each KCQ */
 433         for (i = 0; i < adev->gfx.num_compute_rings; i++) {
 434                 j = i + xcc_id * adev->gfx.num_compute_rings;
 435                 ring = &adev->gfx.compute_ring[j];
 436                 if (!ring->mqd_obj) {
 437                         r = amdgpu_bo_create_kernel(adev, mqd_size, PAGE_SIZE,
 438                                                     domain, &ring->mqd_obj,
 439                                                     &ring->mqd_gpu_addr, &ring->mqd_ptr);
 440                         if (r) {
 441                                 dev_warn(adev->dev, "failed to create ring mqd bo (%d)", r);
 442                                 return r;
 443                         }
 444
 445                         ring->mqd_size = mqd_size;
 446                         /* prepare MQD backup */
 447                         adev->gfx.mec.mqd_backup[j + xcc_id * adev->gfx.num_compute_rings] = kmalloc(mqd_size, GFP_KERNEL);
 448                         if (!adev->gfx.mec.mqd_backup[j + xcc_id * adev->gfx.num_compute_rings])
 449                                 dev_warn(adev->dev, "no memory to create MQD backup for ring %s\n", ring->name);
 450                 }
 451         }
 452
 453         return 0;
 454 }
 455
 456 void amdgpu_gfx_mqd_sw_fini(struct amdgpu_device *adev, int xcc_id)
 457 {
 458         struct amdgpu_ring *ring = NULL;
 459         int i, j;
 460         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 461
 462         if (adev->asic_type >= CHIP_NAVI10 && amdgpu_async_gfx_ring) {
 463                 for (i = 0; i < adev->gfx.num_gfx_rings; i++) {
 464                         ring = &adev->gfx.gfx_ring[i];
 465                         kfree(adev->gfx.me.mqd_backup[i]);
 466                         amdgpu_bo_free_kernel(&ring->mqd_obj,
 467                                               &ring->mqd_gpu_addr,
 468                                               &ring->mqd_ptr);
 469                 }
 470         }
 471
 472         for (i = 0; i < adev->gfx.num_compute_rings; i++) {
 473                 j = i + xcc_id * adev->gfx.num_compute_rings;
 474                 ring = &adev->gfx.compute_ring[j];
 475                 kfree(adev->gfx.mec.mqd_backup[j]);
 476                 amdgpu_bo_free_kernel(&ring->mqd_obj,
 477                                       &ring->mqd_gpu_addr,
 478                                       &ring->mqd_ptr);
 479         }
 480
 481         ring = &kiq->ring;
 482         kfree(kiq->mqd_backup);
 483         kfree(adev->gfx.mec.mqd_backup[AMDGPU_MAX_COMPUTE_RINGS]);
 484         amdgpu_bo_free_kernel(&ring->mqd_obj,
 485                               &ring->mqd_gpu_addr,
 486                               &ring->mqd_ptr);
 487 }
 488
 489 int amdgpu_gfx_disable_kcq(struct amdgpu_device *adev, int xcc_id)
 490 {
 491         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 492         struct amdgpu_ring *kiq_ring = &kiq->ring;
 493         int i, r = 0;
 494         int j;
 495
 496         if (!kiq->pmf || !kiq->pmf->kiq_unmap_queues)
 497                 return -EINVAL;
 498
 499         spin_lock(&kiq->ring_lock);
 500         if (amdgpu_ring_alloc(kiq_ring, kiq->pmf->unmap_queues_size *
 501                                         adev->gfx.num_compute_rings)) {
 502                 spin_unlock(&kiq->ring_lock);
 503                 return -ENOMEM;
 504         }
 505
 506         for (i = 0; i < adev->gfx.num_compute_rings; i++) {
 507                 j = i + xcc_id * adev->gfx.num_compute_rings;
 508                 kiq->pmf->kiq_unmap_queues(kiq_ring,
 509                                            &adev->gfx.compute_ring[i],
 510                                            RESET_QUEUES, 0, 0);
 511         }
 512
 513         if (kiq_ring->sched.ready && !adev->job_hang)
 514                 r = amdgpu_ring_test_helper(kiq_ring);
 515         spin_unlock(&kiq->ring_lock);
 516
 517         return r;
 518 }
 519
 520 int amdgpu_gfx_disable_kgq(struct amdgpu_device *adev, int xcc_id)
 521 {
 522         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 523         struct amdgpu_ring *kiq_ring = &kiq->ring;
 524         int i, r = 0;
 525         int j;
 526
 527         if (!kiq->pmf || !kiq->pmf->kiq_unmap_queues)
 528                 return -EINVAL;
 529
 530         spin_lock(&kiq->ring_lock);
 531         if (amdgpu_gfx_is_master_xcc(adev, xcc_id)) {
 532                 if (amdgpu_ring_alloc(kiq_ring, kiq->pmf->unmap_queues_size *
 533                                                 adev->gfx.num_gfx_rings)) {
 534                         spin_unlock(&kiq->ring_lock);
 535                         return -ENOMEM;
 536                 }
 537
 538                 for (i = 0; i < adev->gfx.num_gfx_rings; i++) {
 539                         j = i + xcc_id * adev->gfx.num_gfx_rings;
 540                         kiq->pmf->kiq_unmap_queues(kiq_ring,
 541                                                    &adev->gfx.gfx_ring[i],
 542                                                    PREEMPT_QUEUES, 0, 0);
 543                 }
 544         }
 545
 546         if (adev->gfx.kiq[0].ring.sched.ready && !adev->job_hang)
 547                 r = amdgpu_ring_test_helper(kiq_ring);
 548         spin_unlock(&kiq->ring_lock);
 549
 550         return r;
 551 }
 552
 553 int amdgpu_queue_mask_bit_to_set_resource_bit(struct amdgpu_device *adev,
 554                                         int queue_bit)
 555 {
 556         int mec, pipe, queue;
 557         int set_resource_bit = 0;
 558
 559         amdgpu_queue_mask_bit_to_mec_queue(adev, queue_bit, &mec, &pipe, &queue);
 560
 561         set_resource_bit = mec * 4 * 8 + pipe * 8 + queue;
 562
 563         return set_resource_bit;
 564 }
 565
 566 int amdgpu_gfx_enable_kcq(struct amdgpu_device *adev, int xcc_id)
 567 {
 568         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 569         struct amdgpu_ring *kiq_ring = &kiq->ring;
 570         uint64_t queue_mask = 0;
 571         int r, i, j;
 572
 573         if (!kiq->pmf || !kiq->pmf->kiq_map_queues || !kiq->pmf->kiq_set_resources)
 574                 return -EINVAL;
 575
 576         for (i = 0; i < AMDGPU_MAX_COMPUTE_QUEUES; ++i) {
 577                 if (!test_bit(i, adev->gfx.mec_bitmap[xcc_id].queue_bitmap))
 578                         continue;
 579
 580                 /* This situation may be hit in the future if a new HW
 581                  * generation exposes more than 64 queues. If so, the
 582                  * definition of queue_mask needs updating */
 583                 if (WARN_ON(i > (sizeof(queue_mask)*8))) {
 584                         DRM_ERROR("Invalid KCQ enabled: %d\n", i);
 585                         break;
 586                 }
 587
 588                 queue_mask |= (1ull << amdgpu_queue_mask_bit_to_set_resource_bit(adev, i));
 589         }
 590
 591         DRM_INFO("kiq ring mec %d pipe %d q %d\n", kiq_ring->me, kiq_ring->pipe,
 592                                                         kiq_ring->queue);
 593         spin_lock(&kiq->ring_lock);
 594         r = amdgpu_ring_alloc(kiq_ring, kiq->pmf->map_queues_size *
 595                                         adev->gfx.num_compute_rings +
 596                                         kiq->pmf->set_resources_size);
 597         if (r) {
 598                 DRM_ERROR("Failed to lock KIQ (%d).\n", r);
 599                 spin_unlock(&kiq->ring_lock);
 600                 return r;
 601         }
 602
 603         if (adev->enable_mes)
 604                 queue_mask = ~0ULL;
 605
 606         kiq->pmf->kiq_set_resources(kiq_ring, queue_mask);
 607         for (i = 0; i < adev->gfx.num_compute_rings; i++) {
 608                 j = i + xcc_id * adev->gfx.num_compute_rings;
 609                         kiq->pmf->kiq_map_queues(kiq_ring,
 610                                                  &adev->gfx.compute_ring[j]);
 611         }
 612
 613         r = amdgpu_ring_test_helper(kiq_ring);
 614         spin_unlock(&kiq->ring_lock);
 615         if (r)
 616                 DRM_ERROR("KCQ enable failed\n");
 617
 618         return r;
 619 }
 620
 621 int amdgpu_gfx_enable_kgq(struct amdgpu_device *adev, int xcc_id)
 622 {
 623         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 624         struct amdgpu_ring *kiq_ring = &kiq->ring;
 625         int r, i, j;
 626
 627         if (!kiq->pmf || !kiq->pmf->kiq_map_queues)
 628                 return -EINVAL;
 629
 630         spin_lock(&kiq->ring_lock);
 631         /* No need to map kcq on the slave */
 632         if (amdgpu_gfx_is_master_xcc(adev, xcc_id)) {
 633                 r = amdgpu_ring_alloc(kiq_ring, kiq->pmf->map_queues_size *
 634                                                 adev->gfx.num_gfx_rings);
 635                 if (r) {
 636                         DRM_ERROR("Failed to lock KIQ (%d).\n", r);
 637                         spin_unlock(&kiq->ring_lock);
 638                         return r;
 639                 }
 640
 641                 for (i = 0; i < adev->gfx.num_gfx_rings; i++) {
 642                         j = i + xcc_id * adev->gfx.num_gfx_rings;
 643                         kiq->pmf->kiq_map_queues(kiq_ring,
 644                                                  &adev->gfx.gfx_ring[i]);
 645                 }
 646         }
 647
 648         r = amdgpu_ring_test_helper(kiq_ring);
 649         spin_unlock(&kiq->ring_lock);
 650         if (r)
 651                 DRM_ERROR("KCQ enable failed\n");
 652
 653         return r;
 654 }
 655
 656 /* amdgpu_gfx_off_ctrl - Handle gfx off feature enable/disable
 657  *
 658  * @adev: amdgpu_device pointer
 659  * @bool enable true: enable gfx off feature, false: disable gfx off feature
 660  *
 661  * 1. gfx off feature will be enabled by gfx ip after gfx cg gp enabled.
 662  * 2. other client can send request to disable gfx off feature, the request should be honored.
 663  * 3. other client can cancel their request of disable gfx off feature
 664  * 4. other client should not send request to enable gfx off feature before disable gfx off feature.
 665  */
 666
 667 void amdgpu_gfx_off_ctrl(struct amdgpu_device *adev, bool enable)
 668 {
 669         unsigned long delay = GFX_OFF_DELAY_ENABLE;
 670
 671         if (!(adev->pm.pp_feature & PP_GFXOFF_MASK))
 672                 return;
 673
 674         mutex_lock(&adev->gfx.gfx_off_mutex);
 675
 676         if (enable) {
 677                 /* If the count is already 0, it means there's an imbalance bug somewhere.
 678                  * Note that the bug may be in a different caller than the one which triggers the
 679                  * WARN_ON_ONCE.
 680                  */
 681                 if (WARN_ON_ONCE(adev->gfx.gfx_off_req_count == 0))
 682                         goto unlock;
 683
 684                 adev->gfx.gfx_off_req_count--;
 685
 686                 if (adev->gfx.gfx_off_req_count == 0 &&
 687                     !adev->gfx.gfx_off_state) {
 688                         /* If going to s2idle, no need to wait */
 689                         if (adev->in_s0ix) {
 690                                 if (!amdgpu_dpm_set_powergating_by_smu(adev,
 691                                                 AMD_IP_BLOCK_TYPE_GFX, true))
 692                                         adev->gfx.gfx_off_state = true;
 693                         } else {
 694                                 schedule_delayed_work(&adev->gfx.gfx_off_delay_work,
 695                                               delay);
 696                         }
 697                 }
 698         } else {
 699                 if (adev->gfx.gfx_off_req_count == 0) {
 700                         cancel_delayed_work_sync(&adev->gfx.gfx_off_delay_work);
 701
 702                         if (adev->gfx.gfx_off_state &&
 703                             !amdgpu_dpm_set_powergating_by_smu(adev, AMD_IP_BLOCK_TYPE_GFX, false)) {
 704                                 adev->gfx.gfx_off_state = false;
 705
 706                                 if (adev->gfx.funcs->init_spm_golden) {
 707                                         dev_dbg(adev->dev,
 708                                                 "GFXOFF is disabled, re-init SPM golden settings\n");
 709                                         amdgpu_gfx_init_spm_golden(adev);
 710                                 }
 711                         }
 712                 }
 713
 714                 adev->gfx.gfx_off_req_count++;
 715         }
 716
 717 unlock:
 718         mutex_unlock(&adev->gfx.gfx_off_mutex);
 719 }
 720
 721 int amdgpu_set_gfx_off_residency(struct amdgpu_device *adev, bool value)
 722 {
 723         int r = 0;
 724
 725         mutex_lock(&adev->gfx.gfx_off_mutex);
 726
 727         r = amdgpu_dpm_set_residency_gfxoff(adev, value);
 728
 729         mutex_unlock(&adev->gfx.gfx_off_mutex);
 730
 731         return r;
 732 }
 733
 734 int amdgpu_get_gfx_off_residency(struct amdgpu_device *adev, u32 *value)
 735 {
 736         int r = 0;
 737
 738         mutex_lock(&adev->gfx.gfx_off_mutex);
 739
 740         r = amdgpu_dpm_get_residency_gfxoff(adev, value);
 741
 742         mutex_unlock(&adev->gfx.gfx_off_mutex);
 743
 744         return r;
 745 }
 746
 747 int amdgpu_get_gfx_off_entrycount(struct amdgpu_device *adev, u64 *value)
 748 {
 749         int r = 0;
 750
 751         mutex_lock(&adev->gfx.gfx_off_mutex);
 752
 753         r = amdgpu_dpm_get_entrycount_gfxoff(adev, value);
 754
 755         mutex_unlock(&adev->gfx.gfx_off_mutex);
 756
 757         return r;
 758 }
 759
 760 int amdgpu_get_gfx_off_status(struct amdgpu_device *adev, uint32_t *value)
 761 {
 762
 763         int r = 0;
 764
 765         mutex_lock(&adev->gfx.gfx_off_mutex);
 766
 767         r = amdgpu_dpm_get_status_gfxoff(adev, value);
 768
 769         mutex_unlock(&adev->gfx.gfx_off_mutex);
 770
 771         return r;
 772 }
 773
 774 int amdgpu_gfx_ras_late_init(struct amdgpu_device *adev, struct ras_common_if *ras_block)
 775 {
 776         int r;
 777
 778         if (amdgpu_ras_is_supported(adev, ras_block->block)) {
 779                 if (!amdgpu_persistent_edc_harvesting_supported(adev))
 780                         amdgpu_ras_reset_error_status(adev, AMDGPU_RAS_BLOCK__GFX);
 781
 782                 r = amdgpu_ras_block_late_init(adev, ras_block);
 783                 if (r)
 784                         return r;
 785
 786                 if (adev->gfx.cp_ecc_error_irq.funcs) {
 787                         r = amdgpu_irq_get(adev, &adev->gfx.cp_ecc_error_irq, 0);
 788                         if (r)
 789                                 goto late_fini;
 790                 }
 791         } else {
 792                 amdgpu_ras_feature_enable_on_boot(adev, ras_block, 0);
 793         }
 794
 795         return 0;
 796 late_fini:
 797         amdgpu_ras_block_late_fini(adev, ras_block);
 798         return r;
 799 }
 800
 801 int amdgpu_gfx_ras_sw_init(struct amdgpu_device *adev)
 802 {
 803         int err = 0;
 804         struct amdgpu_gfx_ras *ras = NULL;
 805
 806         /* adev->gfx.ras is NULL, which means gfx does not
 807          * support ras function, then do nothing here.
 808          */
 809         if (!adev->gfx.ras)
 810                 return 0;
 811
 812         ras = adev->gfx.ras;
 813
 814         err = amdgpu_ras_register_ras_block(adev, &ras->ras_block);
 815         if (err) {
 816                 dev_err(adev->dev, "Failed to register gfx ras block!\n");
 817                 return err;
 818         }
 819
 820         strcpy(ras->ras_block.ras_comm.name, "gfx");
 821         ras->ras_block.ras_comm.block = AMDGPU_RAS_BLOCK__GFX;
 822         ras->ras_block.ras_comm.type = AMDGPU_RAS_ERROR__MULTI_UNCORRECTABLE;
 823         adev->gfx.ras_if = &ras->ras_block.ras_comm;
 824
 825         /* If not define special ras_late_init function, use gfx default ras_late_init */
 826         if (!ras->ras_block.ras_late_init)
 827                 ras->ras_block.ras_late_init = amdgpu_gfx_ras_late_init;
 828
 829         /* If not defined special ras_cb function, use default ras_cb */
 830         if (!ras->ras_block.ras_cb)
 831                 ras->ras_block.ras_cb = amdgpu_gfx_process_ras_data_cb;
 832
 833         return 0;
 834 }
 835
 836 int amdgpu_gfx_poison_consumption_handler(struct amdgpu_device *adev,
 837                                                 struct amdgpu_iv_entry *entry)
 838 {
 839         if (adev->gfx.ras && adev->gfx.ras->poison_consumption_handler)
 840                 return adev->gfx.ras->poison_consumption_handler(adev, entry);
 841
 842         return 0;
 843 }
 844
 845 int amdgpu_gfx_process_ras_data_cb(struct amdgpu_device *adev,
 846                 void *err_data,
 847                 struct amdgpu_iv_entry *entry)
 848 {
 849         /* TODO ue will trigger an interrupt.
 850          *
 851          * When “Full RAS” is enabled, the per-IP interrupt sources should
 852          * be disabled and the driver should only look for the aggregated
 853          * interrupt via sync flood
 854          */
 855         if (!amdgpu_ras_is_supported(adev, AMDGPU_RAS_BLOCK__GFX)) {
 856                 kgd2kfd_set_sram_ecc_flag(adev->kfd.dev);
 857                 if (adev->gfx.ras && adev->gfx.ras->ras_block.hw_ops &&
 858                     adev->gfx.ras->ras_block.hw_ops->query_ras_error_count)
 859                         adev->gfx.ras->ras_block.hw_ops->query_ras_error_count(adev, err_data);
 860                 amdgpu_ras_reset_gpu(adev);
 861         }
 862         return AMDGPU_RAS_SUCCESS;
 863 }
 864
 865 int amdgpu_gfx_cp_ecc_error_irq(struct amdgpu_device *adev,
 866                                   struct amdgpu_irq_src *source,
 867                                   struct amdgpu_iv_entry *entry)
 868 {
 869         struct ras_common_if *ras_if = adev->gfx.ras_if;
 870         struct ras_dispatch_if ih_data = {
 871                 .entry = entry,
 872         };
 873
 874         if (!ras_if)
 875                 return 0;
 876
 877         ih_data.head = *ras_if;
 878
 879         DRM_ERROR("CP ECC ERROR IRQ\n");
 880         amdgpu_ras_interrupt_dispatch(adev, &ih_data);
 881         return 0;
 882 }
 883
 884 uint32_t amdgpu_kiq_rreg(struct amdgpu_device *adev, uint32_t reg)
 885 {
 886         signed long r, cnt = 0;
 887         unsigned long flags;
 888         uint32_t seq, reg_val_offs = 0, value = 0;
 889         struct amdgpu_kiq *kiq = &adev->gfx.kiq[0];
 890         struct amdgpu_ring *ring = &kiq->ring;
 891
 892         if (amdgpu_device_skip_hw_access(adev))
 893                 return 0;
 894
 895         if (adev->mes.ring.sched.ready)
 896                 return amdgpu_mes_rreg(adev, reg);
 897
 898         BUG_ON(!ring->funcs->emit_rreg);
 899
 900         spin_lock_irqsave(&kiq->ring_lock, flags);
 901         if (amdgpu_device_wb_get(adev, &reg_val_offs)) {
 902                 pr_err("critical bug! too many kiq readers\n");
 903                 goto failed_unlock;
 904         }
 905         amdgpu_ring_alloc(ring, 32);
 906         amdgpu_ring_emit_rreg(ring, reg, reg_val_offs);
 907         r = amdgpu_fence_emit_polling(ring, &seq, MAX_KIQ_REG_WAIT);
 908         if (r)
 909                 goto failed_undo;
 910
 911         amdgpu_ring_commit(ring);
 912         spin_unlock_irqrestore(&kiq->ring_lock, flags);
 913
 914         r = amdgpu_fence_wait_polling(ring, seq, MAX_KIQ_REG_WAIT);
 915
 916         /* don't wait anymore for gpu reset case because this way may
 917          * block gpu_recover() routine forever, e.g. this virt_kiq_rreg
 918          * is triggered in TTM and ttm_bo_lock_delayed_workqueue() will
 919          * never return if we keep waiting in virt_kiq_rreg, which cause
 920          * gpu_recover() hang there.
 921          *
 922          * also don't wait anymore for IRQ context
 923          * */
 924         if (r < 1 && (amdgpu_in_reset(adev) || in_interrupt()))
 925                 goto failed_kiq_read;
 926
 927         might_sleep();
 928         while (r < 1 && cnt++ < MAX_KIQ_REG_TRY) {
 929                 msleep(MAX_KIQ_REG_BAILOUT_INTERVAL);
 930                 r = amdgpu_fence_wait_polling(ring, seq, MAX_KIQ_REG_WAIT);
 931         }
 932
 933         if (cnt > MAX_KIQ_REG_TRY)
 934                 goto failed_kiq_read;
 935
 936         mb();
 937         value = adev->wb.wb[reg_val_offs];
 938         amdgpu_device_wb_free(adev, reg_val_offs);
 939         return value;
 940
 941 failed_undo:
 942         amdgpu_ring_undo(ring);
 943 failed_unlock:
 944         spin_unlock_irqrestore(&kiq->ring_lock, flags);
 945 failed_kiq_read:
 946         if (reg_val_offs)
 947                 amdgpu_device_wb_free(adev, reg_val_offs);
 948         dev_err(adev->dev, "failed to read reg:%x\n", reg);
 949         return ~0;
 950 }
 951
 952 void amdgpu_kiq_wreg(struct amdgpu_device *adev, uint32_t reg, uint32_t v)
 953 {
 954         signed long r, cnt = 0;
 955         unsigned long flags;
 956         uint32_t seq;
 957         struct amdgpu_kiq *kiq = &adev->gfx.kiq[0];
 958         struct amdgpu_ring *ring = &kiq->ring;
 959
 960         BUG_ON(!ring->funcs->emit_wreg);
 961
 962         if (amdgpu_device_skip_hw_access(adev))
 963                 return;
 964
 965         if (adev->mes.ring.sched.ready) {
 966                 amdgpu_mes_wreg(adev, reg, v);
 967                 return;
 968         }
 969
 970         spin_lock_irqsave(&kiq->ring_lock, flags);
 971         amdgpu_ring_alloc(ring, 32);
 972         amdgpu_ring_emit_wreg(ring, reg, v);
 973         r = amdgpu_fence_emit_polling(ring, &seq, MAX_KIQ_REG_WAIT);
 974         if (r)
 975                 goto failed_undo;
 976
 977         amdgpu_ring_commit(ring);
 978         spin_unlock_irqrestore(&kiq->ring_lock, flags);
 979
 980         r = amdgpu_fence_wait_polling(ring, seq, MAX_KIQ_REG_WAIT);
 981
 982         /* don't wait anymore for gpu reset case because this way may
 983          * block gpu_recover() routine forever, e.g. this virt_kiq_rreg
 984          * is triggered in TTM and ttm_bo_lock_delayed_workqueue() will
 985          * never return if we keep waiting in virt_kiq_rreg, which cause
 986          * gpu_recover() hang there.
 987          *
 988          * also don't wait anymore for IRQ context
 989          * */
 990         if (r < 1 && (amdgpu_in_reset(adev) || in_interrupt()))
 991                 goto failed_kiq_write;
 992
 993         might_sleep();
 994         while (r < 1 && cnt++ < MAX_KIQ_REG_TRY) {
 995
 996                 msleep(MAX_KIQ_REG_BAILOUT_INTERVAL);
 997                 r = amdgpu_fence_wait_polling(ring, seq, MAX_KIQ_REG_WAIT);
 998         }
 999
1000         if (cnt > MAX_KIQ_REG_TRY)
1001                 goto failed_kiq_write;
1002
1003         return;
1004
1005 failed_undo:
1006         amdgpu_ring_undo(ring);
1007         spin_unlock_irqrestore(&kiq->ring_lock, flags);
1008 failed_kiq_write:
1009         dev_err(adev->dev, "failed to write reg:%x\n", reg);
1010 }
1011
1012 int amdgpu_gfx_get_num_kcq(struct amdgpu_device *adev)
1013 {
1014         if (amdgpu_num_kcq == -1) {
1015                 return 8;
1016         } else if (amdgpu_num_kcq > 8 || amdgpu_num_kcq < 0) {
1017                 dev_warn(adev->dev, "set kernel compute queue number to 8 due to invalid parameter provided by user\n");
1018                 return 8;
1019         }
1020         return amdgpu_num_kcq;
1021 }
1022
1023 void amdgpu_gfx_cp_init_microcode(struct amdgpu_device *adev,
1024                                   uint32_t ucode_id)
1025 {
1026         const struct gfx_firmware_header_v1_0 *cp_hdr;
1027         const struct gfx_firmware_header_v2_0 *cp_hdr_v2_0;
1028         struct amdgpu_firmware_info *info = NULL;
1029         const struct firmware *ucode_fw;
1030         unsigned int fw_size;
1031
1032         switch (ucode_id) {
1033         case AMDGPU_UCODE_ID_CP_PFP:
1034                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1035                         adev->gfx.pfp_fw->data;
1036                 adev->gfx.pfp_fw_version =
1037                         le32_to_cpu(cp_hdr->header.ucode_version);
1038                 adev->gfx.pfp_feature_version =
1039                         le32_to_cpu(cp_hdr->ucode_feature_version);
1040                 ucode_fw = adev->gfx.pfp_fw;
1041                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes);
1042                 break;
1043         case AMDGPU_UCODE_ID_CP_RS64_PFP:
1044                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1045                         adev->gfx.pfp_fw->data;
1046                 adev->gfx.pfp_fw_version =
1047                         le32_to_cpu(cp_hdr_v2_0->header.ucode_version);
1048                 adev->gfx.pfp_feature_version =
1049                         le32_to_cpu(cp_hdr_v2_0->ucode_feature_version);
1050                 ucode_fw = adev->gfx.pfp_fw;
1051                 fw_size = le32_to_cpu(cp_hdr_v2_0->ucode_size_bytes);
1052                 break;
1053         case AMDGPU_UCODE_ID_CP_RS64_PFP_P0_STACK:
1054         case AMDGPU_UCODE_ID_CP_RS64_PFP_P1_STACK:
1055                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1056                         adev->gfx.pfp_fw->data;
1057                 ucode_fw = adev->gfx.pfp_fw;
1058                 fw_size = le32_to_cpu(cp_hdr_v2_0->data_size_bytes);
1059                 break;
1060         case AMDGPU_UCODE_ID_CP_ME:
1061                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1062                         adev->gfx.me_fw->data;
1063                 adev->gfx.me_fw_version =
1064                         le32_to_cpu(cp_hdr->header.ucode_version);
1065                 adev->gfx.me_feature_version =
1066                         le32_to_cpu(cp_hdr->ucode_feature_version);
1067                 ucode_fw = adev->gfx.me_fw;
1068                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes);
1069                 break;
1070         case AMDGPU_UCODE_ID_CP_RS64_ME:
1071                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1072                         adev->gfx.me_fw->data;
1073                 adev->gfx.me_fw_version =
1074                         le32_to_cpu(cp_hdr_v2_0->header.ucode_version);
1075                 adev->gfx.me_feature_version =
1076                         le32_to_cpu(cp_hdr_v2_0->ucode_feature_version);
1077                 ucode_fw = adev->gfx.me_fw;
1078                 fw_size = le32_to_cpu(cp_hdr_v2_0->ucode_size_bytes);
1079                 break;
1080         case AMDGPU_UCODE_ID_CP_RS64_ME_P0_STACK:
1081         case AMDGPU_UCODE_ID_CP_RS64_ME_P1_STACK:
1082                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1083                         adev->gfx.me_fw->data;
1084                 ucode_fw = adev->gfx.me_fw;
1085                 fw_size = le32_to_cpu(cp_hdr_v2_0->data_size_bytes);
1086                 break;
1087         case AMDGPU_UCODE_ID_CP_CE:
1088                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1089                         adev->gfx.ce_fw->data;
1090                 adev->gfx.ce_fw_version =
1091                         le32_to_cpu(cp_hdr->header.ucode_version);
1092                 adev->gfx.ce_feature_version =
1093                         le32_to_cpu(cp_hdr->ucode_feature_version);
1094                 ucode_fw = adev->gfx.ce_fw;
1095                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes);
1096                 break;
1097         case AMDGPU_UCODE_ID_CP_MEC1:
1098                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1099                         adev->gfx.mec_fw->data;
1100                 adev->gfx.mec_fw_version =
1101                         le32_to_cpu(cp_hdr->header.ucode_version);
1102                 adev->gfx.mec_feature_version =
1103                         le32_to_cpu(cp_hdr->ucode_feature_version);
1104                 ucode_fw = adev->gfx.mec_fw;
1105                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes) -
1106                           le32_to_cpu(cp_hdr->jt_size) * 4;
1107                 break;
1108         case AMDGPU_UCODE_ID_CP_MEC1_JT:
1109                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1110                         adev->gfx.mec_fw->data;
1111                 ucode_fw = adev->gfx.mec_fw;
1112                 fw_size = le32_to_cpu(cp_hdr->jt_size) * 4;
1113                 break;
1114         case AMDGPU_UCODE_ID_CP_MEC2:
1115                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1116                         adev->gfx.mec2_fw->data;
1117                 adev->gfx.mec2_fw_version =
1118                         le32_to_cpu(cp_hdr->header.ucode_version);
1119                 adev->gfx.mec2_feature_version =
1120                         le32_to_cpu(cp_hdr->ucode_feature_version);
1121                 ucode_fw = adev->gfx.mec2_fw;
1122                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes) -
1123                           le32_to_cpu(cp_hdr->jt_size) * 4;
1124                 break;
1125         case AMDGPU_UCODE_ID_CP_MEC2_JT:
1126                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1127                         adev->gfx.mec2_fw->data;
1128                 ucode_fw = adev->gfx.mec2_fw;
1129                 fw_size = le32_to_cpu(cp_hdr->jt_size) * 4;
1130                 break;
1131         case AMDGPU_UCODE_ID_CP_RS64_MEC:
1132                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1133                         adev->gfx.mec_fw->data;
1134                 adev->gfx.mec_fw_version =
1135                         le32_to_cpu(cp_hdr_v2_0->header.ucode_version);
1136                 adev->gfx.mec_feature_version =
1137                         le32_to_cpu(cp_hdr_v2_0->ucode_feature_version);
1138                 ucode_fw = adev->gfx.mec_fw;
1139                 fw_size = le32_to_cpu(cp_hdr_v2_0->ucode_size_bytes);
1140                 break;
1141         case AMDGPU_UCODE_ID_CP_RS64_MEC_P0_STACK:
1142         case AMDGPU_UCODE_ID_CP_RS64_MEC_P1_STACK:
1143         case AMDGPU_UCODE_ID_CP_RS64_MEC_P2_STACK:
1144         case AMDGPU_UCODE_ID_CP_RS64_MEC_P3_STACK:
1145                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1146                         adev->gfx.mec_fw->data;
1147                 ucode_fw = adev->gfx.mec_fw;
1148                 fw_size = le32_to_cpu(cp_hdr_v2_0->data_size_bytes);
1149                 break;
1150         default:
1151                 break;
1152         }
1153
1154         if (adev->firmware.load_type == AMDGPU_FW_LOAD_PSP) {
1155                 info = &adev->firmware.ucode[ucode_id];
1156                 info->ucode_id = ucode_id;
1157                 info->fw = ucode_fw;
1158                 adev->firmware.fw_size += ALIGN(fw_size, PAGE_SIZE);
1159         }
1160 }
1161
1162 bool amdgpu_gfx_is_master_xcc(struct amdgpu_device *adev, int xcc_id)
1163 {
1164         return !(xcc_id % (adev->gfx.num_xcc_per_xcp ?
1165                         adev->gfx.num_xcc_per_xcp : 1));
1166 }
1167
1168 static ssize_t amdgpu_gfx_get_current_compute_partition(struct device *dev,
1169                                                 struct device_attribute *addr,
1170                                                 char *buf)
1171 {
1172         struct drm_device *ddev = dev_get_drvdata(dev);
1173         struct amdgpu_device *adev = drm_to_adev(ddev);
1174         enum amdgpu_gfx_partition mode;
1175         char *partition_mode;
1176
1177         mode = adev->gfx.funcs->query_partition_mode(adev);
1178
1179         switch (mode) {
1180         case AMDGPU_SPX_PARTITION_MODE:
1181                 partition_mode = "SPX";
1182                 break;
1183         case AMDGPU_DPX_PARTITION_MODE:
1184                 partition_mode = "DPX";
1185                 break;
1186         case AMDGPU_TPX_PARTITION_MODE:
1187                 partition_mode = "TPX";
1188                 break;
1189         case AMDGPU_QPX_PARTITION_MODE:
1190                 partition_mode = "QPX";
1191                 break;
1192         case AMDGPU_CPX_PARTITION_MODE:
1193                 partition_mode = "CPX";
1194                 break;
1195         default:
1196                 partition_mode = "UNKNOWN";
1197                 break;
1198         }
1199
1200         return sysfs_emit(buf, "%s\n", partition_mode);
1201 }
1202
1203 static ssize_t amdgpu_gfx_set_compute_partition(struct device *dev,
1204                                                 struct device_attribute *addr,
1205                                                 const char *buf, size_t count)
1206 {
1207         struct drm_device *ddev = dev_get_drvdata(dev);
1208         struct amdgpu_device *adev = drm_to_adev(ddev);
1209         enum amdgpu_gfx_partition mode;
1210         int ret = 0, num_xcc;
1211
1212         num_xcc = NUM_XCC(adev->gfx.xcc_mask);
1213         if (num_xcc % 2 != 0)
1214                 return -EINVAL;
1215
1216         if (!strncasecmp("SPX", buf, strlen("SPX"))) {
1217                 mode = AMDGPU_SPX_PARTITION_MODE;
1218         } else if (!strncasecmp("DPX", buf, strlen("DPX"))) {
1219                 if (num_xcc != 4 || num_xcc != 8)
1220                         return -EINVAL;
1221                 mode = AMDGPU_DPX_PARTITION_MODE;
1222         } else if (!strncasecmp("TPX", buf, strlen("TPX"))) {
1223                 if (num_xcc != 6)
1224                         return -EINVAL;
1225                 mode = AMDGPU_TPX_PARTITION_MODE;
1226         } else if (!strncasecmp("QPX", buf, strlen("QPX"))) {
1227                 if (num_xcc != 8)
1228                         return -EINVAL;
1229                 mode = AMDGPU_QPX_PARTITION_MODE;
1230         } else if (!strncasecmp("CPX", buf, strlen("CPX"))) {
1231                 mode = AMDGPU_CPX_PARTITION_MODE;
1232         } else {
1233                 return -EINVAL;
1234         }
1235
1236         if (!adev->kfd.init_complete)
1237                 return -EPERM;
1238
1239         mutex_lock(&adev->gfx.partition_mutex);
1240
1241         if (mode == adev->gfx.funcs->query_partition_mode(adev))
1242                 goto out;
1243
1244         ret = amdgpu_amdkfd_check_and_lock_kfd(adev);
1245         if (ret)
1246                 goto out;
1247
1248         amdgpu_amdkfd_device_fini_sw(adev);
1249
1250         adev->gfx.funcs->switch_partition_mode(adev, mode);
1251
1252         amdgpu_amdkfd_device_probe(adev);
1253         amdgpu_amdkfd_device_init(adev);
1254         /* If KFD init failed, return failure */
1255         if (!adev->kfd.init_complete)
1256                 ret = -EIO;
1257
1258         amdgpu_amdkfd_unlock_kfd(adev);
1259 out:
1260         mutex_unlock(&adev->gfx.partition_mutex);
1261
1262         if (ret)
1263                 return ret;
1264
1265         return count;
1266 }
1267
1268 static ssize_t amdgpu_gfx_get_available_compute_partition(struct device *dev,
1269                                                 struct device_attribute *addr,
1270                                                 char *buf)
1271 {
1272         struct drm_device *ddev = dev_get_drvdata(dev);
1273         struct amdgpu_device *adev = drm_to_adev(ddev);
1274         char *supported_partition;
1275
1276         /* TBD */
1277         switch (NUM_XCC(adev->gfx.xcc_mask)) {
1278         case 8:
1279                 supported_partition = "SPX, DPX, QPX, CPX";
1280                 break;
1281         case 6:
1282                 supported_partition = "SPX, TPX, CPX";
1283                 break;
1284         case 4:
1285                 supported_partition = "SPX, DPX, CPX";
1286                 break;
1287         /* this seems only existing in emulation phase */
1288         case 2:
1289                 supported_partition = "SPX, CPX";
1290                 break;
1291         default:
1292                 supported_partition = "Not supported";
1293                 break;
1294         }
1295
1296         return sysfs_emit(buf, "%s\n", supported_partition);
1297 }
1298
1299 static DEVICE_ATTR(current_compute_partition, S_IRUGO | S_IWUSR,
1300                    amdgpu_gfx_get_current_compute_partition,
1301                    amdgpu_gfx_set_compute_partition);
1302
1303 static DEVICE_ATTR(available_compute_partition, S_IRUGO,
1304                    amdgpu_gfx_get_available_compute_partition, NULL);
1305
1306 int amdgpu_gfx_sysfs_init(struct amdgpu_device *adev)
1307 {
1308         int r;
1309
1310         r = device_create_file(adev->dev, &dev_attr_current_compute_partition);
1311         if (r)
1312                 return r;
1313
1314         r = device_create_file(adev->dev, &dev_attr_available_compute_partition);
1315         if (r)
1316                 return r;
1317
1318         return 0;
1319 }