浏览代码

openmp: add preliminary support for places/partitions routines

Samuel Pitoiset 8 年之前
父节点
当前提交
72407e0967
共有 2 个文件被更改,包括 40 次插入0 次删除
  1. 6 0
      include/starpu_openmp.h
  2. 34 0
      src/util/openmp_runtime_support_omp_api.c

+ 6 - 0
include/starpu_openmp.h

@@ -141,6 +141,12 @@ extern int starpu_omp_get_team_size(int level) __STARPU_OMP_NOTHROW;
 extern int starpu_omp_get_active_level(void) __STARPU_OMP_NOTHROW;
 extern int starpu_omp_in_final(void) __STARPU_OMP_NOTHROW;
 extern enum starpu_omp_proc_bind_value starpu_omp_get_proc_bind(void) __STARPU_OMP_NOTHROW;
+extern int starpu_omp_get_num_places(void) __STARPU_OMP_NOTHROW;
+extern int starpu_omp_get_place_num_procs(int place_num) __STARPU_OMP_NOTHROW;
+extern void starpu_omp_get_place_proc_ids(int place_num, int *ids) __STARPU_OMP_NOTHROW;
+extern int starpu_omp_get_place_num(void) __STARPU_OMP_NOTHROW;
+extern int starpu_omp_get_partition_num_places(void) __STARPU_OMP_NOTHROW;
+extern void starpu_omp_get_partition_place_nums(int *place_nums) __STARPU_OMP_NOTHROW;
 extern void starpu_omp_set_default_device(int device_num) __STARPU_OMP_NOTHROW;
 extern int starpu_omp_get_default_device(void) __STARPU_OMP_NOTHROW;
 extern int starpu_omp_get_num_devices(void) __STARPU_OMP_NOTHROW;

+ 34 - 0
src/util/openmp_runtime_support_omp_api.c

@@ -216,6 +216,40 @@ enum starpu_omp_proc_bind_value starpu_omp_get_proc_bind(void)
 	return proc_bind;
 }
 
+int starpu_omp_get_num_places(void)
+{
+  /* TODO */
+  return 0;
+}
+
+int starpu_omp_get_place_num_procs(int place_num)
+{
+  /* TODO */
+  return 0;
+}
+
+void starpu_omp_get_place_proc_ids(int place_num, int *ids)
+{
+  /* TODO */
+}
+
+int starpu_omp_get_place_num(void)
+{
+  /* TODO */
+  return -1;
+}
+
+int starpu_omp_get_partition_num_places(void)
+{
+  /* TODO */
+  return 0;
+}
+
+void starpu_omp_get_partition_place_nums(int *place_nums)
+{
+  /* TODO */
+}
+
 void starpu_omp_set_default_device(int device_num)
 {
 	(void) device_num;