Browse Source

Fix crash on clearing current context

Samuel Thibault 6 years ago
parent
commit
07cf139b4b
1 changed files with 4 additions and 1 deletions
  1. 4 1
      src/core/sched_ctx.c

+ 4 - 1
src/core/sched_ctx.c

@@ -1661,7 +1661,10 @@ int _starpu_wait_for_no_ready_of_sched_ctx(unsigned sched_ctx_id)
  */
 void starpu_sched_ctx_set_context(unsigned *sched_ctx)
 {
-	STARPU_PTHREAD_SETSPECIFIC(sched_ctx_key, (void*)(uintptr_t)*sched_ctx);
+	if (sched_ctx)
+		STARPU_PTHREAD_SETSPECIFIC(sched_ctx_key, (void*)(uintptr_t)*sched_ctx);
+	else
+		STARPU_PTHREAD_SETSPECIFIC(sched_ctx_key, (void*)(uintptr_t)STARPU_NMAX_SCHED_CTXS);
 }
 
 unsigned starpu_sched_ctx_get_context()