1 /* 2 * Copyright 2014 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 */ 23 24 #include "kfd_mqd_manager.h" 25 #include "amdgpu_amdkfd.h" 26 #include "kfd_device_queue_manager.h" 27 28 /* Mapping queue priority to pipe priority, indexed by queue priority */ 29 int pipe_priority_map[] = { 30 KFD_PIPE_PRIORITY_CS_LOW, 31 KFD_PIPE_PRIORITY_CS_LOW, 32 KFD_PIPE_PRIORITY_CS_LOW, 33 KFD_PIPE_PRIORITY_CS_LOW, 34 KFD_PIPE_PRIORITY_CS_LOW, 35 KFD_PIPE_PRIORITY_CS_LOW, 36 KFD_PIPE_PRIORITY_CS_LOW, 37 KFD_PIPE_PRIORITY_CS_MEDIUM, 38 KFD_PIPE_PRIORITY_CS_MEDIUM, 39 KFD_PIPE_PRIORITY_CS_MEDIUM, 40 KFD_PIPE_PRIORITY_CS_MEDIUM, 41 KFD_PIPE_PRIORITY_CS_HIGH, 42 KFD_PIPE_PRIORITY_CS_HIGH, 43 KFD_PIPE_PRIORITY_CS_HIGH, 44 KFD_PIPE_PRIORITY_CS_HIGH, 45 KFD_PIPE_PRIORITY_CS_HIGH 46 }; 47 48 struct kfd_mem_obj *allocate_hiq_mqd(struct kfd_dev *dev, struct queue_properties *q) 49 { 50 struct kfd_mem_obj *mqd_mem_obj = NULL; 51 52 mqd_mem_obj = kzalloc(sizeof(struct kfd_mem_obj), GFP_KERNEL); 53 if (!mqd_mem_obj) 54 return NULL; 55 56 mqd_mem_obj->gtt_mem = dev->dqm->hiq_sdma_mqd.gtt_mem; 57 mqd_mem_obj->gpu_addr = dev->dqm->hiq_sdma_mqd.gpu_addr; 58 mqd_mem_obj->cpu_ptr = dev->dqm->hiq_sdma_mqd.cpu_ptr; 59 60 return mqd_mem_obj; 61 } 62 63 struct kfd_mem_obj *allocate_sdma_mqd(struct kfd_dev *dev, 64 struct queue_properties *q) 65 { 66 struct kfd_mem_obj *mqd_mem_obj = NULL; 67 uint64_t offset; 68 69 mqd_mem_obj = kzalloc(sizeof(struct kfd_mem_obj), GFP_KERNEL); 70 if (!mqd_mem_obj) 71 return NULL; 72 73 offset = (q->sdma_engine_id * 74 dev->device_info.num_sdma_queues_per_engine + 75 q->sdma_queue_id) * 76 dev->dqm->mqd_mgrs[KFD_MQD_TYPE_SDMA]->mqd_size; 77 78 offset += dev->dqm->mqd_mgrs[KFD_MQD_TYPE_HIQ]->mqd_size; 79 80 mqd_mem_obj->gtt_mem = (void *)((uint64_t)dev->dqm->hiq_sdma_mqd.gtt_mem 81 + offset); 82 mqd_mem_obj->gpu_addr = dev->dqm->hiq_sdma_mqd.gpu_addr + offset; 83 mqd_mem_obj->cpu_ptr = (uint32_t *)((uint64_t) 84 dev->dqm->hiq_sdma_mqd.cpu_ptr + offset); 85 86 return mqd_mem_obj; 87 } 88 89 void free_mqd_hiq_sdma(struct mqd_manager *mm, void *mqd, 90 struct kfd_mem_obj *mqd_mem_obj) 91 { 92 WARN_ON(!mqd_mem_obj->gtt_mem); 93 kfree(mqd_mem_obj); 94 } 95 96 void mqd_symmetrically_map_cu_mask(struct mqd_manager *mm, 97 const uint32_t *cu_mask, uint32_t cu_mask_count, 98 uint32_t *se_mask) 99 { 100 struct kfd_cu_info cu_info; 101 uint32_t cu_per_sh[KFD_MAX_NUM_SE][KFD_MAX_NUM_SH_PER_SE] = {0}; 102 int i, se, sh, cu; 103 amdgpu_amdkfd_get_cu_info(mm->dev->adev, &cu_info); 104 105 if (cu_mask_count > cu_info.cu_active_number) 106 cu_mask_count = cu_info.cu_active_number; 107 108 /* Exceeding these bounds corrupts the stack and indicates a coding error. 109 * Returning with no CU's enabled will hang the queue, which should be 110 * attention grabbing. 111 */ 112 if (cu_info.num_shader_engines > KFD_MAX_NUM_SE) { 113 pr_err("Exceeded KFD_MAX_NUM_SE, chip reports %d\n", cu_info.num_shader_engines); 114 return; 115 } 116 if (cu_info.num_shader_arrays_per_engine > KFD_MAX_NUM_SH_PER_SE) { 117 pr_err("Exceeded KFD_MAX_NUM_SH, chip reports %d\n", 118 cu_info.num_shader_arrays_per_engine * cu_info.num_shader_engines); 119 return; 120 } 121 /* Count active CUs per SH. 122 * 123 * Some CUs in an SH may be disabled. HW expects disabled CUs to be 124 * represented in the high bits of each SH's enable mask (the upper and lower 125 * 16 bits of se_mask) and will take care of the actual distribution of 126 * disabled CUs within each SH automatically. 127 * Each half of se_mask must be filled only on bits 0-cu_per_sh[se][sh]-1. 128 * 129 * See note on Arcturus cu_bitmap layout in gfx_v9_0_get_cu_info. 130 */ 131 for (se = 0; se < cu_info.num_shader_engines; se++) 132 for (sh = 0; sh < cu_info.num_shader_arrays_per_engine; sh++) 133 cu_per_sh[se][sh] = hweight32(cu_info.cu_bitmap[se % 4][sh + (se / 4)]); 134 135 /* Symmetrically map cu_mask to all SEs & SHs: 136 * se_mask programs up to 2 SH in the upper and lower 16 bits. 137 * 138 * Examples 139 * Assuming 1 SH/SE, 4 SEs: 140 * cu_mask[0] bit0 -> se_mask[0] bit0 141 * cu_mask[0] bit1 -> se_mask[1] bit0 142 * ... 143 * cu_mask[0] bit4 -> se_mask[0] bit1 144 * ... 145 * 146 * Assuming 2 SH/SE, 4 SEs 147 * cu_mask[0] bit0 -> se_mask[0] bit0 (SE0,SH0,CU0) 148 * cu_mask[0] bit1 -> se_mask[1] bit0 (SE1,SH0,CU0) 149 * ... 150 * cu_mask[0] bit4 -> se_mask[0] bit16 (SE0,SH1,CU0) 151 * cu_mask[0] bit5 -> se_mask[1] bit16 (SE1,SH1,CU0) 152 * ... 153 * cu_mask[0] bit8 -> se_mask[0] bit1 (SE0,SH0,CU1) 154 * ... 155 * 156 * First ensure all CUs are disabled, then enable user specified CUs. 157 */ 158 for (i = 0; i < cu_info.num_shader_engines; i++) 159 se_mask[i] = 0; 160 161 i = 0; 162 for (cu = 0; cu < 16; cu++) { 163 for (sh = 0; sh < cu_info.num_shader_arrays_per_engine; sh++) { 164 for (se = 0; se < cu_info.num_shader_engines; se++) { 165 if (cu_per_sh[se][sh] > cu) { 166 if (cu_mask[i / 32] & (1 << (i % 32))) 167 se_mask[se] |= 1 << (cu + sh * 16); 168 i++; 169 if (i == cu_mask_count) 170 return; 171 } 172 } 173 } 174 } 175 } 176 177 int kfd_hiq_load_mqd_kiq(struct mqd_manager *mm, void *mqd, 178 uint32_t pipe_id, uint32_t queue_id, 179 struct queue_properties *p, struct mm_struct *mms) 180 { 181 return mm->dev->kfd2kgd->hiq_mqd_load(mm->dev->adev, mqd, pipe_id, 182 queue_id, p->doorbell_off); 183 } 184 185 int kfd_destroy_mqd_cp(struct mqd_manager *mm, void *mqd, 186 enum kfd_preempt_type type, unsigned int timeout, 187 uint32_t pipe_id, uint32_t queue_id) 188 { 189 return mm->dev->kfd2kgd->hqd_destroy(mm->dev->adev, mqd, type, timeout, 190 pipe_id, queue_id); 191 } 192 193 void kfd_free_mqd_cp(struct mqd_manager *mm, void *mqd, 194 struct kfd_mem_obj *mqd_mem_obj) 195 { 196 if (mqd_mem_obj->gtt_mem) { 197 amdgpu_amdkfd_free_gtt_mem(mm->dev->adev, mqd_mem_obj->gtt_mem); 198 kfree(mqd_mem_obj); 199 } else { 200 kfd_gtt_sa_free(mm->dev, mqd_mem_obj); 201 } 202 } 203 204 bool kfd_is_occupied_cp(struct mqd_manager *mm, void *mqd, 205 uint64_t queue_address, uint32_t pipe_id, 206 uint32_t queue_id) 207 { 208 return mm->dev->kfd2kgd->hqd_is_occupied(mm->dev->adev, queue_address, 209 pipe_id, queue_id); 210 } 211 212 int kfd_load_mqd_sdma(struct mqd_manager *mm, void *mqd, 213 uint32_t pipe_id, uint32_t queue_id, 214 struct queue_properties *p, struct mm_struct *mms) 215 { 216 return mm->dev->kfd2kgd->hqd_sdma_load(mm->dev->adev, mqd, 217 (uint32_t __user *)p->write_ptr, 218 mms); 219 } 220 221 /* 222 * preempt type here is ignored because there is only one way 223 * to preempt sdma queue 224 */ 225 int kfd_destroy_mqd_sdma(struct mqd_manager *mm, void *mqd, 226 enum kfd_preempt_type type, 227 unsigned int timeout, uint32_t pipe_id, 228 uint32_t queue_id) 229 { 230 return mm->dev->kfd2kgd->hqd_sdma_destroy(mm->dev->adev, mqd, timeout); 231 } 232 233 bool kfd_is_occupied_sdma(struct mqd_manager *mm, void *mqd, 234 uint64_t queue_address, uint32_t pipe_id, 235 uint32_t queue_id) 236 { 237 return mm->dev->kfd2kgd->hqd_sdma_is_occupied(mm->dev->adev, mqd); 238 } 239