drewcassidy/nvidia-texture-tools
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add tbb code.

Browse Source
This commit is contained in:
castano 2010-12-09 20:57:41 +00:00
parent f4cf5dd32c
commit 8d99f2c6c4
1 changed files with 28 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/nvtt/TaskDispatcher.h
View File

@ -14,11 +14,14 @@
#define HAVE_PPL 1 #define HAVE_PPL 1
#endif #endif
#if HAVE_PPL #if defined(HAVE_PPL)
#include <array> #include <array>
//#include <ppl.h> //#include <ppl.h>
#endif #endif
#if defined(HAVE_TBB)
#include <tbb/parallel_for.h>
#endif
namespace nvtt { namespace nvtt {
@ -50,12 +53,9 @@ namespace nvtt {
struct OpenMPTaskDispatcher : public TaskDispatcher struct OpenMPTaskDispatcher : public TaskDispatcher
{ {
virtual void dispatch(Task * task, void * context, size_t count) { virtual void dispatch(Task * task, void * context, size_t count) {
#pragma omp parallel #pragma omp parallel for
{ for (size_t i = 0; i < count; i++) {
#pragma omp for task(context, i);
for (size_t i = 0; i < count; i++) {
task(context, i);
}
} }
} }
}; };
@ -83,7 +83,7 @@ namespace nvtt {
struct TaskFunctor { struct TaskFunctor {
TaskFunctor(Task * task, void * context) : task(task), context(context) {} TaskFunctor(Task * task, void * context) : task(task), context(context) {}
void operator()(int & n) const { void operator()(int & n) const {
n *= n; task(context, n);
} }
Task * task; Task * task;
void * context; void * context;
@ -105,5 +105,25 @@ namespace nvtt {
#endif #endif
#if defined(HAVE_TBB)
struct TaskFunctor {
TaskFunctor(Task * task, void * context) : task(task), context(context) {}
void operator()(int & n) const {
task(context, n);
}
Task * task;
void * context;
};
struct IntelTaskDispatcher : public TaskDispatcher
{
virtual void dispatch(Task * task, void * context, size_t count) {
parallel_for(blocked_range<size_t>(0, count, 1), TaskFunctor(task, context));
}
};
#endif
} // namespace nvtt } // namespace nvtt