rsCpuCore.cpp revision 14ce007a633b10e3b9a3fae29d8f53a7e8c9b59f
1709a0978ae141198018ca9769f8d96292a8928e6Jason Sams/*
2709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Copyright (C) 2012 The Android Open Source Project
3709a0978ae141198018ca9769f8d96292a8928e6Jason Sams *
4709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Licensed under the Apache License, Version 2.0 (the "License");
5709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * you may not use this file except in compliance with the License.
6709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * You may obtain a copy of the License at
7709a0978ae141198018ca9769f8d96292a8928e6Jason Sams *
8709a0978ae141198018ca9769f8d96292a8928e6Jason Sams *      http://www.apache.org/licenses/LICENSE-2.0
9709a0978ae141198018ca9769f8d96292a8928e6Jason Sams *
10709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Unless required by applicable law or agreed to in writing, software
11709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * distributed under the License is distributed on an "AS IS" BASIS,
12709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * See the License for the specific language governing permissions and
14709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * limitations under the License.
15709a0978ae141198018ca9769f8d96292a8928e6Jason Sams */
16709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
17709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuCore.h"
18709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuScript.h"
19709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuScriptGroup.h"
201ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni#include "rsCpuScriptGroup2.h"
21709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
22709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <malloc.h>
23709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsContext.h"
24709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
25709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/types.h>
26709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/resource.h>
27709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sched.h>
28709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/syscall.h>
2911fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala#include <stdio.h>
30709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <string.h>
31b0934b67b95cc27e2358c2aa4db5f7c1067c8f9bStephen Hines#include <unistd.h>
320b575de8ed0b628d84d256f5846500b0385979bdTim Murray
33b0934b67b95cc27e2358c2aa4db5f7c1067c8f9bStephen Hines#if !defined(RS_SERVER) && !defined(RS_COMPATIBILITY_LIB)
340b575de8ed0b628d84d256f5846500b0385979bdTim Murray#include <cutils/properties.h>
35709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "utils/StopWatch.h"
360b575de8ed0b628d84d256f5846500b0385979bdTim Murray#endif
370b575de8ed0b628d84d256f5846500b0385979bdTim Murray
380b575de8ed0b628d84d256f5846500b0385979bdTim Murray#ifdef RS_SERVER
390b575de8ed0b628d84d256f5846500b0385979bdTim Murray// Android exposes gettid(), standard Linux does not
400b575de8ed0b628d84d256f5846500b0385979bdTim Murraystatic pid_t gettid() {
410b575de8ed0b628d84d256f5846500b0385979bdTim Murray    return syscall(SYS_gettid);
420b575de8ed0b628d84d256f5846500b0385979bdTim Murray}
430b575de8ed0b628d84d256f5846500b0385979bdTim Murray#endif
44709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
45709a0978ae141198018ca9769f8d96292a8928e6Jason Samsusing namespace android;
46709a0978ae141198018ca9769f8d96292a8928e6Jason Samsusing namespace android::renderscript;
47709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
48709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_key_t gThreadTLSKey = 0;
49709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic uint32_t gThreadTLSKeyCount = 0;
50709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_mutex_t gInitMutex = PTHREAD_MUTEX_INITIALIZER;
51709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
52f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Samsbool android::renderscript::gArchUseSIMD = false;
53f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
54709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::~RsdCpuReference() {
55709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
56709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
57709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference * RsdCpuReference::create(Context *rsc, uint32_t version_major,
58cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams        uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn
59b043df0676fef226336deb3a00ead2f31e02343fDavid Gross        , RSSelectRTCallback pSelectRTCallback,
60005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines        const char *pBccPluginName
61cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams        ) {
62709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
63709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    RsdCpuReferenceImpl *cpu = new RsdCpuReferenceImpl(rsc);
64709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (!cpu) {
6544bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        return nullptr;
66709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
67709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (!cpu->init(version_major, version_minor, lfn, slfn)) {
68709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        delete cpu;
6944bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        return nullptr;
70709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
71f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines
721d476620399d54774e4fd386c1d23cc583d49522Stephen Hines    cpu->setSelectRTCallback(pSelectRTCallback);
73005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines    if (pBccPluginName) {
74005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines        cpu->setBccPluginName(pBccPluginName);
75005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines    }
76f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines
77709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return cpu;
78709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
79709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
80709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
81709a0978ae141198018ca9769f8d96292a8928e6Jason SamsContext * RsdCpuReference::getTlsContext() {
82709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey);
83709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return tls->mContext;
84709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
85709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
86709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst Script * RsdCpuReference::getTlsScript() {
87709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey);
88709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return tls->mScript;
89709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
90709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
91f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hinespthread_key_t RsdCpuReference::getThreadTLSKey(){ return gThreadTLSKey; }
92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
93709a0978ae141198018ca9769f8d96292a8928e6Jason Sams////////////////////////////////////////////////////////////
94709a0978ae141198018ca9769f8d96292a8928e6Jason Sams///
95709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
96709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::RsdCpuReferenceImpl(Context *rsc) {
97709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mRSC = rsc;
98709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
99709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    version_major = 0;
100709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    version_minor = 0;
101709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mInForEach = false;
102709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    memset(&mWorkers, 0, sizeof(mWorkers));
103709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    memset(&mTlsStruct, 0, sizeof(mTlsStruct));
104709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mExit = false;
10544bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mSelectRTCallback = nullptr;
1068409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines    mEmbedGlobalInfo = true;
1078409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines    mEmbedGlobalInfoSkipConstant = true;
108709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
109709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
111709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid * RsdCpuReferenceImpl::helperThreadProc(void *vrsc) {
112709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    RsdCpuReferenceImpl *dc = (RsdCpuReferenceImpl *)vrsc;
113709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
1140b575de8ed0b628d84d256f5846500b0385979bdTim Murray    uint32_t idx = __sync_fetch_and_add(&dc->mWorkers.mLaunchCount, 1);
115709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
116709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    //ALOGV("RS helperThread starting %p idx=%i", dc, idx);
117709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
118709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    dc->mWorkers.mLaunchSignals[idx].init();
119709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    dc->mWorkers.mNativeThreadId[idx] = gettid();
120709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
121709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    memset(&dc->mTlsStruct, 0, sizeof(dc->mTlsStruct));
122709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    int status = pthread_setspecific(gThreadTLSKey, &dc->mTlsStruct);
123709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (status) {
124709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        ALOGE("pthread_setspecific %i", status);
125709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
126709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
127709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#if 0
128709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    typedef struct {uint64_t bits[1024 / 64]; } cpu_set_t;
129709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    cpu_set_t cpuset;
130709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    memset(&cpuset, 0, sizeof(cpuset));
131709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    cpuset.bits[idx / 64] |= 1ULL << (idx % 64);
132709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    int ret = syscall(241, rsc->mWorkers.mNativeThreadId[idx],
133709a0978ae141198018ca9769f8d96292a8928e6Jason Sams              sizeof(cpuset), &cpuset);
134709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    ALOGE("SETAFFINITY ret = %i %s", ret, EGLUtils::strerror(ret));
135709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif
136709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
137709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    while (!dc->mExit) {
138709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        dc->mWorkers.mLaunchSignals[idx].wait();
139709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        if (dc->mWorkers.mLaunchCallback) {
140709a0978ae141198018ca9769f8d96292a8928e6Jason Sams           // idx +1 is used because the calling thread is always worker 0.
141709a0978ae141198018ca9769f8d96292a8928e6Jason Sams           dc->mWorkers.mLaunchCallback(dc->mWorkers.mLaunchData, idx+1);
142709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        }
1430b575de8ed0b628d84d256f5846500b0385979bdTim Murray        __sync_fetch_and_sub(&dc->mWorkers.mRunningCount, 1);
144709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        dc->mWorkers.mCompleteSignal.set();
145709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
146709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
147709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    //ALOGV("RS helperThread exited %p idx=%i", dc, idx);
14844bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    return nullptr;
149709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
150709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
15114ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// Launch a kernel.
15214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// The callback function is called to execute the kernel.
153709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::launchThreads(WorkerCallback_t cbk, void *data) {
154709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mLaunchData = data;
155709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mLaunchCallback = cbk;
1564d252d6e807b89764dad123ac845df298c52ca97Tim Murray
1574d252d6e807b89764dad123ac845df298c52ca97Tim Murray    // fast path for very small launches
15814ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    MTLaunchStructCommon *mtls = (MTLaunchStructCommon *)data;
15914ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    if (mtls && mtls->dimPtr->y <= 1 && mtls->end.x <= mtls->start.x + mtls->mSliceSize) {
1604d252d6e807b89764dad123ac845df298c52ca97Tim Murray        if (mWorkers.mLaunchCallback) {
1614d252d6e807b89764dad123ac845df298c52ca97Tim Murray            mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0);
1624d252d6e807b89764dad123ac845df298c52ca97Tim Murray        }
1634d252d6e807b89764dad123ac845df298c52ca97Tim Murray        return;
1644d252d6e807b89764dad123ac845df298c52ca97Tim Murray    }
1654d252d6e807b89764dad123ac845df298c52ca97Tim Murray
1660b575de8ed0b628d84d256f5846500b0385979bdTim Murray    mWorkers.mRunningCount = mWorkers.mCount;
1670b575de8ed0b628d84d256f5846500b0385979bdTim Murray    __sync_synchronize();
1680b575de8ed0b628d84d256f5846500b0385979bdTim Murray
169709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) {
170709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        mWorkers.mLaunchSignals[ct].set();
171709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
172709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
173709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    // We use the calling thread as one of the workers so we can start without
174709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    // the delay of the thread wakeup.
175709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (mWorkers.mLaunchCallback) {
1764d252d6e807b89764dad123ac845df298c52ca97Tim Murray        mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0);
177709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
178709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
1790b575de8ed0b628d84d256f5846500b0385979bdTim Murray    while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) {
180709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        mWorkers.mCompleteSignal.wait();
181709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
182709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
183709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
184709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
185709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::lockMutex() {
186709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    pthread_mutex_lock(&gInitMutex);
187709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
188709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
189709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::unlockMutex() {
190709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    pthread_mutex_unlock(&gInitMutex);
191709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
192709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
19311fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala// Determine if the CPU we're running on supports SIMD instructions.
194f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Samsstatic void GetCpuInfo() {
19511fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala    // Read the CPU flags from /proc/cpuinfo.
19611fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala    FILE *cpuinfo = fopen("/proc/cpuinfo", "r");
197f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
19811fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala    if (!cpuinfo) {
199f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams        return;
200f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    }
201f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
20211fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala    char cpuinfostr[4096];
2035d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang    // fgets() ends with newline or EOF, need to check the whole
2045d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang    // "cpuinfo" file to make sure we can use SIMD or not.
2055d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang    while (fgets(cpuinfostr, sizeof(cpuinfostr), cpuinfo)) {
206074424a4ac5b093331df2c92e7a5bcbfff136b71Jason Sams#if defined(ARCH_ARM_HAVE_VFP) || defined(ARCH_ARM_USE_INTRINSICS)
2075d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang        gArchUseSIMD = strstr(cpuinfostr, " neon") || strstr(cpuinfostr, " asimd");
2087b7060c61e4182b29186849c5a857ea5f0898e56Rose, James#elif defined(ARCH_X86_HAVE_SSSE3)
2095d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang        gArchUseSIMD = strstr(cpuinfostr, " ssse3");
2107b7060c61e4182b29186849c5a857ea5f0898e56Rose, James#endif
2115d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang        if (gArchUseSIMD) {
2125d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang            break;
2135d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang        }
2145d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang    }
2155d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang    fclose(cpuinfo);
216f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams}
217f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
218709a0978ae141198018ca9769f8d96292a8928e6Jason Samsbool RsdCpuReferenceImpl::init(uint32_t version_major, uint32_t version_minor,
219709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                               sym_lookup_t lfn, script_lookup_t slfn) {
220709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mSymLookupFn = lfn;
221709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mScriptLookupFn = slfn;
222709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
223709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    lockMutex();
224709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (!gThreadTLSKeyCount) {
22544bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        int status = pthread_key_create(&gThreadTLSKey, nullptr);
226709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        if (status) {
227709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            ALOGE("Failed to init thread tls key.");
228709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            unlockMutex();
229709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            return false;
230709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        }
231709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
232709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    gThreadTLSKeyCount++;
233709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    unlockMutex();
234709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
235709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mTlsStruct.mContext = mRSC;
23644bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mTlsStruct.mScript = nullptr;
237709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    int status = pthread_setspecific(gThreadTLSKey, &mTlsStruct);
238709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (status) {
239709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        ALOGE("pthread_setspecific %i", status);
240709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
241709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
242f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    GetCpuInfo();
243f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
24477d57a305f4134e78ebc91869011c4009988104eJason Sams    int cpu = sysconf(_SC_NPROCESSORS_CONF);
245709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if(mRSC->props.mDebugMaxThreads) {
246709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        cpu = mRSC->props.mDebugMaxThreads;
247709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
248709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (cpu < 2) {
249709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        mWorkers.mCount = 0;
250709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        return true;
251709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
252709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
253709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    // Subtract one from the cpu count because we also use the command thread as a worker.
254709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mCount = (uint32_t)(cpu - 1);
255709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
2568ca358a2abe7e0dba23993e0fc8d64b8b55bd9caJason Sams    ALOGV("%p Launching thread(s), CPUs %i", mRSC, mWorkers.mCount + 1);
257709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
258709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mThreadId = (pthread_t *) calloc(mWorkers.mCount, sizeof(pthread_t));
259709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mNativeThreadId = (pid_t *) calloc(mWorkers.mCount, sizeof(pid_t));
260709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mLaunchSignals = new Signal[mWorkers.mCount];
26144bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mWorkers.mLaunchCallback = nullptr;
262709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
263709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mCompleteSignal.init();
264709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
2650b575de8ed0b628d84d256f5846500b0385979bdTim Murray    mWorkers.mRunningCount = mWorkers.mCount;
2660b575de8ed0b628d84d256f5846500b0385979bdTim Murray    mWorkers.mLaunchCount = 0;
2670b575de8ed0b628d84d256f5846500b0385979bdTim Murray    __sync_synchronize();
268709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
269709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    pthread_attr_t threadAttr;
270709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    status = pthread_attr_init(&threadAttr);
271709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (status) {
272709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        ALOGE("Failed to init thread attribute.");
273709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        return false;
274709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
275709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
276709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct=0; ct < mWorkers.mCount; ct++) {
277709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        status = pthread_create(&mWorkers.mThreadId[ct], &threadAttr, helperThreadProc, this);
278709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        if (status) {
279709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            mWorkers.mCount = ct;
280709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            ALOGE("Created fewer than expected number of RS threads.");
281709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            break;
282709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        }
283709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
2840b575de8ed0b628d84d256f5846500b0385979bdTim Murray    while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) {
285709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        usleep(100);
286709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
287709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
288709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    pthread_attr_destroy(&threadAttr);
289709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return true;
290709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
291709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
292709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
293709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::setPriority(int32_t priority) {
294709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct=0; ct < mWorkers.mCount; ct++) {
295709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        setpriority(PRIO_PROCESS, mWorkers.mNativeThreadId[ct], priority);
296709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
297709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
298709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
299709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::~RsdCpuReferenceImpl() {
300709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mExit = true;
30144bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mWorkers.mLaunchData = nullptr;
30244bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mWorkers.mLaunchCallback = nullptr;
3030b575de8ed0b628d84d256f5846500b0385979bdTim Murray    mWorkers.mRunningCount = mWorkers.mCount;
3040b575de8ed0b628d84d256f5846500b0385979bdTim Murray    __sync_synchronize();
305709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) {
306709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        mWorkers.mLaunchSignals[ct].set();
307709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
308709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    void *res;
309709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) {
310709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        pthread_join(mWorkers.mThreadId[ct], &res);
311709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
3120b575de8ed0b628d84d256f5846500b0385979bdTim Murray    rsAssert(__sync_fetch_and_or(&mWorkers.mRunningCount, 0) == 0);
31307ef704308b514272ed2f5c3e6a2f4c055550158Jens Gulin    free(mWorkers.mThreadId);
31407ef704308b514272ed2f5c3e6a2f4c055550158Jens Gulin    free(mWorkers.mNativeThreadId);
31507ef704308b514272ed2f5c3e6a2f4c055550158Jens Gulin    delete[] mWorkers.mLaunchSignals;
316709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
317709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    // Global structure cleanup.
318709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    lockMutex();
319709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    --gThreadTLSKeyCount;
320709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (!gThreadTLSKeyCount) {
321709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        pthread_key_delete(gThreadTLSKey);
322709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
323709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    unlockMutex();
324709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
325709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
326709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
32714ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// Set up the appropriate input and output pointers to the kernel driver info structure.
32814ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// Inputs:
32914ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala//   mtls - The MTLaunchStruct holding information about the kernel launch
33014ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala//   fep - The forEach parameters (driver info structure)
33114ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala//   x, y, z, lod, face, a1, a2, a3, a4 - The start offsets into each dimension
33214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Walastatic inline void FepPtrSetup(const MTLaunchStructForEach *mtls, RsExpandKernelDriverInfo *fep,
333c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                               uint32_t x, uint32_t y,
334c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                               uint32_t z = 0, uint32_t lod = 0,
335c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                               RsAllocationCubemapFace face = RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X,
336c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                               uint32_t a1 = 0, uint32_t a2 = 0, uint32_t a3 = 0, uint32_t a4 = 0) {
337c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    for (uint32_t i = 0; i < fep->inLen; i++) {
338c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        fep->inPtr[i] = (const uint8_t *)mtls->ains[i]->getPointerUnchecked(x, y, z, lod, face, a1, a2, a3, a4);
339709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
340c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    if (mtls->aout[0] != nullptr) {
341c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        fep->outPtr[0] = (uint8_t *)mtls->aout[0]->getPointerUnchecked(x, y, z, lod, face, a1, a2, a3, a4);
342c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    }
343709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
344709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
345bf2111d3b3de310932099514f06924e48fa1d7b2Jason Samsstatic uint32_t sliceInt(uint32_t *p, uint32_t val, uint32_t start, uint32_t end) {
346bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    if (start >= end) {
347bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        *p = start;
348bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        return val;
349bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    }
350bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
351bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    uint32_t div = end - start;
352bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
353bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    uint32_t n = val / div;
354bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    *p = (val - (n * div)) + start;
355bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    return n;
356bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams}
357bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
35814ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Walastatic bool SelectOuterSlice(const MTLaunchStructForEach *mtls, RsExpandKernelDriverInfo* fep, uint32_t sliceNum) {
359bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
360bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    uint32_t r = sliceNum;
36159b35f29cc576243a322bf88bc16063a9810da55Jason Sams    r = sliceInt(&fep->current.z, r, mtls->start.z, mtls->end.z);
36259b35f29cc576243a322bf88bc16063a9810da55Jason Sams    r = sliceInt(&fep->current.lod, r, mtls->start.lod, mtls->end.lod);
36359b35f29cc576243a322bf88bc16063a9810da55Jason Sams    r = sliceInt(&fep->current.face, r, mtls->start.face, mtls->end.face);
36459b35f29cc576243a322bf88bc16063a9810da55Jason Sams    r = sliceInt(&fep->current.array[0], r, mtls->start.array[0], mtls->end.array[0]);
36559b35f29cc576243a322bf88bc16063a9810da55Jason Sams    r = sliceInt(&fep->current.array[1], r, mtls->start.array[1], mtls->end.array[1]);
36659b35f29cc576243a322bf88bc16063a9810da55Jason Sams    r = sliceInt(&fep->current.array[2], r, mtls->start.array[2], mtls->end.array[2]);
36759b35f29cc576243a322bf88bc16063a9810da55Jason Sams    r = sliceInt(&fep->current.array[3], r, mtls->start.array[3], mtls->end.array[3]);
368bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    return r == 0;
369bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams}
370bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
371bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
372bf2111d3b3de310932099514f06924e48fa1d7b2Jason Samsstatic void walk_general(void *usr, uint32_t idx) {
37314ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    MTLaunchStructForEach *mtls = (MTLaunchStructForEach *)usr;
374bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    RsExpandKernelDriverInfo fep = mtls->fep;
375bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    fep.lid = idx;
37614ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    ForEachFunc_t fn = mtls->kernel;
377bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
378bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
379bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    while(1) {
380bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1);
381bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
38259b35f29cc576243a322bf88bc16063a9810da55Jason Sams        if (!SelectOuterSlice(mtls, &fep, slice)) {
383bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            return;
384bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        }
385bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
38659b35f29cc576243a322bf88bc16063a9810da55Jason Sams        for (fep.current.y = mtls->start.y; fep.current.y < mtls->end.y;
38759b35f29cc576243a322bf88bc16063a9810da55Jason Sams             fep.current.y++) {
388bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
38959b35f29cc576243a322bf88bc16063a9810da55Jason Sams            FepPtrSetup(mtls, &fep, mtls->start.x,
39059b35f29cc576243a322bf88bc16063a9810da55Jason Sams                        fep.current.y, fep.current.z, fep.current.lod,
39159b35f29cc576243a322bf88bc16063a9810da55Jason Sams                        (RsAllocationCubemapFace)fep.current.face,
39259b35f29cc576243a322bf88bc16063a9810da55Jason Sams                        fep.current.array[0], fep.current.array[1],
39359b35f29cc576243a322bf88bc16063a9810da55Jason Sams                        fep.current.array[2], fep.current.array[3]);
394bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
39559b35f29cc576243a322bf88bc16063a9810da55Jason Sams            fn(&fep, mtls->start.x, mtls->end.x, mtls->fep.outStride[0]);
396bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        }
397bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    }
398bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
399bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams}
400c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams
401f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailesstatic void walk_2d(void *usr, uint32_t idx) {
40214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    MTLaunchStructForEach *mtls = (MTLaunchStructForEach *)usr;
403c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    RsExpandKernelDriverInfo fep = mtls->fep;
404c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    fep.lid = idx;
40514ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    ForEachFunc_t fn = mtls->kernel;
406709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
407c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    while (1) {
408c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        uint32_t slice  = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1);
409bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        uint32_t yStart = mtls->start.y + slice * mtls->mSliceSize;
410c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        uint32_t yEnd   = yStart + mtls->mSliceSize;
411709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
412bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        yEnd = rsMin(yEnd, mtls->end.y);
413709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
414c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        if (yEnd <= yStart) {
415c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            return;
416c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        }
417709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
418c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        for (fep.current.y = yStart; fep.current.y < yEnd; fep.current.y++) {
419bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            FepPtrSetup(mtls, &fep, mtls->start.x, fep.current.y);
420709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
421b0abb140ac51b93d1a85aadaa63fe057f2d29850David Gross            fn(&fep, mtls->start.x, mtls->end.x, fep.outStride[0]);
422709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        }
423c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    }
424f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes}
4254b2bea3dc20865f3a198797702e19912a6a2171cStephen Hines
426f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailesstatic void walk_1d(void *usr, uint32_t idx) {
42714ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    MTLaunchStructForEach *mtls = (MTLaunchStructForEach *)usr;
428c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    RsExpandKernelDriverInfo fep = mtls->fep;
429c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    fep.lid = idx;
43014ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    ForEachFunc_t fn = mtls->kernel;
43180ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
432c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    while (1) {
433c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        uint32_t slice  = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1);
434bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        uint32_t xStart = mtls->start.x + slice * mtls->mSliceSize;
435c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        uint32_t xEnd   = xStart + mtls->mSliceSize;
43680ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
437bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        xEnd = rsMin(xEnd, mtls->end.x);
43880ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
439c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        if (xEnd <= xStart) {
440c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            return;
441c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        }
44280ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
443bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        FepPtrSetup(mtls, &fep, xStart, 0);
44480ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
445b0abb140ac51b93d1a85aadaa63fe057f2d29850David Gross        fn(&fep, xStart, xEnd, fep.outStride[0]);
446c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    }
447709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
448709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
44914ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// Launch a reduce-style kernel.
45014ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// Inputs:
45114ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala//  ain:  The allocation that contains the input
45214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala//  aout: The allocation that will hold the output
45314ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala//  mtls: Holds launch parameters
45414ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Walavoid RsdCpuReferenceImpl::launchReduce(const Allocation *ain,
45514ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala                                       Allocation *aout,
45614ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala                                       MTLaunchStructReduce *mtls) {
45714ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    const uint32_t xStart = mtls->start.x;
45814ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    const uint32_t xEnd = mtls->end.x;
45914ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala
46014ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    if (xStart >= xEnd) {
46114ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala      return;
46214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    }
46314ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala
46414ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    const uint32_t startOffset = ain->getType()->getElementSizeBytes() * xStart;
46514ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala    mtls->kernel(&mtls->inBuf[startOffset], mtls->outBuf, xEnd - xStart);
46614ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala}
46714ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala
46814ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Walavoid RsdCpuReferenceImpl::launchForEach(const Allocation ** ains,
469f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                                        uint32_t inLen,
470f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                                        Allocation* aout,
471f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                                        const RsScriptCall* sc,
47214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala                                        MTLaunchStructForEach* mtls) {
4734b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
4744b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes    //android::StopWatch kernel_time("kernel time");
4754b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
476bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    bool outerDims = (mtls->start.z != mtls->end.z) ||
477bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.face != mtls->end.face) ||
478bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.lod != mtls->end.lod) ||
479bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.array[0] != mtls->end.array[0]) ||
480bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.array[1] != mtls->end.array[1]) ||
481bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.array[2] != mtls->end.array[2]) ||
482bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.array[3] != mtls->end.array[3]);
483bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
4844b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes    if ((mWorkers.mCount >= 1) && mtls->isThreadable && !mInForEach) {
4854b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        const size_t targetByteChunk = 16 * 1024;
4864b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        mInForEach = true;
487f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes
488bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        if (outerDims) {
489bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            // No fancy logic for chunk size
490bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            mtls->mSliceSize = 1;
491bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            launchThreads(walk_general, mtls);
492bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        } else if (mtls->fep.dim.y > 1) {
493c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            uint32_t s1 = mtls->fep.dim.y / ((mWorkers.mCount + 1) * 4);
4944b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            uint32_t s2 = 0;
4954b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
4964b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            // This chooses our slice size to rate limit atomic ops to
4974b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            // one per 16k bytes of reads/writes.
498c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            if ((mtls->aout[0] != nullptr) && mtls->aout[0]->mHal.drvState.lod[0].stride) {
499c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                s2 = targetByteChunk / mtls->aout[0]->mHal.drvState.lod[0].stride;
500a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams            } else if (mtls->ains[0]) {
501c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                s2 = targetByteChunk / mtls->ains[0]->mHal.drvState.lod[0].stride;
502a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams            } else {
503a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams                // Launch option only case
504a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams                // Use s1 based only on the dimensions
505a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams                s2 = s1;
5064b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5074b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            mtls->mSliceSize = rsMin(s1, s2);
5084b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5094b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            if(mtls->mSliceSize < 1) {
5104b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes                mtls->mSliceSize = 1;
5114b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5124b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
513f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes            launchThreads(walk_2d, mtls);
5144b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        } else {
515c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            uint32_t s1 = mtls->fep.dim.x / ((mWorkers.mCount + 1) * 4);
5164b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            uint32_t s2 = 0;
5174b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5184b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            // This chooses our slice size to rate limit atomic ops to
5194b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            // one per 16k bytes of reads/writes.
520c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            if ((mtls->aout[0] != nullptr) && mtls->aout[0]->getType()->getElementSizeBytes()) {
521c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                s2 = targetByteChunk / mtls->aout[0]->getType()->getElementSizeBytes();
522a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams            } else if (mtls->ains[0]) {
523c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                s2 = targetByteChunk / mtls->ains[0]->getType()->getElementSizeBytes();
524a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams            } else {
525a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams                // Launch option only case
526a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams                // Use s1 based only on the dimensions
527a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams                s2 = s1;
5284b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5294b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            mtls->mSliceSize = rsMin(s1, s2);
5304b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5314b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            if (mtls->mSliceSize < 1) {
5324b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes                mtls->mSliceSize = 1;
5334b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5344b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
535f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes            launchThreads(walk_1d, mtls);
5364b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        }
5374b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        mInForEach = false;
5384b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5394b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes    } else {
54014ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala        ForEachFunc_t fn = mtls->kernel;
541bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        uint32_t slice = 0;
54280ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
5434b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
54459b35f29cc576243a322bf88bc16063a9810da55Jason Sams        while(SelectOuterSlice(mtls, &mtls->fep, slice++)) {
545bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            for (mtls->fep.current.y = mtls->start.y;
546bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                 mtls->fep.current.y < mtls->end.y;
547bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                 mtls->fep.current.y++) {
5484b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
549bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                FepPtrSetup(mtls, &mtls->fep, mtls->start.x,
550bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                            mtls->fep.current.y, mtls->fep.current.z, mtls->fep.current.lod,
551bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                            (RsAllocationCubemapFace) mtls->fep.current.face,
552bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                            mtls->fep.current.array[0], mtls->fep.current.array[1],
553bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                            mtls->fep.current.array[2], mtls->fep.current.array[3]);
5544b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
555b0abb140ac51b93d1a85aadaa63fe057f2d29850David Gross                fn(&mtls->fep, mtls->start.x, mtls->end.x, mtls->fep.outStride[0]);
5564b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5574b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        }
5584b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes    }
5594b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes}
5604b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
561709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuScriptImpl * RsdCpuReferenceImpl::setTLS(RsdCpuScriptImpl *sc) {
562709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    //ALOGE("setTls %p", sc);
563709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey);
564709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    rsAssert(tls);
565709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    RsdCpuScriptImpl *old = tls->mImpl;
566709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    tls->mImpl = sc;
567709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    tls->mContext = mRSC;
568709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (sc) {
569709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        tls->mScript = sc->getScript();
570709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    } else {
57144bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        tls->mScript = nullptr;
572709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
573709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return old;
574709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
575709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
576709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst RsdCpuReference::CpuSymbol * RsdCpuReferenceImpl::symLookup(const char *name) {
577709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return mSymLookupFn(mRSC, name);
578709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
579709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
580709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
581709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createScript(const ScriptC *s,
582709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                                    char const *resName, char const *cacheDir,
583709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                                    uint8_t const *bitcode, size_t bitcodeSize,
584709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                                    uint32_t flags) {
585709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
586709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    RsdCpuScriptImpl *i = new RsdCpuScriptImpl(this, s);
587005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines    if (!i->init(resName, cacheDir, bitcode, bitcodeSize, flags
588005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines        , getBccPluginName()
589005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines        )) {
590709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        delete i;
59144bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        return nullptr;
592709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
593709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return i;
594709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
595709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
5967c4b888f2147edf99690b6af75470774ff31c43bJason Samsextern RsdCpuScriptImpl * rsdIntrinsic_3DLUT(RsdCpuReferenceImpl *ctx,
5977c4b888f2147edf99690b6af75470774ff31c43bJason Sams                                             const Script *s, const Element *e);
598c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve3x3(RsdCpuReferenceImpl *ctx,
599c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                                   const Script *s, const Element *e);
600c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_ColorMatrix(RsdCpuReferenceImpl *ctx,
601c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                                   const Script *s, const Element *e);
602c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_LUT(RsdCpuReferenceImpl *ctx,
603c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                           const Script *s, const Element *e);
604c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve5x5(RsdCpuReferenceImpl *ctx,
605c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                                   const Script *s, const Element *e);
606c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blur(RsdCpuReferenceImpl *ctx,
607c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                            const Script *s, const Element *e);
608c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_YuvToRGB(RsdCpuReferenceImpl *ctx,
609c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                                const Script *s, const Element *e);
610c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blend(RsdCpuReferenceImpl *ctx,
611c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                             const Script *s, const Element *e);
6122282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Histogram(RsdCpuReferenceImpl *ctx,
6132282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams                                                 const Script *s, const Element *e);
61439ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Resize(RsdCpuReferenceImpl *ctx,
61539ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams                                              const Script *s, const Element *e);
61664c682b65cd04ac83b51251b40dca14423df351aTim Murrayextern RsdCpuScriptImpl * rsdIntrinsic_BLAS(RsdCpuReferenceImpl *ctx,
61764c682b65cd04ac83b51251b40dca14423df351aTim Murray                                              const Script *s, const Element *e);
618709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
619709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createIntrinsic(const Script *s,
620709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                                    RsScriptIntrinsicID iid, Element *e) {
621709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
62244bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    RsdCpuScriptImpl *i = nullptr;
623709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    switch (iid) {
6247c4b888f2147edf99690b6af75470774ff31c43bJason Sams    case RS_SCRIPT_INTRINSIC_ID_3DLUT:
6257c4b888f2147edf99690b6af75470774ff31c43bJason Sams        i = rsdIntrinsic_3DLUT(this, s, e);
6267c4b888f2147edf99690b6af75470774ff31c43bJason Sams        break;
627709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_3x3:
628c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_Convolve3x3(this, s, e);
629709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
630709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_COLOR_MATRIX:
631c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_ColorMatrix(this, s, e);
632709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
633709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_LUT:
634c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_LUT(this, s, e);
635709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
636709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_5x5:
637c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_Convolve5x5(this, s, e);
638709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
639709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_BLUR:
640c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_Blur(this, s, e);
641709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
642709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_YUV_TO_RGB:
643c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_YuvToRGB(this, s, e);
644709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
645709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_BLEND:
646c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_Blend(this, s, e);
647709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
6482282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams    case RS_SCRIPT_INTRINSIC_ID_HISTOGRAM:
6492282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams        i = rsdIntrinsic_Histogram(this, s, e);
6502282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams        break;
65139ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams    case RS_SCRIPT_INTRINSIC_ID_RESIZE:
65239ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams        i = rsdIntrinsic_Resize(this, s, e);
65339ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams        break;
65464c682b65cd04ac83b51251b40dca14423df351aTim Murray    case RS_SCRIPT_INTRINSIC_ID_BLAS:
65564c682b65cd04ac83b51251b40dca14423df351aTim Murray        i = rsdIntrinsic_BLAS(this, s, e);
65664c682b65cd04ac83b51251b40dca14423df351aTim Murray        break;
657709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
658709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    default:
659709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        rsAssert(0);
660709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
661709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
662709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return i;
663709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
664709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
6651ffd86b448d78366190c540f98f8b6d641cdb6cfYang Nivoid* RsdCpuReferenceImpl::createScriptGroup(const ScriptGroupBase *sg) {
6661ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni  switch (sg->getApiVersion()) {
6671ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni    case ScriptGroupBase::SG_V1: {
6681ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      CpuScriptGroupImpl *sgi = new CpuScriptGroupImpl(this, sg);
6691ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      if (!sgi->init()) {
670709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        delete sgi;
67144bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        return nullptr;
6721ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      }
6731ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      return sgi;
674709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
6751ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni    case ScriptGroupBase::SG_V2: {
6761ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      return new CpuScriptGroup2Impl(this, sg);
6771ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni    }
6781ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni  }
6791ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni  return nullptr;
680709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
681