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

   1 /*
   2  * Copyright 2016 Advanced Micro Devices, Inc.
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice shall be included in
  12  * all copies or substantial portions of the Software.
  13  *
  14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  17  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
  18  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  19  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
  20  * OTHER DEALINGS IN THE SOFTWARE.
  21  *
  22  * Author: Huang Rui
  23  *
  24  */
  25
  26 #include <linux/firmware.h>
  27 #include <linux/module.h>
  28 #include <linux/pci.h>
  29
  30 #include "amdgpu.h"
  31 #include "amdgpu_psp.h"
  32 #include "amdgpu_ucode.h"
  33 #include "soc15_common.h"
  34 #include "psp_v3_1.h"
  35
  36 #include "mp/mp_9_0_offset.h"
  37 #include "mp/mp_9_0_sh_mask.h"
  38 #include "gc/gc_9_0_offset.h"
  39 #include "sdma0/sdma0_4_0_offset.h"
  40 #include "nbio/nbio_6_1_offset.h"
  41
  42 #include "oss/osssys_4_0_offset.h"
  43 #include "oss/osssys_4_0_sh_mask.h"
  44
  45 MODULE_FIRMWARE("amdgpu/vega10_sos.bin");
  46 MODULE_FIRMWARE("amdgpu/vega10_asd.bin");
  47 MODULE_FIRMWARE("amdgpu/vega12_sos.bin");
  48 MODULE_FIRMWARE("amdgpu/vega12_asd.bin");
  49
  50
  51 #define smnMP1_FIRMWARE_FLAGS 0x3010028
  52
  53 static uint32_t sos_old_versions[] = {1517616, 1510592, 1448594, 1446554};
  54
  55 static bool psp_v3_1_support_vmr_ring(struct psp_context *psp);
  56 static int psp_v3_1_ring_stop(struct psp_context *psp,
  57                               enum psp_ring_type ring_type);
  58
  59 static int psp_v3_1_init_microcode(struct psp_context *psp)
  60 {
  61         struct amdgpu_device *adev = psp->adev;
  62         const char *chip_name;
  63         char fw_name[30];
  64         int err = 0;
  65         const struct psp_firmware_header_v1_0 *hdr;
  66
  67         DRM_DEBUG("\n");
  68
  69         switch (adev->asic_type) {
  70         case CHIP_VEGA10:
  71                 chip_name = "vega10";
  72                 break;
  73         case CHIP_VEGA12:
  74                 chip_name = "vega12";
  75                 break;
  76         default: BUG();
  77         }
  78
  79         snprintf(fw_name, sizeof(fw_name), "amdgpu/%s_sos.bin", chip_name);
  80         err = request_firmware(&adev->psp.sos_fw, fw_name, adev->dev);
  81         if (err)
  82                 goto out;
  83
  84         err = amdgpu_ucode_validate(adev->psp.sos_fw);
  85         if (err)
  86                 goto out;
  87
  88         hdr = (const struct psp_firmware_header_v1_0 *)adev->psp.sos_fw->data;
  89         adev->psp.sos_fw_version = le32_to_cpu(hdr->header.ucode_version);
  90         adev->psp.sos_feature_version = le32_to_cpu(hdr->ucode_feature_version);
  91         adev->psp.sos_bin_size = le32_to_cpu(hdr->sos_size_bytes);
  92         adev->psp.sys_bin_size = le32_to_cpu(hdr->header.ucode_size_bytes) -
  93                                         le32_to_cpu(hdr->sos_size_bytes);
  94         adev->psp.sys_start_addr = (uint8_t *)hdr +
  95                                 le32_to_cpu(hdr->header.ucode_array_offset_bytes);
  96         adev->psp.sos_start_addr = (uint8_t *)adev->psp.sys_start_addr +
  97                                 le32_to_cpu(hdr->sos_offset_bytes);
  98
  99         snprintf(fw_name, sizeof(fw_name), "amdgpu/%s_asd.bin", chip_name);
 100         err = request_firmware(&adev->psp.asd_fw, fw_name, adev->dev);
 101         if (err)
 102                 goto out;
 103
 104         err = amdgpu_ucode_validate(adev->psp.asd_fw);
 105         if (err)
 106                 goto out;
 107
 108         hdr = (const struct psp_firmware_header_v1_0 *)adev->psp.asd_fw->data;
 109         adev->psp.asd_fw_version = le32_to_cpu(hdr->header.ucode_version);
 110         adev->psp.asd_feature_version = le32_to_cpu(hdr->ucode_feature_version);
 111         adev->psp.asd_ucode_size = le32_to_cpu(hdr->header.ucode_size_bytes);
 112         adev->psp.asd_start_addr = (uint8_t *)hdr +
 113                                 le32_to_cpu(hdr->header.ucode_array_offset_bytes);
 114
 115         return 0;
 116 out:
 117         if (err) {
 118                 dev_err(adev->dev,
 119                         "psp v3.1: Failed to load firmware \"%s\"\n",
 120                         fw_name);
 121                 release_firmware(adev->psp.sos_fw);
 122                 adev->psp.sos_fw = NULL;
 123                 release_firmware(adev->psp.asd_fw);
 124                 adev->psp.asd_fw = NULL;
 125         }
 126
 127         return err;
 128 }
 129
 130 static int psp_v3_1_bootloader_load_sysdrv(struct psp_context *psp)
 131 {
 132         int ret;
 133         uint32_t psp_gfxdrv_command_reg = 0;
 134         struct amdgpu_device *adev = psp->adev;
 135         uint32_t sol_reg;
 136
 137         /* Check sOS sign of life register to confirm sys driver and sOS
 138          * are already been loaded.
 139          */
 140         sol_reg = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_81);
 141         if (sol_reg)
 142                 return 0;
 143
 144         /* Wait for bootloader to signify that is ready having bit 31 of C2PMSG_35 set to 1 */
 145         ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_35),
 146                            0x80000000, 0x80000000, false);
 147         if (ret)
 148                 return ret;
 149
 150         memset(psp->fw_pri_buf, 0, PSP_1_MEG);
 151
 152         /* Copy PSP System Driver binary to memory */
 153         memcpy(psp->fw_pri_buf, psp->sys_start_addr, psp->sys_bin_size);
 154
 155         /* Provide the sys driver to bootloader */
 156         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36,
 157                (uint32_t)(psp->fw_pri_mc_addr >> 20));
 158         psp_gfxdrv_command_reg = PSP_BL__LOAD_SYSDRV;
 159         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35,
 160                psp_gfxdrv_command_reg);
 161
 162         /* there might be handshake issue with hardware which needs delay */
 163         mdelay(20);
 164
 165         ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_35),
 166                            0x80000000, 0x80000000, false);
 167
 168         return ret;
 169 }
 170
 171 static bool psp_v3_1_match_version(struct amdgpu_device *adev, uint32_t ver)
 172 {
 173         int i;
 174
 175         if (ver == adev->psp.sos_fw_version)
 176                 return true;
 177
 178         /*
 179          * Double check if the latest four legacy versions.
 180          * If yes, it is still the right version.
 181          */
 182         for (i = 0; i < ARRAY_SIZE(sos_old_versions); i++) {
 183                 if (sos_old_versions[i] == adev->psp.sos_fw_version)
 184                         return true;
 185         }
 186
 187         return false;
 188 }
 189
 190 static int psp_v3_1_bootloader_load_sos(struct psp_context *psp)
 191 {
 192         int ret;
 193         unsigned int psp_gfxdrv_command_reg = 0;
 194         struct amdgpu_device *adev = psp->adev;
 195         uint32_t sol_reg, ver;
 196
 197         /* Check sOS sign of life register to confirm sys driver and sOS
 198          * are already been loaded.
 199          */
 200         sol_reg = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_81);
 201         if (sol_reg) {
 202                 psp->sos_fw_version = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_58);
 203                 printk("sos fw version = 0x%x.\n", psp->sos_fw_version);
 204                 return 0;
 205         }
 206
 207         /* Wait for bootloader to signify that is ready having bit 31 of C2PMSG_35 set to 1 */
 208         ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_35),
 209                            0x80000000, 0x80000000, false);
 210         if (ret)
 211                 return ret;
 212
 213         memset(psp->fw_pri_buf, 0, PSP_1_MEG);
 214
 215         /* Copy Secure OS binary to PSP memory */
 216         memcpy(psp->fw_pri_buf, psp->sos_start_addr, psp->sos_bin_size);
 217
 218         /* Provide the PSP secure OS to bootloader */
 219         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36,
 220                (uint32_t)(psp->fw_pri_mc_addr >> 20));
 221         psp_gfxdrv_command_reg = PSP_BL__LOAD_SOSDRV;
 222         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35,
 223                psp_gfxdrv_command_reg);
 224
 225         /* there might be handshake issue with hardware which needs delay */
 226         mdelay(20);
 227         ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_81),
 228                            RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_81),
 229                            0, true);
 230
 231         ver = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_58);
 232         if (!psp_v3_1_match_version(adev, ver))
 233                 DRM_WARN("SOS version doesn't match\n");
 234
 235         return ret;
 236 }
 237
 238 static int psp_v3_1_ring_init(struct psp_context *psp,
 239                               enum psp_ring_type ring_type)
 240 {
 241         int ret = 0;
 242         struct psp_ring *ring;
 243         struct amdgpu_device *adev = psp->adev;
 244
 245         ring = &psp->km_ring;
 246
 247         ring->ring_type = ring_type;
 248
 249         /* allocate 4k Page of Local Frame Buffer memory for ring */
 250         ring->ring_size = 0x1000;
 251         ret = amdgpu_bo_create_kernel(adev, ring->ring_size, PAGE_SIZE,
 252                                       AMDGPU_GEM_DOMAIN_VRAM,
 253                                       &adev->firmware.rbuf,
 254                                       &ring->ring_mem_mc_addr,
 255                                       (void **)&ring->ring_mem);
 256         if (ret) {
 257                 ring->ring_size = 0;
 258                 return ret;
 259         }
 260
 261         return 0;
 262 }
 263
 264 static void psp_v3_1_reroute_ih(struct psp_context *psp)
 265 {
 266         struct amdgpu_device *adev = psp->adev;
 267         uint32_t tmp;
 268
 269         /* Change IH ring for VMC */
 270         tmp = REG_SET_FIELD(0, IH_CLIENT_CFG_DATA, CREDIT_RETURN_ADDR, 0x1244b);
 271         tmp = REG_SET_FIELD(tmp, IH_CLIENT_CFG_DATA, CLIENT_TYPE, 1);
 272         tmp = REG_SET_FIELD(tmp, IH_CLIENT_CFG_DATA, RING_ID, 1);
 273
 274         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_69, 3);
 275         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_70, tmp);
 276         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_64, GFX_CTRL_CMD_ID_GBR_IH_SET);
 277
 278         mdelay(20);
 279         psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_64),
 280                      0x80000000, 0x8000FFFF, false);
 281
 282         /* Change IH ring for UMC */
 283         tmp = REG_SET_FIELD(0, IH_CLIENT_CFG_DATA, CREDIT_RETURN_ADDR, 0x1216b);
 284         tmp = REG_SET_FIELD(tmp, IH_CLIENT_CFG_DATA, RING_ID, 1);
 285
 286         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_69, 4);
 287         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_70, tmp);
 288         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_64, GFX_CTRL_CMD_ID_GBR_IH_SET);
 289
 290         mdelay(20);
 291         psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_64),
 292                      0x80000000, 0x8000FFFF, false);
 293 }
 294
 295 static int psp_v3_1_ring_create(struct psp_context *psp,
 296                                 enum psp_ring_type ring_type)
 297 {
 298         int ret = 0;
 299         unsigned int psp_ring_reg = 0;
 300         struct psp_ring *ring = &psp->km_ring;
 301         struct amdgpu_device *adev = psp->adev;
 302
 303         psp_v3_1_reroute_ih(psp);
 304
 305         if (psp_v3_1_support_vmr_ring(psp)) {
 306                 ret = psp_v3_1_ring_stop(psp, ring_type);
 307                 if (ret) {
 308                         DRM_ERROR("psp_v3_1_ring_stop_sriov failed!\n");
 309                         return ret;
 310                 }
 311
 312                 /* Write low address of the ring to C2PMSG_102 */
 313                 psp_ring_reg = lower_32_bits(ring->ring_mem_mc_addr);
 314                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_102, psp_ring_reg);
 315                 /* Write high address of the ring to C2PMSG_103 */
 316                 psp_ring_reg = upper_32_bits(ring->ring_mem_mc_addr);
 317                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_103, psp_ring_reg);
 318                 /* No size initialization for sriov  */
 319                 /* Write the ring initialization command to C2PMSG_101 */
 320                 psp_ring_reg = ring_type;
 321                 psp_ring_reg = psp_ring_reg << 16;
 322                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_101, psp_ring_reg);
 323
 324                 /* there might be hardware handshake issue which needs delay */
 325                 mdelay(20);
 326
 327                 /* Wait for response flag (bit 31) in C2PMSG_101 */
 328                 ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0,
 329                                         mmMP0_SMN_C2PMSG_101), 0x80000000,
 330                                         0x8000FFFF, false);
 331         } else {
 332
 333                 /* Write low address of the ring to C2PMSG_69 */
 334                 psp_ring_reg = lower_32_bits(ring->ring_mem_mc_addr);
 335                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_69, psp_ring_reg);
 336                 /* Write high address of the ring to C2PMSG_70 */
 337                 psp_ring_reg = upper_32_bits(ring->ring_mem_mc_addr);
 338                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_70, psp_ring_reg);
 339                 /* Write size of ring to C2PMSG_71 */
 340                 psp_ring_reg = ring->ring_size;
 341                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_71, psp_ring_reg);
 342                 /* Write the ring initialization command to C2PMSG_64 */
 343                 psp_ring_reg = ring_type;
 344                 psp_ring_reg = psp_ring_reg << 16;
 345                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_64, psp_ring_reg);
 346
 347                 /* there might be hardware handshake issue which needs delay */
 348                 mdelay(20);
 349
 350                 /* Wait for response flag (bit 31) in C2PMSG_64 */
 351                 ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0,
 352                                         mmMP0_SMN_C2PMSG_64), 0x80000000,
 353                                         0x8000FFFF, false);
 354
 355         }
 356         return ret;
 357 }
 358
 359 static int psp_v3_1_ring_stop(struct psp_context *psp,
 360                               enum psp_ring_type ring_type)
 361 {
 362         int ret = 0;
 363         unsigned int psp_ring_reg = 0;
 364         struct amdgpu_device *adev = psp->adev;
 365
 366         if (psp_v3_1_support_vmr_ring(psp)) {
 367                 /* Write the Destroy GPCOM ring command to C2PMSG_101 */
 368                 psp_ring_reg = GFX_CTRL_CMD_ID_DESTROY_GPCOM_RING;
 369                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_101, psp_ring_reg);
 370
 371                 /* there might be handshake issue which needs delay */
 372                 mdelay(20);
 373
 374                 /* Wait for response flag (bit 31) in C2PMSG_101 */
 375                 ret = psp_wait_for(psp,
 376                                 SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_101),
 377                                 0x80000000, 0x80000000, false);
 378         } else {
 379                 /* Write the ring destroy command to C2PMSG_64 */
 380                 psp_ring_reg = 3 << 16;
 381                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_64, psp_ring_reg);
 382
 383                 /* there might be handshake issue which needs delay */
 384                 mdelay(20);
 385
 386                 /* Wait for response flag (bit 31) in C2PMSG_64 */
 387                 ret = psp_wait_for(psp,
 388                                 SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_64),
 389                                 0x80000000, 0x80000000, false);
 390         }
 391
 392         return ret;
 393 }
 394
 395 static int psp_v3_1_ring_destroy(struct psp_context *psp,
 396                                  enum psp_ring_type ring_type)
 397 {
 398         int ret = 0;
 399         struct psp_ring *ring = &psp->km_ring;
 400         struct amdgpu_device *adev = psp->adev;
 401
 402         ret = psp_v3_1_ring_stop(psp, ring_type);
 403         if (ret)
 404                 DRM_ERROR("Fail to stop psp ring\n");
 405
 406         amdgpu_bo_free_kernel(&adev->firmware.rbuf,
 407                               &ring->ring_mem_mc_addr,
 408                               (void **)&ring->ring_mem);
 409
 410         return ret;
 411 }
 412
 413 static int
 414 psp_v3_1_sram_map(struct amdgpu_device *adev,
 415                   unsigned int *sram_offset, unsigned int *sram_addr_reg_offset,
 416                   unsigned int *sram_data_reg_offset,
 417                   enum AMDGPU_UCODE_ID ucode_id)
 418 {
 419         int ret = 0;
 420
 421         switch(ucode_id) {
 422 /* TODO: needs to confirm */
 423 #if 0
 424         case AMDGPU_UCODE_ID_SMC:
 425                 *sram_offset = 0;
 426                 *sram_addr_reg_offset = 0;
 427                 *sram_data_reg_offset = 0;
 428                 break;
 429 #endif
 430
 431         case AMDGPU_UCODE_ID_CP_CE:
 432                 *sram_offset = 0x0;
 433                 *sram_addr_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_CE_UCODE_ADDR);
 434                 *sram_data_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_CE_UCODE_DATA);
 435                 break;
 436
 437         case AMDGPU_UCODE_ID_CP_PFP:
 438                 *sram_offset = 0x0;
 439                 *sram_addr_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_PFP_UCODE_ADDR);
 440                 *sram_data_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_PFP_UCODE_DATA);
 441                 break;
 442
 443         case AMDGPU_UCODE_ID_CP_ME:
 444                 *sram_offset = 0x0;
 445                 *sram_addr_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_HYP_ME_UCODE_ADDR);
 446                 *sram_data_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_HYP_ME_UCODE_DATA);
 447                 break;
 448
 449         case AMDGPU_UCODE_ID_CP_MEC1:
 450                 *sram_offset = 0x10000;
 451                 *sram_addr_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_MEC_ME1_UCODE_ADDR);
 452                 *sram_data_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_MEC_ME1_UCODE_DATA);
 453                 break;
 454
 455         case AMDGPU_UCODE_ID_CP_MEC2:
 456                 *sram_offset = 0x10000;
 457                 *sram_addr_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_HYP_MEC2_UCODE_ADDR);
 458                 *sram_data_reg_offset = SOC15_REG_OFFSET(GC, 0, mmCP_HYP_MEC2_UCODE_DATA);
 459                 break;
 460
 461         case AMDGPU_UCODE_ID_RLC_G:
 462                 *sram_offset = 0x2000;
 463                 *sram_addr_reg_offset = SOC15_REG_OFFSET(GC, 0, mmRLC_GPM_UCODE_ADDR);
 464                 *sram_data_reg_offset = SOC15_REG_OFFSET(GC, 0, mmRLC_GPM_UCODE_DATA);
 465                 break;
 466
 467         case AMDGPU_UCODE_ID_SDMA0:
 468                 *sram_offset = 0x0;
 469                 *sram_addr_reg_offset = SOC15_REG_OFFSET(SDMA0, 0, mmSDMA0_UCODE_ADDR);
 470                 *sram_data_reg_offset = SOC15_REG_OFFSET(SDMA0, 0, mmSDMA0_UCODE_DATA);
 471                 break;
 472
 473 /* TODO: needs to confirm */
 474 #if 0
 475         case AMDGPU_UCODE_ID_SDMA1:
 476                 *sram_offset = ;
 477                 *sram_addr_reg_offset = ;
 478                 break;
 479
 480         case AMDGPU_UCODE_ID_UVD:
 481                 *sram_offset = ;
 482                 *sram_addr_reg_offset = ;
 483                 break;
 484
 485         case AMDGPU_UCODE_ID_VCE:
 486                 *sram_offset = ;
 487                 *sram_addr_reg_offset = ;
 488                 break;
 489 #endif
 490
 491         case AMDGPU_UCODE_ID_MAXIMUM:
 492         default:
 493                 ret = -EINVAL;
 494                 break;
 495         }
 496
 497         return ret;
 498 }
 499
 500 static bool psp_v3_1_compare_sram_data(struct psp_context *psp,
 501                                        struct amdgpu_firmware_info *ucode,
 502                                        enum AMDGPU_UCODE_ID ucode_type)
 503 {
 504         int err = 0;
 505         unsigned int fw_sram_reg_val = 0;
 506         unsigned int fw_sram_addr_reg_offset = 0;
 507         unsigned int fw_sram_data_reg_offset = 0;
 508         unsigned int ucode_size;
 509         uint32_t *ucode_mem = NULL;
 510         struct amdgpu_device *adev = psp->adev;
 511
 512         err = psp_v3_1_sram_map(adev, &fw_sram_reg_val, &fw_sram_addr_reg_offset,
 513                                 &fw_sram_data_reg_offset, ucode_type);
 514         if (err)
 515                 return false;
 516
 517         WREG32(fw_sram_addr_reg_offset, fw_sram_reg_val);
 518
 519         ucode_size = ucode->ucode_size;
 520         ucode_mem = (uint32_t *)ucode->kaddr;
 521         while (ucode_size) {
 522                 fw_sram_reg_val = RREG32(fw_sram_data_reg_offset);
 523
 524                 if (*ucode_mem != fw_sram_reg_val)
 525                         return false;
 526
 527                 ucode_mem++;
 528                 /* 4 bytes */
 529                 ucode_size -= 4;
 530         }
 531
 532         return true;
 533 }
 534
 535 static bool psp_v3_1_smu_reload_quirk(struct psp_context *psp)
 536 {
 537         struct amdgpu_device *adev = psp->adev;
 538         uint32_t reg;
 539
 540         reg = RREG32_PCIE(smnMP1_FIRMWARE_FLAGS | 0x03b00000);
 541         return (reg & MP1_FIRMWARE_FLAGS__INTERRUPTS_ENABLED_MASK) ? true : false;
 542 }
 543
 544 static int psp_v3_1_mode1_reset(struct psp_context *psp)
 545 {
 546         int ret;
 547         uint32_t offset;
 548         struct amdgpu_device *adev = psp->adev;
 549
 550         offset = SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_64);
 551
 552         ret = psp_wait_for(psp, offset, 0x80000000, 0x8000FFFF, false);
 553
 554         if (ret) {
 555                 DRM_INFO("psp is not working correctly before mode1 reset!\n");
 556                 return -EINVAL;
 557         }
 558
 559         /*send the mode 1 reset command*/
 560         WREG32(offset, GFX_CTRL_CMD_ID_MODE1_RST);
 561
 562         msleep(500);
 563
 564         offset = SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_33);
 565
 566         ret = psp_wait_for(psp, offset, 0x80000000, 0x80000000, false);
 567
 568         if (ret) {
 569                 DRM_INFO("psp mode 1 reset failed!\n");
 570                 return -EINVAL;
 571         }
 572
 573         DRM_INFO("psp mode1 reset succeed \n");
 574
 575         return 0;
 576 }
 577
 578 static bool psp_v3_1_support_vmr_ring(struct psp_context *psp)
 579 {
 580         if (amdgpu_sriov_vf(psp->adev))
 581                 return true;
 582
 583         return false;
 584 }
 585
 586 static uint32_t psp_v3_1_ring_get_wptr(struct psp_context *psp)
 587 {
 588         uint32_t data;
 589         struct amdgpu_device *adev = psp->adev;
 590
 591         if (psp_v3_1_support_vmr_ring(psp))
 592                 data = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_102);
 593         else
 594                 data = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_67);
 595         return data;
 596 }
 597
 598 static void psp_v3_1_ring_set_wptr(struct psp_context *psp, uint32_t value)
 599 {
 600         struct amdgpu_device *adev = psp->adev;
 601
 602         if (psp_v3_1_support_vmr_ring(psp)) {
 603                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_102, value);
 604                 /* send interrupt to PSP for SRIOV ring write pointer update */
 605                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_101,
 606                         GFX_CTRL_CMD_ID_CONSUME_CMD);
 607         } else
 608                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_67, value);
 609 }
 610
 611 static const struct psp_funcs psp_v3_1_funcs = {
 612         .init_microcode = psp_v3_1_init_microcode,
 613         .bootloader_load_sysdrv = psp_v3_1_bootloader_load_sysdrv,
 614         .bootloader_load_sos = psp_v3_1_bootloader_load_sos,
 615         .ring_init = psp_v3_1_ring_init,
 616         .ring_create = psp_v3_1_ring_create,
 617         .ring_stop = psp_v3_1_ring_stop,
 618         .ring_destroy = psp_v3_1_ring_destroy,
 619         .compare_sram_data = psp_v3_1_compare_sram_data,
 620         .smu_reload_quirk = psp_v3_1_smu_reload_quirk,
 621         .mode1_reset = psp_v3_1_mode1_reset,
 622         .support_vmr_ring = psp_v3_1_support_vmr_ring,
 623         .ring_get_wptr = psp_v3_1_ring_get_wptr,
 624         .ring_set_wptr = psp_v3_1_ring_set_wptr,
 625 };
 626
 627 void psp_v3_1_set_psp_funcs(struct psp_context *psp)
 628 {
 629         psp->funcs = &psp_v3_1_funcs;
 630 }