Fix T41222 Blender gives weird output when baking (4096*4096) resolution on GPU
[blender-staging.git] / intern / cycles / device / device_cuda.cpp
index 6629069c6c675989c2731ae618224897cdd4c784..d76ffb1078671596609517060b6e420c784f4e70 100644 (file)
@@ -25,6 +25,7 @@
 
 #include "cuew.h"
 #include "util_debug.h"
+#include "util_foreach.h"
 #include "util_map.h"
 #include "util_opengl.h"
 #include "util_path.h"
@@ -966,7 +967,10 @@ public:
 
        int get_split_task_count(DeviceTask& task)
        {
-               return 1;
+               if (task.type == DeviceTask::SHADER)
+                       return task.get_subtask_count(TaskScheduler::num_threads(), 1024 * 1024);
+               else
+                       return 1;
        }
 
        void task_add(DeviceTask& task)
@@ -979,6 +983,15 @@ public:
                        cuda_assert(cuCtxSynchronize());
                        cuda_pop_context();
                }
+               else if(task.type == DeviceTask::SHADER) {
+                       /* split task into smaller ones */
+                       list<DeviceTask> tasks;
+
+                       task.split(tasks, TaskScheduler::num_threads(), 1024 * 1024);
+
+                       foreach(DeviceTask& task, tasks)
+                               task_pool.push(new CUDADeviceTask(this, task));
+               }
                else {
                        task_pool.push(new CUDADeviceTask(this, task));
                }