Add function starpu_sched_ctx_set_user_data()

(partially cherry picked from commit f88dc05a29e9adf998d78682c88bc461e4f550d3)

include/starpu_sched_ctx.h

@@ -111,6 +111,8 @@ int starpu_sched_ctx_max_priority_is_set(unsigned sched_ctx_id);
 
 void *starpu_sched_ctx_get_user_data(unsigned sched_ctx_id);
 
+void starpu_sched_ctx_set_user_data(unsigned sched_ctx_id, void* user_data);
+
 struct starpu_worker_collection *starpu_sched_ctx_create_worker_collection(unsigned sched_ctx_id, enum starpu_worker_collection_type type) STARPU_ATTRIBUTE_MALLOC;
 
 void starpu_sched_ctx_delete_worker_collection(unsigned sched_ctx_id);

src/core/sched_ctx.c

@@ -2649,6 +2649,13 @@ void *starpu_sched_ctx_get_user_data(unsigned sched_ctx_id)
 	return sched_ctx->user_data;
 }
 
+void starpu_sched_ctx_set_user_data(unsigned sched_ctx_id, void* user_data)
+{
+	struct _starpu_sched_ctx *sched_ctx = _starpu_get_sched_ctx_struct(sched_ctx_id);
+	STARPU_ASSERT(sched_ctx != NULL);
+	sched_ctx->user_data = user_data;
+}
+
 void _starpu_worker_apply_deferred_ctx_changes(void)
 {
 	int workerid = starpu_worker_get_id_check();