소스 검색

add global for model

Corentin Salingue 8 년 전
부모
커밋
200c25371d
2개의 변경된 파일6개의 추가작업 그리고 0개의 파일을 삭제
  1. 5 0
      src/core/perfmodel/perfmodel.c
  2. 1 0
      src/starpu_parameters.h

+ 5 - 0
src/core/perfmodel/perfmodel.c

@@ -123,6 +123,8 @@ double starpu_worker_get_relative_speedup(struct starpu_perfmodel_arch* perf_arc
 			coef = _STARPU_MIC_ALPHA;
 		else if (perf_arch->devices[dev].type == STARPU_SCC_WORKER)
 			coef = _STARPU_SCC_ALPHA;
+		else if (perf_arch->devices[dev].type == STARPU_MPI_WORKER)
+			coef = _STARPU_MPI_MS_ALPHA;
 
 		speedup += coef * (perf_arch->devices[dev].ncores);
 	}
@@ -263,6 +265,9 @@ double starpu_task_expected_conversion_time(struct starpu_task *task,
 			case STARPU_SCC_WORKER:
 				node_kind = STARPU_SCC_RAM;
 				break;
+			case STARPU_MPI_WORKER:
+				node_kind = STARPU_MPI_MS_RAM;
+				break;
 			default:
 				STARPU_ABORT();
 				break;

+ 1 - 0
src/starpu_parameters.h

@@ -27,4 +27,5 @@
 #define _STARPU_OPENCL_ALPHA	12.22f
 #define _STARPU_MIC_ALPHA	0.5f
 #define _STARPU_SCC_ALPHA	1.0f
+#define _STARPU_MPI_MS_ALPHA	1.0f
 #endif /* _STARPU_PARAMETERS_H */