|
@@ -53,9 +53,10 @@ static void memset_cuda(void *descr[], void *arg)
|
|
|
|
|
|
int ret;
|
|
|
|
|
|
-
|
|
|
+#ifdef STARPU_USE_OPENCL
|
|
|
extern void memset0_opencl(void *buffers[], void *args);
|
|
|
extern void memset_opencl(void *buffers[], void *args);
|
|
|
+#endif
|
|
|
|
|
|
void memset0_cpu(void *descr[], void *arg)
|
|
|
{
|
|
@@ -103,8 +104,10 @@ static struct starpu_codelet memset_cl =
|
|
|
.cuda_funcs = {memset_cuda},
|
|
|
.cuda_flags = {STARPU_CUDA_ASYNC},
|
|
|
#endif
|
|
|
+#ifdef STARPU_USE_OPENCL
|
|
|
.opencl_funcs = {memset0_opencl, memset_opencl},
|
|
|
.opencl_flags = {STARPU_OPENCL_ASYNC},
|
|
|
+#endif
|
|
|
.cpu_funcs = {memset0_cpu, memset_cpu},
|
|
|
.cpu_funcs_name = {"memset0_cpu", "memset_cpu"},
|
|
|
.model = &model,
|
|
@@ -118,8 +121,10 @@ static struct starpu_codelet nl_memset_cl =
|
|
|
.cuda_funcs = {memset_cuda},
|
|
|
.cuda_flags = {STARPU_CUDA_ASYNC},
|
|
|
#endif
|
|
|
+#ifdef STARPU_USE_OPENCL
|
|
|
.opencl_funcs = {memset0_opencl, memset_opencl},
|
|
|
.opencl_flags = {STARPU_OPENCL_ASYNC},
|
|
|
+#endif
|
|
|
.cpu_funcs = {memset0_cpu, memset_cpu},
|
|
|
.cpu_funcs_name = {"memset0_cpu", "memset_cpu"},
|
|
|
.model = &nl_model,
|
|
@@ -296,7 +301,9 @@ static void compare_performance(int size, struct starpu_codelet *codelet, struct
|
|
|
|
|
|
}
|
|
|
|
|
|
+#ifdef STARPU_USE_OPENCL
|
|
|
struct starpu_opencl_program opencl_program;
|
|
|
+#endif
|
|
|
|
|
|
int main(int argc, char **argv)
|
|
|
{
|
|
@@ -316,11 +323,15 @@ int main(int argc, char **argv)
|
|
|
ret = starpu_initialize(&conf, &argc, &argv);
|
|
|
if (ret == -ENODEV) return STARPU_TEST_SKIPPED;
|
|
|
STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
|
|
|
+#ifdef STARPU_USE_OPENCL
|
|
|
|
|
|
ret = starpu_opencl_load_opencl_from_file("/home/makni/makni/starpu.git/tests/perfmodels/opencl_memset_kernel_01.cl",
|
|
|
&opencl_program, NULL);
|
|
|
STARPU_CHECK_RETURN_VALUE(ret, "starpu_opencl_load_opencl_from_file");
|
|
|
|
|
|
+#endif
|
|
|
+
|
|
|
+
|
|
|
int size;
|
|
|
for (size = STARTlin; size < END; size *= 2)
|
|
|
{
|
|
@@ -381,9 +392,13 @@ int main(int argc, char **argv)
|
|
|
starpu_data_unregister(handle);
|
|
|
free(dummy_buffer);
|
|
|
|
|
|
+#ifdef STARPU_USE_OPENCL
|
|
|
|
|
|
ret = starpu_opencl_unload_opencl(&opencl_program);
|
|
|
STARPU_CHECK_RETURN_VALUE(ret, "starpu_opencl_unload_opencl");
|
|
|
+
|
|
|
+#endif
|
|
|
+
|
|
|
starpu_shutdown();
|
|
|
|
|
|
return EXIT_SUCCESS;
|