gru_assign_gru_context
while (!gru_assign_gru_context(kgts)) {
if (!gru_assign_gru_context(gts)) {
extern struct gru_state *gru_assign_gru_context(struct gru_thread_state *gts);