1 /* 2 * Copyright 2015 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 * Authors: monk liu <monk.liu@amd.com> 23 */ 24 25 #include <drm/drm_auth.h> 26 #include "amdgpu.h" 27 #include "amdgpu_sched.h" 28 #include "amdgpu_ras.h" 29 30 #define to_amdgpu_ctx_entity(e) \ 31 container_of((e), struct amdgpu_ctx_entity, entity) 32 33 const unsigned int amdgpu_ctx_num_entities[AMDGPU_HW_IP_NUM] = { 34 [AMDGPU_HW_IP_GFX] = 1, 35 [AMDGPU_HW_IP_COMPUTE] = 4, 36 [AMDGPU_HW_IP_DMA] = 2, 37 [AMDGPU_HW_IP_UVD] = 1, 38 [AMDGPU_HW_IP_VCE] = 1, 39 [AMDGPU_HW_IP_UVD_ENC] = 1, 40 [AMDGPU_HW_IP_VCN_DEC] = 1, 41 [AMDGPU_HW_IP_VCN_ENC] = 1, 42 [AMDGPU_HW_IP_VCN_JPEG] = 1, 43 }; 44 45 static int amdgput_ctx_total_num_entities(void) 46 { 47 unsigned i, num_entities = 0; 48 49 for (i = 0; i < AMDGPU_HW_IP_NUM; ++i) 50 num_entities += amdgpu_ctx_num_entities[i]; 51 52 return num_entities; 53 } 54 55 static int amdgpu_ctx_priority_permit(struct drm_file *filp, 56 enum drm_sched_priority priority) 57 { 58 /* NORMAL and below are accessible by everyone */ 59 if (priority <= DRM_SCHED_PRIORITY_NORMAL) 60 return 0; 61 62 if (capable(CAP_SYS_NICE)) 63 return 0; 64 65 if (drm_is_current_master(filp)) 66 return 0; 67 68 return -EACCES; 69 } 70 71 static int amdgpu_ctx_init(struct amdgpu_device *adev, 72 enum drm_sched_priority priority, 73 struct drm_file *filp, 74 struct amdgpu_ctx *ctx) 75 { 76 unsigned num_entities = amdgput_ctx_total_num_entities(); 77 unsigned i, j; 78 int r; 79 80 if (priority < 0 || priority >= DRM_SCHED_PRIORITY_MAX) 81 return -EINVAL; 82 83 r = amdgpu_ctx_priority_permit(filp, priority); 84 if (r) 85 return r; 86 87 memset(ctx, 0, sizeof(*ctx)); 88 ctx->adev = adev; 89 90 ctx->fences = kcalloc(amdgpu_sched_jobs * num_entities, 91 sizeof(struct dma_fence*), GFP_KERNEL); 92 if (!ctx->fences) 93 return -ENOMEM; 94 95 ctx->entities[0] = kcalloc(num_entities, 96 sizeof(struct amdgpu_ctx_entity), 97 GFP_KERNEL); 98 if (!ctx->entities[0]) { 99 r = -ENOMEM; 100 goto error_free_fences; 101 } 102 103 for (i = 0; i < num_entities; ++i) { 104 struct amdgpu_ctx_entity *entity = &ctx->entities[0][i]; 105 106 entity->sequence = 1; 107 entity->fences = &ctx->fences[amdgpu_sched_jobs * i]; 108 } 109 for (i = 1; i < AMDGPU_HW_IP_NUM; ++i) 110 ctx->entities[i] = ctx->entities[i - 1] + 111 amdgpu_ctx_num_entities[i - 1]; 112 113 kref_init(&ctx->refcount); 114 spin_lock_init(&ctx->ring_lock); 115 mutex_init(&ctx->lock); 116 117 ctx->reset_counter = atomic_read(&adev->gpu_reset_counter); 118 ctx->reset_counter_query = ctx->reset_counter; 119 ctx->vram_lost_counter = atomic_read(&adev->vram_lost_counter); 120 ctx->init_priority = priority; 121 ctx->override_priority = DRM_SCHED_PRIORITY_UNSET; 122 123 for (i = 0; i < AMDGPU_HW_IP_NUM; ++i) { 124 struct amdgpu_ring *rings[AMDGPU_MAX_RINGS]; 125 struct drm_sched_rq *rqs[AMDGPU_MAX_RINGS]; 126 unsigned num_rings; 127 unsigned num_rqs = 0; 128 129 switch (i) { 130 case AMDGPU_HW_IP_GFX: 131 rings[0] = &adev->gfx.gfx_ring[0]; 132 num_rings = 1; 133 break; 134 case AMDGPU_HW_IP_COMPUTE: 135 for (j = 0; j < adev->gfx.num_compute_rings; ++j) 136 rings[j] = &adev->gfx.compute_ring[j]; 137 num_rings = adev->gfx.num_compute_rings; 138 break; 139 case AMDGPU_HW_IP_DMA: 140 for (j = 0; j < adev->sdma.num_instances; ++j) 141 rings[j] = &adev->sdma.instance[j].ring; 142 num_rings = adev->sdma.num_instances; 143 break; 144 case AMDGPU_HW_IP_UVD: 145 rings[0] = &adev->uvd.inst[0].ring; 146 num_rings = 1; 147 break; 148 case AMDGPU_HW_IP_VCE: 149 rings[0] = &adev->vce.ring[0]; 150 num_rings = 1; 151 break; 152 case AMDGPU_HW_IP_UVD_ENC: 153 rings[0] = &adev->uvd.inst[0].ring_enc[0]; 154 num_rings = 1; 155 break; 156 case AMDGPU_HW_IP_VCN_DEC: 157 rings[0] = &adev->vcn.ring_dec; 158 num_rings = 1; 159 break; 160 case AMDGPU_HW_IP_VCN_ENC: 161 rings[0] = &adev->vcn.ring_enc[0]; 162 num_rings = 1; 163 break; 164 case AMDGPU_HW_IP_VCN_JPEG: 165 rings[0] = &adev->vcn.ring_jpeg; 166 num_rings = 1; 167 break; 168 } 169 170 for (j = 0; j < num_rings; ++j) { 171 if (!rings[j]->adev) 172 continue; 173 174 rqs[num_rqs++] = &rings[j]->sched.sched_rq[priority]; 175 } 176 177 for (j = 0; j < amdgpu_ctx_num_entities[i]; ++j) 178 r = drm_sched_entity_init(&ctx->entities[i][j].entity, 179 rqs, num_rqs, &ctx->guilty); 180 if (r) 181 goto error_cleanup_entities; 182 } 183 184 return 0; 185 186 error_cleanup_entities: 187 for (i = 0; i < num_entities; ++i) 188 drm_sched_entity_destroy(&ctx->entities[0][i].entity); 189 kfree(ctx->entities[0]); 190 191 error_free_fences: 192 kfree(ctx->fences); 193 ctx->fences = NULL; 194 return r; 195 } 196 197 static void amdgpu_ctx_fini(struct kref *ref) 198 { 199 struct amdgpu_ctx *ctx = container_of(ref, struct amdgpu_ctx, refcount); 200 unsigned num_entities = amdgput_ctx_total_num_entities(); 201 struct amdgpu_device *adev = ctx->adev; 202 unsigned i, j; 203 204 if (!adev) 205 return; 206 207 for (i = 0; i < num_entities; ++i) 208 for (j = 0; j < amdgpu_sched_jobs; ++j) 209 dma_fence_put(ctx->entities[0][i].fences[j]); 210 kfree(ctx->fences); 211 kfree(ctx->entities[0]); 212 213 mutex_destroy(&ctx->lock); 214 215 kfree(ctx); 216 } 217 218 int amdgpu_ctx_get_entity(struct amdgpu_ctx *ctx, u32 hw_ip, u32 instance, 219 u32 ring, struct drm_sched_entity **entity) 220 { 221 if (hw_ip >= AMDGPU_HW_IP_NUM) { 222 DRM_ERROR("unknown HW IP type: %d\n", hw_ip); 223 return -EINVAL; 224 } 225 226 /* Right now all IPs have only one instance - multiple rings. */ 227 if (instance != 0) { 228 DRM_DEBUG("invalid ip instance: %d\n", instance); 229 return -EINVAL; 230 } 231 232 if (ring >= amdgpu_ctx_num_entities[hw_ip]) { 233 DRM_DEBUG("invalid ring: %d %d\n", hw_ip, ring); 234 return -EINVAL; 235 } 236 237 *entity = &ctx->entities[hw_ip][ring].entity; 238 return 0; 239 } 240 241 static int amdgpu_ctx_alloc(struct amdgpu_device *adev, 242 struct amdgpu_fpriv *fpriv, 243 struct drm_file *filp, 244 enum drm_sched_priority priority, 245 uint32_t *id) 246 { 247 struct amdgpu_ctx_mgr *mgr = &fpriv->ctx_mgr; 248 struct amdgpu_ctx *ctx; 249 int r; 250 251 ctx = kmalloc(sizeof(*ctx), GFP_KERNEL); 252 if (!ctx) 253 return -ENOMEM; 254 255 mutex_lock(&mgr->lock); 256 r = idr_alloc(&mgr->ctx_handles, ctx, 1, AMDGPU_VM_MAX_NUM_CTX, GFP_KERNEL); 257 if (r < 0) { 258 mutex_unlock(&mgr->lock); 259 kfree(ctx); 260 return r; 261 } 262 263 *id = (uint32_t)r; 264 r = amdgpu_ctx_init(adev, priority, filp, ctx); 265 if (r) { 266 idr_remove(&mgr->ctx_handles, *id); 267 *id = 0; 268 kfree(ctx); 269 } 270 mutex_unlock(&mgr->lock); 271 return r; 272 } 273 274 static void amdgpu_ctx_do_release(struct kref *ref) 275 { 276 struct amdgpu_ctx *ctx; 277 unsigned num_entities; 278 u32 i; 279 280 ctx = container_of(ref, struct amdgpu_ctx, refcount); 281 282 num_entities = 0; 283 for (i = 0; i < AMDGPU_HW_IP_NUM; i++) 284 num_entities += amdgpu_ctx_num_entities[i]; 285 286 for (i = 0; i < num_entities; i++) 287 drm_sched_entity_destroy(&ctx->entities[0][i].entity); 288 289 amdgpu_ctx_fini(ref); 290 } 291 292 static int amdgpu_ctx_free(struct amdgpu_fpriv *fpriv, uint32_t id) 293 { 294 struct amdgpu_ctx_mgr *mgr = &fpriv->ctx_mgr; 295 struct amdgpu_ctx *ctx; 296 297 mutex_lock(&mgr->lock); 298 ctx = idr_remove(&mgr->ctx_handles, id); 299 if (ctx) 300 kref_put(&ctx->refcount, amdgpu_ctx_do_release); 301 mutex_unlock(&mgr->lock); 302 return ctx ? 0 : -EINVAL; 303 } 304 305 static int amdgpu_ctx_query(struct amdgpu_device *adev, 306 struct amdgpu_fpriv *fpriv, uint32_t id, 307 union drm_amdgpu_ctx_out *out) 308 { 309 struct amdgpu_ctx *ctx; 310 struct amdgpu_ctx_mgr *mgr; 311 unsigned reset_counter; 312 313 if (!fpriv) 314 return -EINVAL; 315 316 mgr = &fpriv->ctx_mgr; 317 mutex_lock(&mgr->lock); 318 ctx = idr_find(&mgr->ctx_handles, id); 319 if (!ctx) { 320 mutex_unlock(&mgr->lock); 321 return -EINVAL; 322 } 323 324 /* TODO: these two are always zero */ 325 out->state.flags = 0x0; 326 out->state.hangs = 0x0; 327 328 /* determine if a GPU reset has occured since the last call */ 329 reset_counter = atomic_read(&adev->gpu_reset_counter); 330 /* TODO: this should ideally return NO, GUILTY, or INNOCENT. */ 331 if (ctx->reset_counter_query == reset_counter) 332 out->state.reset_status = AMDGPU_CTX_NO_RESET; 333 else 334 out->state.reset_status = AMDGPU_CTX_UNKNOWN_RESET; 335 ctx->reset_counter_query = reset_counter; 336 337 mutex_unlock(&mgr->lock); 338 return 0; 339 } 340 341 static int amdgpu_ctx_query2(struct amdgpu_device *adev, 342 struct amdgpu_fpriv *fpriv, uint32_t id, 343 union drm_amdgpu_ctx_out *out) 344 { 345 struct amdgpu_ctx *ctx; 346 struct amdgpu_ctx_mgr *mgr; 347 uint32_t ras_counter; 348 349 if (!fpriv) 350 return -EINVAL; 351 352 mgr = &fpriv->ctx_mgr; 353 mutex_lock(&mgr->lock); 354 ctx = idr_find(&mgr->ctx_handles, id); 355 if (!ctx) { 356 mutex_unlock(&mgr->lock); 357 return -EINVAL; 358 } 359 360 out->state.flags = 0x0; 361 out->state.hangs = 0x0; 362 363 if (ctx->reset_counter != atomic_read(&adev->gpu_reset_counter)) 364 out->state.flags |= AMDGPU_CTX_QUERY2_FLAGS_RESET; 365 366 if (ctx->vram_lost_counter != atomic_read(&adev->vram_lost_counter)) 367 out->state.flags |= AMDGPU_CTX_QUERY2_FLAGS_VRAMLOST; 368 369 if (atomic_read(&ctx->guilty)) 370 out->state.flags |= AMDGPU_CTX_QUERY2_FLAGS_GUILTY; 371 372 /*query ue count*/ 373 ras_counter = amdgpu_ras_query_error_count(adev, false); 374 /*ras counter is monotonic increasing*/ 375 if (ras_counter != ctx->ras_counter_ue) { 376 out->state.flags |= AMDGPU_CTX_QUERY2_FLAGS_RAS_UE; 377 ctx->ras_counter_ue = ras_counter; 378 } 379 380 /*query ce count*/ 381 ras_counter = amdgpu_ras_query_error_count(adev, true); 382 if (ras_counter != ctx->ras_counter_ce) { 383 out->state.flags |= AMDGPU_CTX_QUERY2_FLAGS_RAS_CE; 384 ctx->ras_counter_ce = ras_counter; 385 } 386 387 mutex_unlock(&mgr->lock); 388 return 0; 389 } 390 391 int amdgpu_ctx_ioctl(struct drm_device *dev, void *data, 392 struct drm_file *filp) 393 { 394 int r; 395 uint32_t id; 396 enum drm_sched_priority priority; 397 398 union drm_amdgpu_ctx *args = data; 399 struct amdgpu_device *adev = dev->dev_private; 400 struct amdgpu_fpriv *fpriv = filp->driver_priv; 401 402 r = 0; 403 id = args->in.ctx_id; 404 priority = amdgpu_to_sched_priority(args->in.priority); 405 406 /* For backwards compatibility reasons, we need to accept 407 * ioctls with garbage in the priority field */ 408 if (priority == DRM_SCHED_PRIORITY_INVALID) 409 priority = DRM_SCHED_PRIORITY_NORMAL; 410 411 switch (args->in.op) { 412 case AMDGPU_CTX_OP_ALLOC_CTX: 413 r = amdgpu_ctx_alloc(adev, fpriv, filp, priority, &id); 414 args->out.alloc.ctx_id = id; 415 break; 416 case AMDGPU_CTX_OP_FREE_CTX: 417 r = amdgpu_ctx_free(fpriv, id); 418 break; 419 case AMDGPU_CTX_OP_QUERY_STATE: 420 r = amdgpu_ctx_query(adev, fpriv, id, &args->out); 421 break; 422 case AMDGPU_CTX_OP_QUERY_STATE2: 423 r = amdgpu_ctx_query2(adev, fpriv, id, &args->out); 424 break; 425 default: 426 return -EINVAL; 427 } 428 429 return r; 430 } 431 432 struct amdgpu_ctx *amdgpu_ctx_get(struct amdgpu_fpriv *fpriv, uint32_t id) 433 { 434 struct amdgpu_ctx *ctx; 435 struct amdgpu_ctx_mgr *mgr; 436 437 if (!fpriv) 438 return NULL; 439 440 mgr = &fpriv->ctx_mgr; 441 442 mutex_lock(&mgr->lock); 443 ctx = idr_find(&mgr->ctx_handles, id); 444 if (ctx) 445 kref_get(&ctx->refcount); 446 mutex_unlock(&mgr->lock); 447 return ctx; 448 } 449 450 int amdgpu_ctx_put(struct amdgpu_ctx *ctx) 451 { 452 if (ctx == NULL) 453 return -EINVAL; 454 455 kref_put(&ctx->refcount, amdgpu_ctx_do_release); 456 return 0; 457 } 458 459 void amdgpu_ctx_add_fence(struct amdgpu_ctx *ctx, 460 struct drm_sched_entity *entity, 461 struct dma_fence *fence, uint64_t* handle) 462 { 463 struct amdgpu_ctx_entity *centity = to_amdgpu_ctx_entity(entity); 464 uint64_t seq = centity->sequence; 465 struct dma_fence *other = NULL; 466 unsigned idx = 0; 467 468 idx = seq & (amdgpu_sched_jobs - 1); 469 other = centity->fences[idx]; 470 if (other) 471 BUG_ON(!dma_fence_is_signaled(other)); 472 473 dma_fence_get(fence); 474 475 spin_lock(&ctx->ring_lock); 476 centity->fences[idx] = fence; 477 centity->sequence++; 478 spin_unlock(&ctx->ring_lock); 479 480 dma_fence_put(other); 481 if (handle) 482 *handle = seq; 483 } 484 485 struct dma_fence *amdgpu_ctx_get_fence(struct amdgpu_ctx *ctx, 486 struct drm_sched_entity *entity, 487 uint64_t seq) 488 { 489 struct amdgpu_ctx_entity *centity = to_amdgpu_ctx_entity(entity); 490 struct dma_fence *fence; 491 492 spin_lock(&ctx->ring_lock); 493 494 if (seq == ~0ull) 495 seq = centity->sequence - 1; 496 497 if (seq >= centity->sequence) { 498 spin_unlock(&ctx->ring_lock); 499 return ERR_PTR(-EINVAL); 500 } 501 502 503 if (seq + amdgpu_sched_jobs < centity->sequence) { 504 spin_unlock(&ctx->ring_lock); 505 return NULL; 506 } 507 508 fence = dma_fence_get(centity->fences[seq & (amdgpu_sched_jobs - 1)]); 509 spin_unlock(&ctx->ring_lock); 510 511 return fence; 512 } 513 514 void amdgpu_ctx_priority_override(struct amdgpu_ctx *ctx, 515 enum drm_sched_priority priority) 516 { 517 unsigned num_entities = amdgput_ctx_total_num_entities(); 518 enum drm_sched_priority ctx_prio; 519 unsigned i; 520 521 ctx->override_priority = priority; 522 523 ctx_prio = (ctx->override_priority == DRM_SCHED_PRIORITY_UNSET) ? 524 ctx->init_priority : ctx->override_priority; 525 526 for (i = 0; i < num_entities; i++) { 527 struct drm_sched_entity *entity = &ctx->entities[0][i].entity; 528 529 drm_sched_entity_set_priority(entity, ctx_prio); 530 } 531 } 532 533 int amdgpu_ctx_wait_prev_fence(struct amdgpu_ctx *ctx, 534 struct drm_sched_entity *entity) 535 { 536 struct amdgpu_ctx_entity *centity = to_amdgpu_ctx_entity(entity); 537 struct dma_fence *other; 538 unsigned idx; 539 long r; 540 541 spin_lock(&ctx->ring_lock); 542 idx = centity->sequence & (amdgpu_sched_jobs - 1); 543 other = dma_fence_get(centity->fences[idx]); 544 spin_unlock(&ctx->ring_lock); 545 546 if (!other) 547 return 0; 548 549 r = dma_fence_wait(other, true); 550 if (r < 0 && r != -ERESTARTSYS) 551 DRM_ERROR("Error (%ld) waiting for fence!\n", r); 552 553 dma_fence_put(other); 554 return r; 555 } 556 557 void amdgpu_ctx_mgr_init(struct amdgpu_ctx_mgr *mgr) 558 { 559 mutex_init(&mgr->lock); 560 idr_init(&mgr->ctx_handles); 561 } 562 563 long amdgpu_ctx_mgr_entity_flush(struct amdgpu_ctx_mgr *mgr, long timeout) 564 { 565 unsigned num_entities = amdgput_ctx_total_num_entities(); 566 struct amdgpu_ctx *ctx; 567 struct idr *idp; 568 uint32_t id, i; 569 570 idp = &mgr->ctx_handles; 571 572 mutex_lock(&mgr->lock); 573 idr_for_each_entry(idp, ctx, id) { 574 for (i = 0; i < num_entities; i++) { 575 struct drm_sched_entity *entity; 576 577 entity = &ctx->entities[0][i].entity; 578 timeout = drm_sched_entity_flush(entity, timeout); 579 } 580 } 581 mutex_unlock(&mgr->lock); 582 return timeout; 583 } 584 585 void amdgpu_ctx_mgr_entity_fini(struct amdgpu_ctx_mgr *mgr) 586 { 587 unsigned num_entities = amdgput_ctx_total_num_entities(); 588 struct amdgpu_ctx *ctx; 589 struct idr *idp; 590 uint32_t id, i; 591 592 idp = &mgr->ctx_handles; 593 594 idr_for_each_entry(idp, ctx, id) { 595 if (kref_read(&ctx->refcount) != 1) { 596 DRM_ERROR("ctx %p is still alive\n", ctx); 597 continue; 598 } 599 600 for (i = 0; i < num_entities; i++) 601 drm_sched_entity_fini(&ctx->entities[0][i].entity); 602 } 603 } 604 605 void amdgpu_ctx_mgr_fini(struct amdgpu_ctx_mgr *mgr) 606 { 607 struct amdgpu_ctx *ctx; 608 struct idr *idp; 609 uint32_t id; 610 611 amdgpu_ctx_mgr_entity_fini(mgr); 612 613 idp = &mgr->ctx_handles; 614 615 idr_for_each_entry(idp, ctx, id) { 616 if (kref_put(&ctx->refcount, amdgpu_ctx_fini) != 1) 617 DRM_ERROR("ctx %p is still alive\n", ctx); 618 } 619 620 idr_destroy(&mgr->ctx_handles); 621 mutex_destroy(&mgr->lock); 622 } 623