155 	struct gru_thread_state *kgts;  in gru_load_kernel_context()  local
166 	kgts = bs->bs_kgts;  in gru_load_kernel_context()
168 	if (!kgts->ts_gru) {  in gru_load_kernel_context()
171 		kgts->ts_cbr_au_count = GRU_CB_COUNT_TO_AU(  in gru_load_kernel_context()
173 		kgts->ts_dsr_au_count = GRU_DS_BYTES_TO_AU(  in gru_load_kernel_context()
176 		while (!gru_assign_gru_context(kgts)) {  in gru_load_kernel_context()
178 			gru_steal_context(kgts);  in gru_load_kernel_context()
180 		gru_load_context(kgts);  in gru_load_kernel_context()
183 		ctxnum = kgts->ts_ctxnum;  in gru_load_kernel_context()
197 	struct gru_thread_state *kgts;  in gru_free_kernel_contexts()  local
207 			kgts = bs->bs_kgts;  in gru_free_kernel_contexts()
208 			if (kgts && kgts->ts_gru)  in gru_free_kernel_contexts()
209 				gru_unload_context(kgts, 0);  in gru_free_kernel_contexts()
212 			kfree(kgts);  in gru_free_kernel_contexts()
301 	struct gru_thread_state *kgts;  in gru_reserve_async_resources()  local
314 	kgts = bs->bs_kgts;  in gru_reserve_async_resources()
317 	if (kgts && kgts->ts_gru)  in gru_reserve_async_resources()
318 		gru_unload_context(kgts, 0);  in gru_reserve_async_resources()
398 	struct gru_thread_state *kgts = NULL;  in gru_get_cb_exception_detail()  local
410 		kgts = gru_base[bid]->bs_kgts;  in gru_get_cb_exception_detail()
411 		if (!kgts || !kgts->ts_gru)  in gru_get_cb_exception_detail()
413 		off = cb - kgts->ts_gru->gs_gru_base_vaddr;  in gru_get_cb_exception_detail()
416 		kgts = NULL;  in gru_get_cb_exception_detail()
418 	BUG_ON(!kgts);  in gru_get_cb_exception_detail()
419 	cbrnum = thread_cbr_number(kgts, get_cb_number(cb));  in gru_get_cb_exception_detail()