c++枚举类型StarPU实现矩阵乘

1) 枚举类型 

#pragma once
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<windows.h>
#include<iostream>
#include <time.h>enum Color {RED,				//RED=0GREEN,		//GREEN=1BLUE			//BLUE=2
};
int main()
{//定义数据类型为枚举类型Color的数组enum Color sky[3] = { RED,RED,RED};for (int i = 0; i < 3; i++) {printf("%d\n",sky[i]);}
}

2) StarPU实现矩阵乘 

#include <starpu.h>#define N 100// 定义在CPU上执行的任务
static void cpu_task(void *buffers[], void *cl_arg) {// 获取输入和输出矩阵float *A = (float *)STARPU_VECTOR_GET_PTR(buffers[0]);float *B = (float *)STARPU_VECTOR_GET_PTR(buffers[1]);float *C = (float *)STARPU_VECTOR_GET_PTR(buffers[2]);// 执行矩阵乘法for (int i = 0; i < N; i++) {for (int j = 0; j < N; j++) {C[i * N + j] = 0;for (int k = 0; k < N; k++) {C[i * N + j] += A[i * N + k] * B[k * N + j];}}}
}                        
// 定义在GPU上执行的任务
static void gpu_task(void *buffers[], void *cl_arg) {// 获取输入和输出矩阵float *A = (float *)STARPU_VECTOR_GET_PTR(buffers[0]);float *B = (float *)STARPU_VECTOR_GET_PTR(buffers[1]);float *C = (float *)STARPU_VECTOR_GET_PTR(buffers[2]);// 使用CUDA进行矩阵乘法// 这里省略具体的CUDA代码
}int main(int argc, char **argv) {// 初始化StarPUstarpu_init(NULL);// 分配数据float *A = (float *)malloc(N * sizeof(float));float *B = (float *)malloc(N * sizeof(float));float *C = (float *)malloc(N * sizeof(float));// 创建句柄starpu_data_handle_t handle_A, handle_B, handle_C;starpu_vector_data_register(&handle_A, STARPU_MAIN_RAM, (uintptr_t)A, N, sizeof(float));starpu_vector_data_register(&handle_B, STARPU_MAIN_RAM, (uintptr_t)B, N, sizeof(float));starpu_vector_data_register(&handle_C, STARPU_MAIN_RAM, (uintptr_t)C, N, sizeof(float));// 定义任务代码struct starpu_codelet cl;starpu_codelet_init(&cl);cl.where = STARPU_CPU | STARPU_CUDA;cl.cpu_funcs[0] = cpu_task;cl.cuda_funcs[0] = gpu_task;cl.nbuffers = 3;cl.modes[0] = STARPU_R;cl.modes[1] = STARPU_R;cl.modes[2] = STARPU_W;// 创建任务struct starpu_task *task = starpu_task_create();task->cl = &cl;//向缓冲区中填充数据task->handles[0] = handle_A;task->handles[1] = handle_B;task->handles[2] = handle_C;// 提交任务starpu_task_submit(task);// 等待任务完成starpu_task_wait_for_all();// 释放句柄starpu_data_unregister(handle_A);starpu_data_unregister(handle_B);starpu_data_unregister(handle_C);// 清理free(A);free(B);free(C);starpu_shutdown();return 0;
}