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_xcd = (adev->gfx.num_xcd > 1) ? adev->gfx.num_xcd : 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_xcd; 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_xcd; 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_xcd; 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->doorbell_index = adev->doorbell_index.kiq;
 317         ring->xcc_id = xcc_id;
 318         ring->vm_hub = AMDGPU_GFXHUB(xcc_id);
 319         if (xcc_id >= 1)
 320                 ring->doorbell_index = adev->doorbell_index.xcc1_kiq_start +
 321                                         xcc_id - 1;
 322         else
 323                 ring->doorbell_index = adev->doorbell_index.kiq;
 324
 325         r = amdgpu_gfx_kiq_acquire(adev, ring, xcc_id);
 326         if (r)
 327                 return r;
 328
 329         ring->eop_gpu_addr = kiq->eop_gpu_addr;
 330         ring->no_scheduler = true;
 331         sprintf(ring->name, "kiq_%d.%d.%d.%d", xcc_id, ring->me, ring->pipe, ring->queue);
 332         r = amdgpu_ring_init(adev, ring, 1024, irq, AMDGPU_CP_KIQ_IRQ_DRIVER0,
 333                              AMDGPU_RING_PRIO_DEFAULT, NULL);
 334         if (r)
 335                 dev_warn(adev->dev, "(%d) failed to init kiq ring\n", r);
 336
 337         return r;
 338 }
 339
 340 void amdgpu_gfx_kiq_free_ring(struct amdgpu_ring *ring)
 341 {
 342         amdgpu_ring_fini(ring);
 343 }
 344
 345 void amdgpu_gfx_kiq_fini(struct amdgpu_device *adev, int xcc_id)
 346 {
 347         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 348
 349         amdgpu_bo_free_kernel(&kiq->eop_obj, &kiq->eop_gpu_addr, NULL);
 350 }
 351
 352 int amdgpu_gfx_kiq_init(struct amdgpu_device *adev,
 353                         unsigned hpd_size, int xcc_id)
 354 {
 355         int r;
 356         u32 *hpd;
 357         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 358
 359         r = amdgpu_bo_create_kernel(adev, hpd_size, PAGE_SIZE,
 360                                     AMDGPU_GEM_DOMAIN_GTT, &kiq->eop_obj,
 361                                     &kiq->eop_gpu_addr, (void **)&hpd);
 362         if (r) {
 363                 dev_warn(adev->dev, "failed to create KIQ bo (%d).\n", r);
 364                 return r;
 365         }
 366
 367         memset(hpd, 0, hpd_size);
 368
 369         r = amdgpu_bo_reserve(kiq->eop_obj, true);
 370         if (unlikely(r != 0))
 371                 dev_warn(adev->dev, "(%d) reserve kiq eop bo failed\n", r);
 372         amdgpu_bo_kunmap(kiq->eop_obj);
 373         amdgpu_bo_unreserve(kiq->eop_obj);
 374
 375         return 0;
 376 }
 377
 378 /* create MQD for each compute/gfx queue */
 379 int amdgpu_gfx_mqd_sw_init(struct amdgpu_device *adev,
 380                            unsigned mqd_size, int xcc_id)
 381 {
 382         int r, i, j;
 383         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 384         struct amdgpu_ring *ring = &kiq->ring;
 385         u32 domain = AMDGPU_GEM_DOMAIN_GTT;
 386
 387         /* Only enable on gfx10 and 11 for now to avoid changing behavior on older chips */
 388         if (adev->ip_versions[GC_HWIP][0] >= IP_VERSION(10, 0, 0))
 389                 domain |= AMDGPU_GEM_DOMAIN_VRAM;
 390
 391         /* create MQD for KIQ */
 392         if (!adev->enable_mes_kiq && !ring->mqd_obj) {
 393                 /* originaly the KIQ MQD is put in GTT domain, but for SRIOV VRAM domain is a must
 394                  * otherwise hypervisor trigger SAVE_VF fail after driver unloaded which mean MQD
 395                  * deallocated and gart_unbind, to strict diverage we decide to use VRAM domain for
 396                  * KIQ MQD no matter SRIOV or Bare-metal
 397                  */
 398                 r = amdgpu_bo_create_kernel(adev, mqd_size, PAGE_SIZE,
 399                                             AMDGPU_GEM_DOMAIN_VRAM |
 400                                             AMDGPU_GEM_DOMAIN_GTT,
 401                                             &ring->mqd_obj,
 402                                             &ring->mqd_gpu_addr,
 403                                             &ring->mqd_ptr);
 404                 if (r) {
 405                         dev_warn(adev->dev, "failed to create ring mqd ob (%d)", r);
 406                         return r;
 407                 }
 408
 409                 /* prepare MQD backup */
 410                 kiq->mqd_backup = kmalloc(mqd_size, GFP_KERNEL);
 411                 if (!kiq->mqd_backup)
 412                                 dev_warn(adev->dev, "no memory to create MQD backup for ring %s\n", ring->name);
 413         }
 414
 415         if (adev->asic_type >= CHIP_NAVI10 && amdgpu_async_gfx_ring) {
 416                 /* create MQD for each KGQ */
 417                 for (i = 0; i < adev->gfx.num_gfx_rings; i++) {
 418                         ring = &adev->gfx.gfx_ring[i];
 419                         if (!ring->mqd_obj) {
 420                                 r = amdgpu_bo_create_kernel(adev, mqd_size, PAGE_SIZE,
 421                                                             domain, &ring->mqd_obj,
 422                                                             &ring->mqd_gpu_addr, &ring->mqd_ptr);
 423                                 if (r) {
 424                                         dev_warn(adev->dev, "failed to create ring mqd bo (%d)", r);
 425                                         return r;
 426                                 }
 427
 428                                 ring->mqd_size = mqd_size;
 429                                 /* prepare MQD backup */
 430                                 adev->gfx.me.mqd_backup[i] = kmalloc(mqd_size, GFP_KERNEL);
 431                                 if (!adev->gfx.me.mqd_backup[i])
 432                                         dev_warn(adev->dev, "no memory to create MQD backup for ring %s\n", ring->name);
 433                         }
 434                 }
 435         }
 436
 437         /* create MQD for each KCQ */
 438         for (i = 0; i < adev->gfx.num_compute_rings; i++) {
 439                 j = i + xcc_id * adev->gfx.num_compute_rings;
 440                 ring = &adev->gfx.compute_ring[j];
 441                 if (!ring->mqd_obj) {
 442                         r = amdgpu_bo_create_kernel(adev, mqd_size, PAGE_SIZE,
 443                                                     domain, &ring->mqd_obj,
 444                                                     &ring->mqd_gpu_addr, &ring->mqd_ptr);
 445                         if (r) {
 446                                 dev_warn(adev->dev, "failed to create ring mqd bo (%d)", r);
 447                                 return r;
 448                         }
 449
 450                         ring->mqd_size = mqd_size;
 451                         /* prepare MQD backup */
 452                         adev->gfx.mec.mqd_backup[j + xcc_id * adev->gfx.num_compute_rings] = kmalloc(mqd_size, GFP_KERNEL);
 453                         if (!adev->gfx.mec.mqd_backup[j + xcc_id * adev->gfx.num_compute_rings])
 454                                 dev_warn(adev->dev, "no memory to create MQD backup for ring %s\n", ring->name);
 455                 }
 456         }
 457
 458         return 0;
 459 }
 460
 461 void amdgpu_gfx_mqd_sw_fini(struct amdgpu_device *adev, int xcc_id)
 462 {
 463         struct amdgpu_ring *ring = NULL;
 464         int i, j;
 465         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 466
 467         if (adev->asic_type >= CHIP_NAVI10 && amdgpu_async_gfx_ring) {
 468                 for (i = 0; i < adev->gfx.num_gfx_rings; i++) {
 469                         ring = &adev->gfx.gfx_ring[i];
 470                         kfree(adev->gfx.me.mqd_backup[i]);
 471                         amdgpu_bo_free_kernel(&ring->mqd_obj,
 472                                               &ring->mqd_gpu_addr,
 473                                               &ring->mqd_ptr);
 474                 }
 475         }
 476
 477         for (i = 0; i < adev->gfx.num_compute_rings; i++) {
 478                 j = i + xcc_id * adev->gfx.num_compute_rings;
 479                 ring = &adev->gfx.compute_ring[j];
 480                 kfree(adev->gfx.mec.mqd_backup[j]);
 481                 amdgpu_bo_free_kernel(&ring->mqd_obj,
 482                                       &ring->mqd_gpu_addr,
 483                                       &ring->mqd_ptr);
 484         }
 485
 486         ring = &kiq->ring;
 487         kfree(kiq->mqd_backup);
 488         kfree(adev->gfx.mec.mqd_backup[AMDGPU_MAX_COMPUTE_RINGS]);
 489         amdgpu_bo_free_kernel(&ring->mqd_obj,
 490                               &ring->mqd_gpu_addr,
 491                               &ring->mqd_ptr);
 492 }
 493
 494 int amdgpu_gfx_disable_kcq(struct amdgpu_device *adev, int xcc_id)
 495 {
 496         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 497         struct amdgpu_ring *kiq_ring = &kiq->ring;
 498         int i, r = 0;
 499         int j;
 500
 501         if (!kiq->pmf || !kiq->pmf->kiq_unmap_queues)
 502                 return -EINVAL;
 503
 504         spin_lock(&kiq->ring_lock);
 505         if (amdgpu_ring_alloc(kiq_ring, kiq->pmf->unmap_queues_size *
 506                                         adev->gfx.num_compute_rings)) {
 507                 spin_unlock(&kiq->ring_lock);
 508                 return -ENOMEM;
 509         }
 510
 511         for (i = 0; i < adev->gfx.num_compute_rings; i++) {
 512                 j = i + xcc_id * adev->gfx.num_compute_rings;
 513                 kiq->pmf->kiq_unmap_queues(kiq_ring,
 514                                            &adev->gfx.compute_ring[i],
 515                                            RESET_QUEUES, 0, 0);
 516         }
 517
 518         if (kiq_ring->sched.ready && !adev->job_hang)
 519                 r = amdgpu_ring_test_helper(kiq_ring);
 520         spin_unlock(&kiq->ring_lock);
 521
 522         return r;
 523 }
 524
 525 int amdgpu_gfx_disable_kgq(struct amdgpu_device *adev, int xcc_id)
 526 {
 527         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 528         struct amdgpu_ring *kiq_ring = &kiq->ring;
 529         int i, r = 0;
 530         int j;
 531
 532         if (!kiq->pmf || !kiq->pmf->kiq_unmap_queues)
 533                 return -EINVAL;
 534
 535         spin_lock(&kiq->ring_lock);
 536         if (amdgpu_gfx_is_master_xcc(adev, xcc_id)) {
 537                 if (amdgpu_ring_alloc(kiq_ring, kiq->pmf->unmap_queues_size *
 538                                                 adev->gfx.num_gfx_rings)) {
 539                         spin_unlock(&kiq->ring_lock);
 540                         return -ENOMEM;
 541                 }
 542
 543                 for (i = 0; i < adev->gfx.num_gfx_rings; i++) {
 544                         j = i + xcc_id * adev->gfx.num_gfx_rings;
 545                         kiq->pmf->kiq_unmap_queues(kiq_ring,
 546                                                    &adev->gfx.gfx_ring[i],
 547                                                    PREEMPT_QUEUES, 0, 0);
 548                 }
 549         }
 550
 551         if (adev->gfx.kiq[0].ring.sched.ready && !adev->job_hang)
 552                 r = amdgpu_ring_test_helper(kiq_ring);
 553         spin_unlock(&kiq->ring_lock);
 554
 555         return r;
 556 }
 557
 558 int amdgpu_queue_mask_bit_to_set_resource_bit(struct amdgpu_device *adev,
 559                                         int queue_bit)
 560 {
 561         int mec, pipe, queue;
 562         int set_resource_bit = 0;
 563
 564         amdgpu_queue_mask_bit_to_mec_queue(adev, queue_bit, &mec, &pipe, &queue);
 565
 566         set_resource_bit = mec * 4 * 8 + pipe * 8 + queue;
 567
 568         return set_resource_bit;
 569 }
 570
 571 int amdgpu_gfx_enable_kcq(struct amdgpu_device *adev, int xcc_id)
 572 {
 573         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 574         struct amdgpu_ring *kiq_ring = &kiq->ring;
 575         uint64_t queue_mask = 0;
 576         int r, i, j;
 577
 578         if (!kiq->pmf || !kiq->pmf->kiq_map_queues || !kiq->pmf->kiq_set_resources)
 579                 return -EINVAL;
 580
 581         for (i = 0; i < AMDGPU_MAX_COMPUTE_QUEUES; ++i) {
 582                 if (!test_bit(i, adev->gfx.mec_bitmap[xcc_id].queue_bitmap))
 583                         continue;
 584
 585                 /* This situation may be hit in the future if a new HW
 586                  * generation exposes more than 64 queues. If so, the
 587                  * definition of queue_mask needs updating */
 588                 if (WARN_ON(i > (sizeof(queue_mask)*8))) {
 589                         DRM_ERROR("Invalid KCQ enabled: %d\n", i);
 590                         break;
 591                 }
 592
 593                 queue_mask |= (1ull << amdgpu_queue_mask_bit_to_set_resource_bit(adev, i));
 594         }
 595
 596         DRM_INFO("kiq ring mec %d pipe %d q %d\n", kiq_ring->me, kiq_ring->pipe,
 597                                                         kiq_ring->queue);
 598         spin_lock(&kiq->ring_lock);
 599         r = amdgpu_ring_alloc(kiq_ring, kiq->pmf->map_queues_size *
 600                                         adev->gfx.num_compute_rings +
 601                                         kiq->pmf->set_resources_size);
 602         if (r) {
 603                 DRM_ERROR("Failed to lock KIQ (%d).\n", r);
 604                 spin_unlock(&kiq->ring_lock);
 605                 return r;
 606         }
 607
 608         if (adev->enable_mes)
 609                 queue_mask = ~0ULL;
 610
 611         kiq->pmf->kiq_set_resources(kiq_ring, queue_mask);
 612         for (i = 0; i < adev->gfx.num_compute_rings; i++) {
 613                 j = i + xcc_id * adev->gfx.num_compute_rings;
 614                         kiq->pmf->kiq_map_queues(kiq_ring,
 615                                                  &adev->gfx.compute_ring[j]);
 616         }
 617
 618         r = amdgpu_ring_test_helper(kiq_ring);
 619         spin_unlock(&kiq->ring_lock);
 620         if (r)
 621                 DRM_ERROR("KCQ enable failed\n");
 622
 623         return r;
 624 }
 625
 626 int amdgpu_gfx_enable_kgq(struct amdgpu_device *adev, int xcc_id)
 627 {
 628         struct amdgpu_kiq *kiq = &adev->gfx.kiq[xcc_id];
 629         struct amdgpu_ring *kiq_ring = &kiq->ring;
 630         int r, i, j;
 631
 632         if (!kiq->pmf || !kiq->pmf->kiq_map_queues)
 633                 return -EINVAL;
 634
 635         spin_lock(&kiq->ring_lock);
 636         /* No need to map kcq on the slave */
 637         if (amdgpu_gfx_is_master_xcc(adev, xcc_id)) {
 638                 r = amdgpu_ring_alloc(kiq_ring, kiq->pmf->map_queues_size *
 639                                                 adev->gfx.num_gfx_rings);
 640                 if (r) {
 641                         DRM_ERROR("Failed to lock KIQ (%d).\n", r);
 642                         spin_unlock(&kiq->ring_lock);
 643                         return r;
 644                 }
 645
 646                 for (i = 0; i < adev->gfx.num_gfx_rings; i++) {
 647                         j = i + xcc_id * adev->gfx.num_gfx_rings;
 648                         kiq->pmf->kiq_map_queues(kiq_ring,
 649                                                  &adev->gfx.gfx_ring[i]);
 650                 }
 651         }
 652
 653         r = amdgpu_ring_test_helper(kiq_ring);
 654         spin_unlock(&kiq->ring_lock);
 655         if (r)
 656                 DRM_ERROR("KCQ enable failed\n");
 657
 658         return r;
 659 }
 660
 661 /* amdgpu_gfx_off_ctrl - Handle gfx off feature enable/disable
 662  *
 663  * @adev: amdgpu_device pointer
 664  * @bool enable true: enable gfx off feature, false: disable gfx off feature
 665  *
 666  * 1. gfx off feature will be enabled by gfx ip after gfx cg gp enabled.
 667  * 2. other client can send request to disable gfx off feature, the request should be honored.
 668  * 3. other client can cancel their request of disable gfx off feature
 669  * 4. other client should not send request to enable gfx off feature before disable gfx off feature.
 670  */
 671
 672 void amdgpu_gfx_off_ctrl(struct amdgpu_device *adev, bool enable)
 673 {
 674         unsigned long delay = GFX_OFF_DELAY_ENABLE;
 675
 676         if (!(adev->pm.pp_feature & PP_GFXOFF_MASK))
 677                 return;
 678
 679         mutex_lock(&adev->gfx.gfx_off_mutex);
 680
 681         if (enable) {
 682                 /* If the count is already 0, it means there's an imbalance bug somewhere.
 683                  * Note that the bug may be in a different caller than the one which triggers the
 684                  * WARN_ON_ONCE.
 685                  */
 686                 if (WARN_ON_ONCE(adev->gfx.gfx_off_req_count == 0))
 687                         goto unlock;
 688
 689                 adev->gfx.gfx_off_req_count--;
 690
 691                 if (adev->gfx.gfx_off_req_count == 0 &&
 692                     !adev->gfx.gfx_off_state) {
 693                         /* If going to s2idle, no need to wait */
 694                         if (adev->in_s0ix) {
 695                                 if (!amdgpu_dpm_set_powergating_by_smu(adev,
 696                                                 AMD_IP_BLOCK_TYPE_GFX, true))
 697                                         adev->gfx.gfx_off_state = true;
 698                         } else {
 699                                 schedule_delayed_work(&adev->gfx.gfx_off_delay_work,
 700                                               delay);
 701                         }
 702                 }
 703         } else {
 704                 if (adev->gfx.gfx_off_req_count == 0) {
 705                         cancel_delayed_work_sync(&adev->gfx.gfx_off_delay_work);
 706
 707                         if (adev->gfx.gfx_off_state &&
 708                             !amdgpu_dpm_set_powergating_by_smu(adev, AMD_IP_BLOCK_TYPE_GFX, false)) {
 709                                 adev->gfx.gfx_off_state = false;
 710
 711                                 if (adev->gfx.funcs->init_spm_golden) {
 712                                         dev_dbg(adev->dev,
 713                                                 "GFXOFF is disabled, re-init SPM golden settings\n");
 714                                         amdgpu_gfx_init_spm_golden(adev);
 715                                 }
 716                         }
 717                 }
 718
 719                 adev->gfx.gfx_off_req_count++;
 720         }
 721
 722 unlock:
 723         mutex_unlock(&adev->gfx.gfx_off_mutex);
 724 }
 725
 726 int amdgpu_set_gfx_off_residency(struct amdgpu_device *adev, bool value)
 727 {
 728         int r = 0;
 729
 730         mutex_lock(&adev->gfx.gfx_off_mutex);
 731
 732         r = amdgpu_dpm_set_residency_gfxoff(adev, value);
 733
 734         mutex_unlock(&adev->gfx.gfx_off_mutex);
 735
 736         return r;
 737 }
 738
 739 int amdgpu_get_gfx_off_residency(struct amdgpu_device *adev, u32 *value)
 740 {
 741         int r = 0;
 742
 743         mutex_lock(&adev->gfx.gfx_off_mutex);
 744
 745         r = amdgpu_dpm_get_residency_gfxoff(adev, value);
 746
 747         mutex_unlock(&adev->gfx.gfx_off_mutex);
 748
 749         return r;
 750 }
 751
 752 int amdgpu_get_gfx_off_entrycount(struct amdgpu_device *adev, u64 *value)
 753 {
 754         int r = 0;
 755
 756         mutex_lock(&adev->gfx.gfx_off_mutex);
 757
 758         r = amdgpu_dpm_get_entrycount_gfxoff(adev, value);
 759
 760         mutex_unlock(&adev->gfx.gfx_off_mutex);
 761
 762         return r;
 763 }
 764
 765 int amdgpu_get_gfx_off_status(struct amdgpu_device *adev, uint32_t *value)
 766 {
 767
 768         int r = 0;
 769
 770         mutex_lock(&adev->gfx.gfx_off_mutex);
 771
 772         r = amdgpu_dpm_get_status_gfxoff(adev, value);
 773
 774         mutex_unlock(&adev->gfx.gfx_off_mutex);
 775
 776         return r;
 777 }
 778
 779 int amdgpu_gfx_ras_late_init(struct amdgpu_device *adev, struct ras_common_if *ras_block)
 780 {
 781         int r;
 782
 783         if (amdgpu_ras_is_supported(adev, ras_block->block)) {
 784                 if (!amdgpu_persistent_edc_harvesting_supported(adev))
 785                         amdgpu_ras_reset_error_status(adev, AMDGPU_RAS_BLOCK__GFX);
 786
 787                 r = amdgpu_ras_block_late_init(adev, ras_block);
 788                 if (r)
 789                         return r;
 790
 791                 if (adev->gfx.cp_ecc_error_irq.funcs) {
 792                         r = amdgpu_irq_get(adev, &adev->gfx.cp_ecc_error_irq, 0);
 793                         if (r)
 794                                 goto late_fini;
 795                 }
 796         } else {
 797                 amdgpu_ras_feature_enable_on_boot(adev, ras_block, 0);
 798         }
 799
 800         return 0;
 801 late_fini:
 802         amdgpu_ras_block_late_fini(adev, ras_block);
 803         return r;
 804 }
 805
 806 int amdgpu_gfx_ras_sw_init(struct amdgpu_device *adev)
 807 {
 808         int err = 0;
 809         struct amdgpu_gfx_ras *ras = NULL;
 810
 811         /* adev->gfx.ras is NULL, which means gfx does not
 812          * support ras function, then do nothing here.
 813          */
 814         if (!adev->gfx.ras)
 815                 return 0;
 816
 817         ras = adev->gfx.ras;
 818
 819         err = amdgpu_ras_register_ras_block(adev, &ras->ras_block);
 820         if (err) {
 821                 dev_err(adev->dev, "Failed to register gfx ras block!\n");
 822                 return err;
 823         }
 824
 825         strcpy(ras->ras_block.ras_comm.name, "gfx");
 826         ras->ras_block.ras_comm.block = AMDGPU_RAS_BLOCK__GFX;
 827         ras->ras_block.ras_comm.type = AMDGPU_RAS_ERROR__MULTI_UNCORRECTABLE;
 828         adev->gfx.ras_if = &ras->ras_block.ras_comm;
 829
 830         /* If not define special ras_late_init function, use gfx default ras_late_init */
 831         if (!ras->ras_block.ras_late_init)
 832                 ras->ras_block.ras_late_init = amdgpu_gfx_ras_late_init;
 833
 834         /* If not defined special ras_cb function, use default ras_cb */
 835         if (!ras->ras_block.ras_cb)
 836                 ras->ras_block.ras_cb = amdgpu_gfx_process_ras_data_cb;
 837
 838         return 0;
 839 }
 840
 841 int amdgpu_gfx_poison_consumption_handler(struct amdgpu_device *adev,
 842                                                 struct amdgpu_iv_entry *entry)
 843 {
 844         if (adev->gfx.ras && adev->gfx.ras->poison_consumption_handler)
 845                 return adev->gfx.ras->poison_consumption_handler(adev, entry);
 846
 847         return 0;
 848 }
 849
 850 int amdgpu_gfx_process_ras_data_cb(struct amdgpu_device *adev,
 851                 void *err_data,
 852                 struct amdgpu_iv_entry *entry)
 853 {
 854         /* TODO ue will trigger an interrupt.
 855          *
 856          * When “Full RAS” is enabled, the per-IP interrupt sources should
 857          * be disabled and the driver should only look for the aggregated
 858          * interrupt via sync flood
 859          */
 860         if (!amdgpu_ras_is_supported(adev, AMDGPU_RAS_BLOCK__GFX)) {
 861                 kgd2kfd_set_sram_ecc_flag(adev->kfd.dev);
 862                 if (adev->gfx.ras && adev->gfx.ras->ras_block.hw_ops &&
 863                     adev->gfx.ras->ras_block.hw_ops->query_ras_error_count)
 864                         adev->gfx.ras->ras_block.hw_ops->query_ras_error_count(adev, err_data);
 865                 amdgpu_ras_reset_gpu(adev);
 866         }
 867         return AMDGPU_RAS_SUCCESS;
 868 }
 869
 870 int amdgpu_gfx_cp_ecc_error_irq(struct amdgpu_device *adev,
 871                                   struct amdgpu_irq_src *source,
 872                                   struct amdgpu_iv_entry *entry)
 873 {
 874         struct ras_common_if *ras_if = adev->gfx.ras_if;
 875         struct ras_dispatch_if ih_data = {
 876                 .entry = entry,
 877         };
 878
 879         if (!ras_if)
 880                 return 0;
 881
 882         ih_data.head = *ras_if;
 883
 884         DRM_ERROR("CP ECC ERROR IRQ\n");
 885         amdgpu_ras_interrupt_dispatch(adev, &ih_data);
 886         return 0;
 887 }
 888
 889 uint32_t amdgpu_kiq_rreg(struct amdgpu_device *adev, uint32_t reg)
 890 {
 891         signed long r, cnt = 0;
 892         unsigned long flags;
 893         uint32_t seq, reg_val_offs = 0, value = 0;
 894         struct amdgpu_kiq *kiq = &adev->gfx.kiq[0];
 895         struct amdgpu_ring *ring = &kiq->ring;
 896
 897         if (amdgpu_device_skip_hw_access(adev))
 898                 return 0;
 899
 900         if (adev->mes.ring.sched.ready)
 901                 return amdgpu_mes_rreg(adev, reg);
 902
 903         BUG_ON(!ring->funcs->emit_rreg);
 904
 905         spin_lock_irqsave(&kiq->ring_lock, flags);
 906         if (amdgpu_device_wb_get(adev, &reg_val_offs)) {
 907                 pr_err("critical bug! too many kiq readers\n");
 908                 goto failed_unlock;
 909         }
 910         amdgpu_ring_alloc(ring, 32);
 911         amdgpu_ring_emit_rreg(ring, reg, reg_val_offs);
 912         r = amdgpu_fence_emit_polling(ring, &seq, MAX_KIQ_REG_WAIT);
 913         if (r)
 914                 goto failed_undo;
 915
 916         amdgpu_ring_commit(ring);
 917         spin_unlock_irqrestore(&kiq->ring_lock, flags);
 918
 919         r = amdgpu_fence_wait_polling(ring, seq, MAX_KIQ_REG_WAIT);
 920
 921         /* don't wait anymore for gpu reset case because this way may
 922          * block gpu_recover() routine forever, e.g. this virt_kiq_rreg
 923          * is triggered in TTM and ttm_bo_lock_delayed_workqueue() will
 924          * never return if we keep waiting in virt_kiq_rreg, which cause
 925          * gpu_recover() hang there.
 926          *
 927          * also don't wait anymore for IRQ context
 928          * */
 929         if (r < 1 && (amdgpu_in_reset(adev) || in_interrupt()))
 930                 goto failed_kiq_read;
 931
 932         might_sleep();
 933         while (r < 1 && cnt++ < MAX_KIQ_REG_TRY) {
 934                 msleep(MAX_KIQ_REG_BAILOUT_INTERVAL);
 935                 r = amdgpu_fence_wait_polling(ring, seq, MAX_KIQ_REG_WAIT);
 936         }
 937
 938         if (cnt > MAX_KIQ_REG_TRY)
 939                 goto failed_kiq_read;
 940
 941         mb();
 942         value = adev->wb.wb[reg_val_offs];
 943         amdgpu_device_wb_free(adev, reg_val_offs);
 944         return value;
 945
 946 failed_undo:
 947         amdgpu_ring_undo(ring);
 948 failed_unlock:
 949         spin_unlock_irqrestore(&kiq->ring_lock, flags);
 950 failed_kiq_read:
 951         if (reg_val_offs)
 952                 amdgpu_device_wb_free(adev, reg_val_offs);
 953         dev_err(adev->dev, "failed to read reg:%x\n", reg);
 954         return ~0;
 955 }
 956
 957 void amdgpu_kiq_wreg(struct amdgpu_device *adev, uint32_t reg, uint32_t v)
 958 {
 959         signed long r, cnt = 0;
 960         unsigned long flags;
 961         uint32_t seq;
 962         struct amdgpu_kiq *kiq = &adev->gfx.kiq[0];
 963         struct amdgpu_ring *ring = &kiq->ring;
 964
 965         BUG_ON(!ring->funcs->emit_wreg);
 966
 967         if (amdgpu_device_skip_hw_access(adev))
 968                 return;
 969
 970         if (adev->mes.ring.sched.ready) {
 971                 amdgpu_mes_wreg(adev, reg, v);
 972                 return;
 973         }
 974
 975         spin_lock_irqsave(&kiq->ring_lock, flags);
 976         amdgpu_ring_alloc(ring, 32);
 977         amdgpu_ring_emit_wreg(ring, reg, v);
 978         r = amdgpu_fence_emit_polling(ring, &seq, MAX_KIQ_REG_WAIT);
 979         if (r)
 980                 goto failed_undo;
 981
 982         amdgpu_ring_commit(ring);
 983         spin_unlock_irqrestore(&kiq->ring_lock, flags);
 984
 985         r = amdgpu_fence_wait_polling(ring, seq, MAX_KIQ_REG_WAIT);
 986
 987         /* don't wait anymore for gpu reset case because this way may
 988          * block gpu_recover() routine forever, e.g. this virt_kiq_rreg
 989          * is triggered in TTM and ttm_bo_lock_delayed_workqueue() will
 990          * never return if we keep waiting in virt_kiq_rreg, which cause
 991          * gpu_recover() hang there.
 992          *
 993          * also don't wait anymore for IRQ context
 994          * */
 995         if (r < 1 && (amdgpu_in_reset(adev) || in_interrupt()))
 996                 goto failed_kiq_write;
 997
 998         might_sleep();
 999         while (r < 1 && cnt++ < MAX_KIQ_REG_TRY) {
1000
1001                 msleep(MAX_KIQ_REG_BAILOUT_INTERVAL);
1002                 r = amdgpu_fence_wait_polling(ring, seq, MAX_KIQ_REG_WAIT);
1003         }
1004
1005         if (cnt > MAX_KIQ_REG_TRY)
1006                 goto failed_kiq_write;
1007
1008         return;
1009
1010 failed_undo:
1011         amdgpu_ring_undo(ring);
1012         spin_unlock_irqrestore(&kiq->ring_lock, flags);
1013 failed_kiq_write:
1014         dev_err(adev->dev, "failed to write reg:%x\n", reg);
1015 }
1016
1017 int amdgpu_gfx_get_num_kcq(struct amdgpu_device *adev)
1018 {
1019         if (amdgpu_num_kcq == -1) {
1020                 return 8;
1021         } else if (amdgpu_num_kcq > 8 || amdgpu_num_kcq < 0) {
1022                 dev_warn(adev->dev, "set kernel compute queue number to 8 due to invalid parameter provided by user\n");
1023                 return 8;
1024         }
1025         return amdgpu_num_kcq;
1026 }
1027
1028 void amdgpu_gfx_cp_init_microcode(struct amdgpu_device *adev,
1029                                   uint32_t ucode_id)
1030 {
1031         const struct gfx_firmware_header_v1_0 *cp_hdr;
1032         const struct gfx_firmware_header_v2_0 *cp_hdr_v2_0;
1033         struct amdgpu_firmware_info *info = NULL;
1034         const struct firmware *ucode_fw;
1035         unsigned int fw_size;
1036
1037         switch (ucode_id) {
1038         case AMDGPU_UCODE_ID_CP_PFP:
1039                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1040                         adev->gfx.pfp_fw->data;
1041                 adev->gfx.pfp_fw_version =
1042                         le32_to_cpu(cp_hdr->header.ucode_version);
1043                 adev->gfx.pfp_feature_version =
1044                         le32_to_cpu(cp_hdr->ucode_feature_version);
1045                 ucode_fw = adev->gfx.pfp_fw;
1046                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes);
1047                 break;
1048         case AMDGPU_UCODE_ID_CP_RS64_PFP:
1049                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1050                         adev->gfx.pfp_fw->data;
1051                 adev->gfx.pfp_fw_version =
1052                         le32_to_cpu(cp_hdr_v2_0->header.ucode_version);
1053                 adev->gfx.pfp_feature_version =
1054                         le32_to_cpu(cp_hdr_v2_0->ucode_feature_version);
1055                 ucode_fw = adev->gfx.pfp_fw;
1056                 fw_size = le32_to_cpu(cp_hdr_v2_0->ucode_size_bytes);
1057                 break;
1058         case AMDGPU_UCODE_ID_CP_RS64_PFP_P0_STACK:
1059         case AMDGPU_UCODE_ID_CP_RS64_PFP_P1_STACK:
1060                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1061                         adev->gfx.pfp_fw->data;
1062                 ucode_fw = adev->gfx.pfp_fw;
1063                 fw_size = le32_to_cpu(cp_hdr_v2_0->data_size_bytes);
1064                 break;
1065         case AMDGPU_UCODE_ID_CP_ME:
1066                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1067                         adev->gfx.me_fw->data;
1068                 adev->gfx.me_fw_version =
1069                         le32_to_cpu(cp_hdr->header.ucode_version);
1070                 adev->gfx.me_feature_version =
1071                         le32_to_cpu(cp_hdr->ucode_feature_version);
1072                 ucode_fw = adev->gfx.me_fw;
1073                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes);
1074                 break;
1075         case AMDGPU_UCODE_ID_CP_RS64_ME:
1076                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1077                         adev->gfx.me_fw->data;
1078                 adev->gfx.me_fw_version =
1079                         le32_to_cpu(cp_hdr_v2_0->header.ucode_version);
1080                 adev->gfx.me_feature_version =
1081                         le32_to_cpu(cp_hdr_v2_0->ucode_feature_version);
1082                 ucode_fw = adev->gfx.me_fw;
1083                 fw_size = le32_to_cpu(cp_hdr_v2_0->ucode_size_bytes);
1084                 break;
1085         case AMDGPU_UCODE_ID_CP_RS64_ME_P0_STACK:
1086         case AMDGPU_UCODE_ID_CP_RS64_ME_P1_STACK:
1087                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1088                         adev->gfx.me_fw->data;
1089                 ucode_fw = adev->gfx.me_fw;
1090                 fw_size = le32_to_cpu(cp_hdr_v2_0->data_size_bytes);
1091                 break;
1092         case AMDGPU_UCODE_ID_CP_CE:
1093                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1094                         adev->gfx.ce_fw->data;
1095                 adev->gfx.ce_fw_version =
1096                         le32_to_cpu(cp_hdr->header.ucode_version);
1097                 adev->gfx.ce_feature_version =
1098                         le32_to_cpu(cp_hdr->ucode_feature_version);
1099                 ucode_fw = adev->gfx.ce_fw;
1100                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes);
1101                 break;
1102         case AMDGPU_UCODE_ID_CP_MEC1:
1103                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1104                         adev->gfx.mec_fw->data;
1105                 adev->gfx.mec_fw_version =
1106                         le32_to_cpu(cp_hdr->header.ucode_version);
1107                 adev->gfx.mec_feature_version =
1108                         le32_to_cpu(cp_hdr->ucode_feature_version);
1109                 ucode_fw = adev->gfx.mec_fw;
1110                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes) -
1111                           le32_to_cpu(cp_hdr->jt_size) * 4;
1112                 break;
1113         case AMDGPU_UCODE_ID_CP_MEC1_JT:
1114                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1115                         adev->gfx.mec_fw->data;
1116                 ucode_fw = adev->gfx.mec_fw;
1117                 fw_size = le32_to_cpu(cp_hdr->jt_size) * 4;
1118                 break;
1119         case AMDGPU_UCODE_ID_CP_MEC2:
1120                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1121                         adev->gfx.mec2_fw->data;
1122                 adev->gfx.mec2_fw_version =
1123                         le32_to_cpu(cp_hdr->header.ucode_version);
1124                 adev->gfx.mec2_feature_version =
1125                         le32_to_cpu(cp_hdr->ucode_feature_version);
1126                 ucode_fw = adev->gfx.mec2_fw;
1127                 fw_size = le32_to_cpu(cp_hdr->header.ucode_size_bytes) -
1128                           le32_to_cpu(cp_hdr->jt_size) * 4;
1129                 break;
1130         case AMDGPU_UCODE_ID_CP_MEC2_JT:
1131                 cp_hdr = (const struct gfx_firmware_header_v1_0 *)
1132                         adev->gfx.mec2_fw->data;
1133                 ucode_fw = adev->gfx.mec2_fw;
1134                 fw_size = le32_to_cpu(cp_hdr->jt_size) * 4;
1135                 break;
1136         case AMDGPU_UCODE_ID_CP_RS64_MEC:
1137                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1138                         adev->gfx.mec_fw->data;
1139                 adev->gfx.mec_fw_version =
1140                         le32_to_cpu(cp_hdr_v2_0->header.ucode_version);
1141                 adev->gfx.mec_feature_version =
1142                         le32_to_cpu(cp_hdr_v2_0->ucode_feature_version);
1143                 ucode_fw = adev->gfx.mec_fw;
1144                 fw_size = le32_to_cpu(cp_hdr_v2_0->ucode_size_bytes);
1145                 break;
1146         case AMDGPU_UCODE_ID_CP_RS64_MEC_P0_STACK:
1147         case AMDGPU_UCODE_ID_CP_RS64_MEC_P1_STACK:
1148         case AMDGPU_UCODE_ID_CP_RS64_MEC_P2_STACK:
1149         case AMDGPU_UCODE_ID_CP_RS64_MEC_P3_STACK:
1150                 cp_hdr_v2_0 = (const struct gfx_firmware_header_v2_0 *)
1151                         adev->gfx.mec_fw->data;
1152                 ucode_fw = adev->gfx.mec_fw;
1153                 fw_size = le32_to_cpu(cp_hdr_v2_0->data_size_bytes);
1154                 break;
1155         default:
1156                 break;
1157         }
1158
1159         if (adev->firmware.load_type == AMDGPU_FW_LOAD_PSP) {
1160                 info = &adev->firmware.ucode[ucode_id];
1161                 info->ucode_id = ucode_id;
1162                 info->fw = ucode_fw;
1163                 adev->firmware.fw_size += ALIGN(fw_size, PAGE_SIZE);
1164         }
1165 }
1166
1167 bool amdgpu_gfx_is_master_xcc(struct amdgpu_device *adev, int xcc_id)
1168 {
1169         return !(xcc_id % (adev->gfx.num_xcc_per_xcp ?
1170                         adev->gfx.num_xcc_per_xcp : 1));
1171 }
1172
1173 static ssize_t amdgpu_gfx_get_current_compute_partition(struct device *dev,
1174                                                 struct device_attribute *addr,
1175                                                 char *buf)
1176 {
1177         struct drm_device *ddev = dev_get_drvdata(dev);
1178         struct amdgpu_device *adev = drm_to_adev(ddev);
1179         enum amdgpu_gfx_partition mode;
1180         char *partition_mode;
1181
1182         mode = adev->gfx.funcs->query_partition_mode(adev);
1183
1184         switch (mode) {
1185         case AMDGPU_SPX_PARTITION_MODE:
1186                 partition_mode = "SPX";
1187                 break;
1188         case AMDGPU_DPX_PARTITION_MODE:
1189                 partition_mode = "DPX";
1190                 break;
1191         case AMDGPU_TPX_PARTITION_MODE:
1192                 partition_mode = "TPX";
1193                 break;
1194         case AMDGPU_QPX_PARTITION_MODE:
1195                 partition_mode = "QPX";
1196                 break;
1197         case AMDGPU_CPX_PARTITION_MODE:
1198                 partition_mode = "CPX";
1199                 break;
1200         default:
1201                 partition_mode = "UNKNOWN";
1202                 break;
1203         }
1204
1205         return sysfs_emit(buf, "%s\n", partition_mode);
1206 }
1207
1208 static ssize_t amdgpu_gfx_set_compute_partition(struct device *dev,
1209                                                 struct device_attribute *addr,
1210                                                 const char *buf, size_t count)
1211 {
1212         struct drm_device *ddev = dev_get_drvdata(dev);
1213         struct amdgpu_device *adev = drm_to_adev(ddev);
1214         enum amdgpu_gfx_partition mode;
1215         int ret;
1216
1217         if (adev->gfx.num_xcd % 2 != 0)
1218                 return -EINVAL;
1219
1220         if (!strncasecmp("SPX", buf, strlen("SPX"))) {
1221                 mode = AMDGPU_SPX_PARTITION_MODE;
1222         } else if (!strncasecmp("DPX", buf, strlen("DPX"))) {
1223                 if (adev->gfx.num_xcd != 4 || adev->gfx.num_xcd != 8)
1224                         return -EINVAL;
1225                 mode = AMDGPU_DPX_PARTITION_MODE;
1226         } else if (!strncasecmp("TPX", buf, strlen("TPX"))) {
1227                 if (adev->gfx.num_xcd != 6)
1228                         return -EINVAL;
1229                 mode = AMDGPU_TPX_PARTITION_MODE;
1230         } else if (!strncasecmp("QPX", buf, strlen("QPX"))) {
1231                 if (adev->gfx.num_xcd != 8)
1232                         return -EINVAL;
1233                 mode = AMDGPU_QPX_PARTITION_MODE;
1234         } else if (!strncasecmp("CPX", buf, strlen("CPX"))) {
1235                 mode = AMDGPU_CPX_PARTITION_MODE;
1236         } else {
1237                 return -EINVAL;
1238         }
1239
1240         mutex_lock(&adev->gfx.partition_mutex);
1241
1242         ret = adev->gfx.funcs->switch_partition_mode(adev, mode);
1243
1244         mutex_unlock(&adev->gfx.partition_mutex);
1245
1246         if (ret)
1247                 return ret;
1248
1249         return count;
1250 }
1251
1252 static ssize_t amdgpu_gfx_get_available_compute_partition(struct device *dev,
1253                                                 struct device_attribute *addr,
1254                                                 char *buf)
1255 {
1256         struct drm_device *ddev = dev_get_drvdata(dev);
1257         struct amdgpu_device *adev = drm_to_adev(ddev);
1258         char *supported_partition;
1259
1260         /* TBD */
1261         switch (adev->gfx.num_xcd) {
1262         case 8:
1263                 supported_partition = "SPX, DPX, QPX, CPX";
1264                 break;
1265         case 6:
1266                 supported_partition = "SPX, TPX, CPX";
1267                 break;
1268         case 4:
1269                 supported_partition = "SPX, DPX, CPX";
1270                 break;
1271         /* this seems only existing in emulation phase */
1272         case 2:
1273                 supported_partition = "SPX, CPX";
1274                 break;
1275         default:
1276                 supported_partition = "Not supported";
1277                 break;
1278         }
1279
1280         return sysfs_emit(buf, "%s\n", supported_partition);
1281 }
1282
1283 static DEVICE_ATTR(current_compute_partition, S_IRUGO | S_IWUSR,
1284                    amdgpu_gfx_get_current_compute_partition,
1285                    amdgpu_gfx_set_compute_partition);
1286
1287 static DEVICE_ATTR(available_compute_partition, S_IRUGO,
1288                    amdgpu_gfx_get_available_compute_partition, NULL);
1289
1290 int amdgpu_gfx_sysfs_init(struct amdgpu_device *adev)
1291 {
1292         int r;
1293
1294         r = device_create_file(adev->dev, &dev_attr_current_compute_partition);
1295         if (r)
1296                 return r;
1297
1298         r = device_create_file(adev->dev, &dev_attr_available_compute_partition);
1299         if (r)
1300                 return r;
1301
1302         return 0;
1303 }