Mercurial > hg > Game > Cerium
annotate TaskManager/Cuda/CudaScheduler.h @ 1965:59105550c175 draft
fix CudaScheduler. remove event.
| author | Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp> |
|---|---|
| date | Thu, 13 Feb 2014 00:29:49 +0900 |
| parents | 6988e5478a8c |
| children | c3b4083c4467 |
| rev | line source |
|---|---|
| 1908 | 1 #ifndef INCLUDE_CUDA_SCHEDULER |
| 2 #define INCLUDE_CUDA_SCHEDULER | |
| 3 | |
| 4 #include "MainScheduler.h" | |
| 5 #include "FifoDmaManager.h" | |
| 6 #include "CudaThreads.h" | |
| 7 #include "HTask.h" | |
| 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> |
| 1908 | 10 |
| 11 extern TaskObject cuda_task_list[MAX_TASK_OBJECT]; | |
| 12 | |
|
1935
67e50779feb4
CudaScheduler is runnig.
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1927
diff
changeset
|
13 #define STAGE 8 |
|
67e50779feb4
CudaScheduler is runnig.
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1927
diff
changeset
|
14 |
| 1908 | 15 class CudaScheduler : public MainScheduler { |
| 16 public: | |
| 17 typedef struct cudabuffer { | |
| 18 int allcate_size; | |
| 1924 | 19 int in_size; |
| 20 int out_size; | |
| 21 CUdeviceptr* memin; | |
| 22 CUdeviceptr* memout; | |
| 23 CUstream stream; | |
|
1963
6988e5478a8c
fix CudaScheduler
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1935
diff
changeset
|
24 void** kernelParams; |
| 1924 | 25 } CudaBuffer, *CudaBufferPtr; |
| 1908 | 26 CudaScheduler(); |
| 27 virtual ~CudaScheduler(); | |
| 28 void init_gpu(); | |
|
1965
59105550c175
fix CudaScheduler. remove event.
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1963
diff
changeset
|
29 void wait_for_event(CudaBufferPtr m, TaskListPtr taskList, int cur); |
| 1908 | 30 void run(); |
| 1924 | 31 |
| 1908 | 32 // platform platform; |
| 33 // platform は OpenCL が複数のメーカーの GPU に対応してるから必要 | |
| 34 // Cuda の場合、NVIDIA だけなので必要ない? | |
| 35 CUdevice device; | |
| 36 unsigned int ret_num_platforms; // たぶん要らない | |
|
1925
cd5bbd8ec5d6
fix CudaScheduler
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1924
diff
changeset
|
37 int ret_num_devices; |
| 1908 | 38 CUcontext context; |
| 39 // command_queue command_queue; | |
| 40 // Cuda には command_queue に相当するものはない | |
| 41 // Closest approximation would be the CUDA Stream mechanism. らしい... | |
| 42 int ret; | |
| 43 memaddr reply; | |
| 44 // cl_kernel に相当 | |
| 45 // 変数名は function にすべきか kernel にすべきか | |
| 46 // とりあえず、kernel で | |
|
1935
67e50779feb4
CudaScheduler is runnig.
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1927
diff
changeset
|
47 CUfunction kernel[STAGE]; |
|
67e50779feb4
CudaScheduler is runnig.
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1927
diff
changeset
|
48 CudaBuffer cudabuffer[STAGE]; |
|
67e50779feb4
CudaScheduler is runnig.
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1927
diff
changeset
|
49 |
|
67e50779feb4
CudaScheduler is runnig.
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1927
diff
changeset
|
50 HTask::htask_flag flag[STAGE]; |
| 1908 | 51 |
|
1925
cd5bbd8ec5d6
fix CudaScheduler
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1924
diff
changeset
|
52 private: |
| 1908 | 53 int load_kernel(int cmd); |
|
1935
67e50779feb4
CudaScheduler is runnig.
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1927
diff
changeset
|
54 void createBuffer(CudaBufferPtr cudabuffer, CUdeviceptr* mem, int i, size_t size); |
| 1908 | 55 void initCudaBuffer(CudaBufferPtr m); |
| 56 void destroyCudaBuffer(CudaBufferPtr m); | |
|
1925
cd5bbd8ec5d6
fix CudaScheduler
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
1924
diff
changeset
|
57 void CudaTaskError(CudaBufferPtr cudabuffer, int cur, TaskListPtr taskList, int ret); |
| 1908 | 58 }; |
| 59 | |
| 60 #define CudaSchedRegister(str, filename, functionname) \ | |
| 61 cuda_register_task(str, filename, functionname); | |
| 62 #endif | |
| 1927 | 63 |
| 64 extern void cuda_register_task(int cmd, const char* filename, const char* functionname); |