155 	struct gru_thread_state *kgts;  in gru_load_kernel_context()  local
171 	kgts = bs->bs_kgts;  in gru_load_kernel_context()
173 	if (!kgts->ts_gru) {  in gru_load_kernel_context()
176 		kgts->ts_cbr_au_count = GRU_CB_COUNT_TO_AU(  in gru_load_kernel_context()
178 		kgts->ts_dsr_au_count = GRU_DS_BYTES_TO_AU(  in gru_load_kernel_context()
181 		while (!gru_assign_gru_context(kgts)) {  in gru_load_kernel_context()
183 			gru_steal_context(kgts);  in gru_load_kernel_context()
185 		gru_load_context(kgts);  in gru_load_kernel_context()
188 		ctxnum = kgts->ts_ctxnum;  in gru_load_kernel_context()
202 	struct gru_thread_state *kgts;  in gru_free_kernel_contexts()  local
212 			kgts = bs->bs_kgts;  in gru_free_kernel_contexts()
213 			if (kgts && kgts->ts_gru)  in gru_free_kernel_contexts()
214 				gru_unload_context(kgts, 0);  in gru_free_kernel_contexts()
217 			kfree(kgts);  in gru_free_kernel_contexts()
306 	struct gru_thread_state *kgts;  in gru_reserve_async_resources()  local
319 	kgts = bs->bs_kgts;  in gru_reserve_async_resources()
322 	if (kgts && kgts->ts_gru)  in gru_reserve_async_resources()
323 		gru_unload_context(kgts, 0);  in gru_reserve_async_resources()
403 	struct gru_thread_state *kgts = NULL;  in gru_get_cb_exception_detail()  local
415 		kgts = gru_base[bid]->bs_kgts;  in gru_get_cb_exception_detail()
416 		if (!kgts || !kgts->ts_gru)  in gru_get_cb_exception_detail()
418 		off = cb - kgts->ts_gru->gs_gru_base_vaddr;  in gru_get_cb_exception_detail()
421 		kgts = NULL;  in gru_get_cb_exception_detail()
423 	BUG_ON(!kgts);  in gru_get_cb_exception_detail()
424 	cbrnum = thread_cbr_number(kgts, get_cb_number(cb));  in gru_get_cb_exception_detail()