Mercurial > hg > Game > Cerium

annotate TaskManager/Cuda/CudaScheduler.h @ 1924:14f9fc88872c draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
fix
author Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
date Thu, 30 Jan 2014 16:22:51 +0900
parents e801016bd47c
children cd5bbd8ec5d6
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1908
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
1 #ifndef INCLUDE_CUDA_SCHEDULER
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
2 #define INCLUDE_CUDA_SCHEDULER
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
3
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
4 #include "MainScheduler.h"
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
5 #include "FifoDmaManager.h"
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
6 #include "CudaThreads.h"
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
7 #include "HTask.h"
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
8 #include "TaskManager.h"
1915
effb5653fd5c update cuda, yet running
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1908
diff changeset
9 #include <cuda.h>
effb5653fd5c update cuda, yet running
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1908
diff changeset
10 #include <cuda_runtime.h>
effb5653fd5c update cuda, yet running
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1908
diff changeset
11
1908
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
12
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
13 extern TaskObject cuda_task_list[MAX_TASK_OBJECT];
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
14
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
15 class CudaScheduler : public MainScheduler {
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
16 public:
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
17 typedef struct cudabuffer {
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
18 int allcate_size;
1924
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1923
diff changeset
19 int in_size;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1923
diff changeset
20 int out_size;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1923
diff changeset
21 CUdeviceptr* memin;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1923
diff changeset
22 CUdeviceptr* memout;
1908
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
23 CUevent* event;
1924
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1923
diff changeset
24 CUstream stream;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1923
diff changeset
25 } CudaBuffer, *CudaBufferPtr;
1908
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
26 CudaScheduler();
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
27 virtual ~CudaScheduler();
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
28 void init_gpu();
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
29 void wait_for_event(CUevent* event, CudaBufferPtr m, TaskListPtr taskList, int cur);
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
30 void run();
1924
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1923
diff changeset
31
1908
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
32 // platform platform;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
33 // platform は OpenCL が複数のメーカーの GPU に対応してるから必要
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
34 // Cuda の場合、NVIDIA だけなので必要ない?
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
35 CUdevice device;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
36 unsigned int ret_num_platforms; // たぶん要らない
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
37 unsigned int ret_num_devices;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
38 CUcontext context;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
39 // command_queue command_queue;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
40 // Cuda には command_queue に相当するものはない
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
41 // Closest approximation would be the CUDA Stream mechanism. らしい...
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
42 int ret;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
43 memaddr reply;
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
44 // cl_kernel に相当
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
45 // 変数名は function にすべきか kernel にすべきか
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
46 // とりあえず、kernel で
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
47 CUfunction kernel[2];
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
48 CUevent kernel_event[2];
1924
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents: 1923
diff changeset
49 CudaBuffer* cudabuffer;
1908
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
50 HTask::htask_flag[2];
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
51
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
52 privete:
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
53 int load_kernel(int cmd);
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
54 CUdeviceptr createBuffer(CudaBufferPtr m, int i, CUcontext context, /* mem_flag mem_flag, */size_t size, int* error);
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
55 void initCudaBuffer(CudaBufferPtr m);
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
56 void destroyCudaBuffer(CudaBufferPtr m);
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
57 void CudaTaskError(int cur, TaskListPtr taskList, int ret);
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
58 };
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
59
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
60 #define CudaSchedRegister(str, filename, functionname) \
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
61 cuda_register_task(str, filename, functionname);
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff changeset
62 #endif