rsCpuCore.cpp revision bf2111d3b3de310932099514f06924e48fa1d7b2
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>
29709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <string.h>
30b0934b67b95cc27e2358c2aa4db5f7c1067c8f9bStephen Hines#include <unistd.h>
310b575de8ed0b628d84d256f5846500b0385979bdTim Murray
32f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams#include <stdio.h>
33f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams#include <stdlib.h>
34f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams#include <fcntl.h>
35f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
36b0934b67b95cc27e2358c2aa4db5f7c1067c8f9bStephen Hines#if !defined(RS_SERVER) && !defined(RS_COMPATIBILITY_LIB)
370b575de8ed0b628d84d256f5846500b0385979bdTim Murray#include <cutils/properties.h>
38709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "utils/StopWatch.h"
390b575de8ed0b628d84d256f5846500b0385979bdTim Murray#endif
400b575de8ed0b628d84d256f5846500b0385979bdTim Murray
410b575de8ed0b628d84d256f5846500b0385979bdTim Murray#ifdef RS_SERVER
420b575de8ed0b628d84d256f5846500b0385979bdTim Murray// Android exposes gettid(), standard Linux does not
430b575de8ed0b628d84d256f5846500b0385979bdTim Murraystatic pid_t gettid() {
440b575de8ed0b628d84d256f5846500b0385979bdTim Murray    return syscall(SYS_gettid);
450b575de8ed0b628d84d256f5846500b0385979bdTim Murray}
460b575de8ed0b628d84d256f5846500b0385979bdTim Murray#endif
47709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
48709a0978ae141198018ca9769f8d96292a8928e6Jason Samsusing namespace android;
49709a0978ae141198018ca9769f8d96292a8928e6Jason Samsusing namespace android::renderscript;
50709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
51709a0978ae141198018ca9769f8d96292a8928e6Jason Samstypedef void (*outer_foreach_t)(
5280ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes    const android::renderscript::RsExpandKernelParams *,
539ed79105cc6a8dbfaf959875249f36022cc2c798Chris Wailes    uint32_t x1, uint32_t x2, uint32_t outstep);
54709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
55709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
56709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_key_t gThreadTLSKey = 0;
57709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic uint32_t gThreadTLSKeyCount = 0;
58709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_mutex_t gInitMutex = PTHREAD_MUTEX_INITIALIZER;
59709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
60f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Samsbool android::renderscript::gArchUseSIMD = false;
61f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
62709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::~RsdCpuReference() {
63709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
64709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
65709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference * RsdCpuReference::create(Context *rsc, uint32_t version_major,
66cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams        uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn
671d476620399d54774e4fd386c1d23cc583d49522Stephen Hines        , bcc::RSLinkRuntimeCallback pLinkRuntimeCallback,
68005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines        RSSelectRTCallback pSelectRTCallback,
69005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines        const char *pBccPluginName
70cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams        ) {
71709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
72709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    RsdCpuReferenceImpl *cpu = new RsdCpuReferenceImpl(rsc);
73709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (!cpu) {
7444bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        return nullptr;
75709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
76709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (!cpu->init(version_major, version_minor, lfn, slfn)) {
77709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        delete cpu;
7844bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        return nullptr;
79709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
80f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines
81f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines    cpu->setLinkRuntimeCallback(pLinkRuntimeCallback);
821d476620399d54774e4fd386c1d23cc583d49522Stephen Hines    cpu->setSelectRTCallback(pSelectRTCallback);
83005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines    if (pBccPluginName) {
84005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines        cpu->setBccPluginName(pBccPluginName);
85005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines    }
86f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines
87709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return cpu;
88709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
89709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
90709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
91709a0978ae141198018ca9769f8d96292a8928e6Jason SamsContext * RsdCpuReference::getTlsContext() {
92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey);
93709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return tls->mContext;
94709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
95709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
96709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst Script * RsdCpuReference::getTlsScript() {
97709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey);
98709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return tls->mScript;
99709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
100709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
101f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hinespthread_key_t RsdCpuReference::getThreadTLSKey(){ return gThreadTLSKey; }
102709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
103709a0978ae141198018ca9769f8d96292a8928e6Jason Sams////////////////////////////////////////////////////////////
104709a0978ae141198018ca9769f8d96292a8928e6Jason Sams///
105709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
106709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::RsdCpuReferenceImpl(Context *rsc) {
107709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mRSC = rsc;
108709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
109709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    version_major = 0;
110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    version_minor = 0;
111709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mInForEach = false;
112709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    memset(&mWorkers, 0, sizeof(mWorkers));
113709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    memset(&mTlsStruct, 0, sizeof(mTlsStruct));
114709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mExit = false;
11544bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mLinkRuntimeCallback = nullptr;
11644bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mSelectRTCallback = nullptr;
11744bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mSetupCompilerCallback = nullptr;
118709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
119709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
120709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
121709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid * RsdCpuReferenceImpl::helperThreadProc(void *vrsc) {
122709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    RsdCpuReferenceImpl *dc = (RsdCpuReferenceImpl *)vrsc;
123709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
1240b575de8ed0b628d84d256f5846500b0385979bdTim Murray    uint32_t idx = __sync_fetch_and_add(&dc->mWorkers.mLaunchCount, 1);
125709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
126709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    //ALOGV("RS helperThread starting %p idx=%i", dc, idx);
127709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
128709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    dc->mWorkers.mLaunchSignals[idx].init();
129709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    dc->mWorkers.mNativeThreadId[idx] = gettid();
130709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
131709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    memset(&dc->mTlsStruct, 0, sizeof(dc->mTlsStruct));
132709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    int status = pthread_setspecific(gThreadTLSKey, &dc->mTlsStruct);
133709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (status) {
134709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        ALOGE("pthread_setspecific %i", status);
135709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
136709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
137709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#if 0
138709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    typedef struct {uint64_t bits[1024 / 64]; } cpu_set_t;
139709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    cpu_set_t cpuset;
140709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    memset(&cpuset, 0, sizeof(cpuset));
141709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    cpuset.bits[idx / 64] |= 1ULL << (idx % 64);
142709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    int ret = syscall(241, rsc->mWorkers.mNativeThreadId[idx],
143709a0978ae141198018ca9769f8d96292a8928e6Jason Sams              sizeof(cpuset), &cpuset);
144709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    ALOGE("SETAFFINITY ret = %i %s", ret, EGLUtils::strerror(ret));
145709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif
146709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
147709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    while (!dc->mExit) {
148709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        dc->mWorkers.mLaunchSignals[idx].wait();
149709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        if (dc->mWorkers.mLaunchCallback) {
150709a0978ae141198018ca9769f8d96292a8928e6Jason Sams           // idx +1 is used because the calling thread is always worker 0.
151709a0978ae141198018ca9769f8d96292a8928e6Jason Sams           dc->mWorkers.mLaunchCallback(dc->mWorkers.mLaunchData, idx+1);
152709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        }
1530b575de8ed0b628d84d256f5846500b0385979bdTim Murray        __sync_fetch_and_sub(&dc->mWorkers.mRunningCount, 1);
154709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        dc->mWorkers.mCompleteSignal.set();
155709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
156709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
157709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    //ALOGV("RS helperThread exited %p idx=%i", dc, idx);
15844bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    return nullptr;
159709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
160709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
161709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::launchThreads(WorkerCallback_t cbk, void *data) {
162709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mLaunchData = data;
163709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mLaunchCallback = cbk;
1644d252d6e807b89764dad123ac845df298c52ca97Tim Murray
1654d252d6e807b89764dad123ac845df298c52ca97Tim Murray    // fast path for very small launches
1664d252d6e807b89764dad123ac845df298c52ca97Tim Murray    MTLaunchStruct *mtls = (MTLaunchStruct *)data;
167bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    if (mtls && mtls->fep.dim.y <= 1 && mtls->end.x <= mtls->start.x + mtls->mSliceSize) {
1684d252d6e807b89764dad123ac845df298c52ca97Tim Murray        if (mWorkers.mLaunchCallback) {
1694d252d6e807b89764dad123ac845df298c52ca97Tim Murray            mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0);
1704d252d6e807b89764dad123ac845df298c52ca97Tim Murray        }
1714d252d6e807b89764dad123ac845df298c52ca97Tim Murray        return;
1724d252d6e807b89764dad123ac845df298c52ca97Tim Murray    }
1734d252d6e807b89764dad123ac845df298c52ca97Tim Murray
1740b575de8ed0b628d84d256f5846500b0385979bdTim Murray    mWorkers.mRunningCount = mWorkers.mCount;
1750b575de8ed0b628d84d256f5846500b0385979bdTim Murray    __sync_synchronize();
1760b575de8ed0b628d84d256f5846500b0385979bdTim Murray
177709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) {
178709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        mWorkers.mLaunchSignals[ct].set();
179709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
180709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
181709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    // We use the calling thread as one of the workers so we can start without
182709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    // the delay of the thread wakeup.
183709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (mWorkers.mLaunchCallback) {
1844d252d6e807b89764dad123ac845df298c52ca97Tim Murray        mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0);
185709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
186709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
1870b575de8ed0b628d84d256f5846500b0385979bdTim Murray    while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) {
188709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        mWorkers.mCompleteSignal.wait();
189709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
190709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
191709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
192709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
193709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::lockMutex() {
194709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    pthread_mutex_lock(&gInitMutex);
195709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
196709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
197709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::unlockMutex() {
198709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    pthread_mutex_unlock(&gInitMutex);
199709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
200709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
201f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Samsstatic int
202f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Samsread_file(const char*  pathname, char*  buffer, size_t  buffsize)
203f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams{
204f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    int  fd, len;
205f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
206f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    fd = open(pathname, O_RDONLY);
207f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    if (fd < 0)
208f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams        return -1;
209f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
210f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    do {
211f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams        len = read(fd, buffer, buffsize);
212f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    } while (len < 0 && errno == EINTR);
213f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
214f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    close(fd);
215f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
216f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    return len;
217f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams}
218f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
219f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Samsstatic void GetCpuInfo() {
220f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    char cpuinfo[4096];
221f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    int  cpuinfo_len;
222f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
223f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    cpuinfo_len = read_file("/proc/cpuinfo", cpuinfo, sizeof cpuinfo);
224f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    if (cpuinfo_len < 0)  /* should not happen */ {
225f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams        return;
226f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    }
227f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
228074424a4ac5b093331df2c92e7a5bcbfff136b71Jason Sams#if defined(ARCH_ARM_HAVE_VFP) || defined(ARCH_ARM_USE_INTRINSICS)
229074424a4ac5b093331df2c92e7a5bcbfff136b71Jason Sams    gArchUseSIMD = (!!strstr(cpuinfo, " neon")) ||
230074424a4ac5b093331df2c92e7a5bcbfff136b71Jason Sams                   (!!strstr(cpuinfo, " asimd"));
2317b7060c61e4182b29186849c5a857ea5f0898e56Rose, James#elif defined(ARCH_X86_HAVE_SSSE3)
2327b7060c61e4182b29186849c5a857ea5f0898e56Rose, James    gArchUseSIMD = !!strstr(cpuinfo, " ssse3");
2337b7060c61e4182b29186849c5a857ea5f0898e56Rose, James#endif
234f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams}
235f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
236709a0978ae141198018ca9769f8d96292a8928e6Jason Samsbool RsdCpuReferenceImpl::init(uint32_t version_major, uint32_t version_minor,
237709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                               sym_lookup_t lfn, script_lookup_t slfn) {
238709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
239709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mSymLookupFn = lfn;
240709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mScriptLookupFn = slfn;
241709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
242709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    lockMutex();
243709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (!gThreadTLSKeyCount) {
24444bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        int status = pthread_key_create(&gThreadTLSKey, nullptr);
245709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        if (status) {
246709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            ALOGE("Failed to init thread tls key.");
247709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            unlockMutex();
248709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            return false;
249709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        }
250709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
251709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    gThreadTLSKeyCount++;
252709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    unlockMutex();
253709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
254709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mTlsStruct.mContext = mRSC;
25544bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mTlsStruct.mScript = nullptr;
256709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    int status = pthread_setspecific(gThreadTLSKey, &mTlsStruct);
257709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (status) {
258709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        ALOGE("pthread_setspecific %i", status);
259709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
260709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
261f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams    GetCpuInfo();
262f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams
26377d57a305f4134e78ebc91869011c4009988104eJason Sams    int cpu = sysconf(_SC_NPROCESSORS_CONF);
264709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if(mRSC->props.mDebugMaxThreads) {
265709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        cpu = mRSC->props.mDebugMaxThreads;
266709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
267709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (cpu < 2) {
268709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        mWorkers.mCount = 0;
269709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        return true;
270709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
271709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
272709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    // Subtract one from the cpu count because we also use the command thread as a worker.
273709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mCount = (uint32_t)(cpu - 1);
274709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
2758ca358a2abe7e0dba23993e0fc8d64b8b55bd9caJason Sams    ALOGV("%p Launching thread(s), CPUs %i", mRSC, mWorkers.mCount + 1);
276709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
277709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mThreadId = (pthread_t *) calloc(mWorkers.mCount, sizeof(pthread_t));
278709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mNativeThreadId = (pid_t *) calloc(mWorkers.mCount, sizeof(pid_t));
279709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mLaunchSignals = new Signal[mWorkers.mCount];
28044bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mWorkers.mLaunchCallback = nullptr;
281709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
282709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mWorkers.mCompleteSignal.init();
283709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
2840b575de8ed0b628d84d256f5846500b0385979bdTim Murray    mWorkers.mRunningCount = mWorkers.mCount;
2850b575de8ed0b628d84d256f5846500b0385979bdTim Murray    mWorkers.mLaunchCount = 0;
2860b575de8ed0b628d84d256f5846500b0385979bdTim Murray    __sync_synchronize();
287709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
288709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    pthread_attr_t threadAttr;
289709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    status = pthread_attr_init(&threadAttr);
290709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (status) {
291709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        ALOGE("Failed to init thread attribute.");
292709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        return false;
293709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
294709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
295709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct=0; ct < mWorkers.mCount; ct++) {
296709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        status = pthread_create(&mWorkers.mThreadId[ct], &threadAttr, helperThreadProc, this);
297709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        if (status) {
298709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            mWorkers.mCount = ct;
299709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            ALOGE("Created fewer than expected number of RS threads.");
300709a0978ae141198018ca9769f8d96292a8928e6Jason Sams            break;
301709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        }
302709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
3030b575de8ed0b628d84d256f5846500b0385979bdTim Murray    while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) {
304709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        usleep(100);
305709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
306709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
307709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    pthread_attr_destroy(&threadAttr);
308709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return true;
309709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
310709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
311709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
312709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::setPriority(int32_t priority) {
313709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct=0; ct < mWorkers.mCount; ct++) {
314709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        setpriority(PRIO_PROCESS, mWorkers.mNativeThreadId[ct], priority);
315709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
316709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
317709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
318709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::~RsdCpuReferenceImpl() {
319709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    mExit = true;
32044bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mWorkers.mLaunchData = nullptr;
32144bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    mWorkers.mLaunchCallback = nullptr;
3220b575de8ed0b628d84d256f5846500b0385979bdTim Murray    mWorkers.mRunningCount = mWorkers.mCount;
3230b575de8ed0b628d84d256f5846500b0385979bdTim Murray    __sync_synchronize();
324709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) {
325709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        mWorkers.mLaunchSignals[ct].set();
326709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
327709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    void *res;
328709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) {
329709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        pthread_join(mWorkers.mThreadId[ct], &res);
330709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
3310b575de8ed0b628d84d256f5846500b0385979bdTim Murray    rsAssert(__sync_fetch_and_or(&mWorkers.mRunningCount, 0) == 0);
33207ef704308b514272ed2f5c3e6a2f4c055550158Jens Gulin    free(mWorkers.mThreadId);
33307ef704308b514272ed2f5c3e6a2f4c055550158Jens Gulin    free(mWorkers.mNativeThreadId);
33407ef704308b514272ed2f5c3e6a2f4c055550158Jens Gulin    delete[] mWorkers.mLaunchSignals;
335709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
336709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    // Global structure cleanup.
337709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    lockMutex();
338709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    --gThreadTLSKeyCount;
339709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (!gThreadTLSKeyCount) {
340709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        pthread_key_delete(gThreadTLSKey);
341709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
342709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    unlockMutex();
343709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
344709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
345709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
346709a0978ae141198018ca9769f8d96292a8928e6Jason Samstypedef void (*rs_t)(const void *, void *, const void *, uint32_t, uint32_t, uint32_t, uint32_t);
347c0d68470b978a79ce024fde56f23ea3690603ccdJason Samstypedef void (*walk_loop_t)(const MTLaunchStruct*,
348c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                            RsExpandKernelDriverInfo,
349f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                            outer_foreach_t);
350709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
351c0d68470b978a79ce024fde56f23ea3690603ccdJason Samsstatic void kparamSetup(RsExpandKernelParams *kparams, const RsExpandKernelDriverInfo *fep) {
352c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    //ALOGE("kp  usr %p", fep->usr);
353c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    //ALOGE("kp  slot %i", fep->slot);
354c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    //ALOGE("kp  dim %i %i %i", fep->dim.x, fep->dim.y, fep->dim.z);
355c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    //ALOGE("kp  lid %i", fep->lid);
356c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    //ALOGE("kp  in[0] stide %i  ptr %p", fep->inStride[0], fep->inPtr[0]);
357c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    //ALOGE("kp  out[0] ptr %p", fep->outPtr[0]);
358c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    //ALOGE("kp  loc %i %i %i", fep->current.x, fep->current.y, fep->current.z);
359c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams
360c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->usr  = fep->usr;
361c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->slot = fep->slot;
362c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->dimX = fep->dim.x;
363c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->dimY = fep->dim.y;
364c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->dimZ = fep->dim.z;
365c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->lid = fep->lid;
366c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->inEStrides = (uint32_t *)&fep->inStride[0];
367c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->ins = (const void **)&fep->inPtr[0];
368c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->out = fep->outPtr[0];
369c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->y = fep->current.y;
370c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    kparams->z = fep->current.z;
371c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams}
372f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes
373bf2111d3b3de310932099514f06924e48fa1d7b2Jason Samsstatic inline void FepPtrSetup(const MTLaunchStruct *mtls, RsExpandKernelDriverInfo *fep,
374c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                               uint32_t x, uint32_t y,
375c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                               uint32_t z = 0, uint32_t lod = 0,
376c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                               RsAllocationCubemapFace face = RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X,
377c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                               uint32_t a1 = 0, uint32_t a2 = 0, uint32_t a3 = 0, uint32_t a4 = 0) {
3784b2bea3dc20865f3a198797702e19912a6a2171cStephen Hines
379c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    for (uint32_t i = 0; i < fep->inLen; i++) {
380c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        fep->inPtr[i] = (const uint8_t *)mtls->ains[i]->getPointerUnchecked(x, y, z, lod, face, a1, a2, a3, a4);
381709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
382709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
383c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    if (mtls->aout[0] != nullptr) {
384c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        fep->outPtr[0] = (uint8_t *)mtls->aout[0]->getPointerUnchecked(x, y, z, lod, face, a1, a2, a3, a4);
385c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    }
386709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
387709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
388bf2111d3b3de310932099514f06924e48fa1d7b2Jason Samsstatic uint32_t sliceInt(uint32_t *p, uint32_t val, uint32_t start, uint32_t end) {
389bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    if (start >= end) {
390bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        *p = start;
391bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        return val;
392bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    }
393bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
394bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    uint32_t div = end - start;
395bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
396bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    uint32_t n = val / div;
397bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    *p = (val - (n * div)) + start;
398bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    return n;
399bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams}
400bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
401bf2111d3b3de310932099514f06924e48fa1d7b2Jason Samsstatic bool SelectOuterSlice(MTLaunchStruct* mtls, uint32_t sliceNum) {
402bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
403bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    uint32_t r = sliceNum;
404bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    r = sliceInt(&mtls->fep.current.z, r, mtls->start.z, mtls->end.z);
405bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    r = sliceInt(&mtls->fep.current.lod, r, mtls->start.lod, mtls->end.lod);
406bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    r = sliceInt(&mtls->fep.current.face, r, mtls->start.face, mtls->end.face);
407bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    r = sliceInt(&mtls->fep.current.array[0], r, mtls->start.array[0], mtls->end.array[0]);
408bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    r = sliceInt(&mtls->fep.current.array[1], r, mtls->start.array[1], mtls->end.array[1]);
409bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    r = sliceInt(&mtls->fep.current.array[2], r, mtls->start.array[2], mtls->end.array[2]);
410bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    r = sliceInt(&mtls->fep.current.array[3], r, mtls->start.array[3], mtls->end.array[3]);
411bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    return r == 0;
412bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams}
413bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
414bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
415bf2111d3b3de310932099514f06924e48fa1d7b2Jason Samsstatic void walk_general(void *usr, uint32_t idx) {
416bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    MTLaunchStruct *mtls = (MTLaunchStruct *)usr;
417bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    RsExpandKernelDriverInfo fep = mtls->fep;
418bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    fep.lid = idx;
419bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    outer_foreach_t fn = (outer_foreach_t) mtls->kernel;
420bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
421bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
422bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    while(1) {
423bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1);
424bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
425bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        if (!SelectOuterSlice(mtls, slice)) {
426bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            return;
427bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        }
428bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
429bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        for (mtls->fep.current.y = mtls->start.y;
430bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams             mtls->fep.current.y < mtls->end.y;
431bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams             mtls->fep.current.y++) {
432bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
433bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            FepPtrSetup(mtls, &mtls->fep, mtls->start.x,
434bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                        mtls->fep.current.y, mtls->fep.current.z, mtls->fep.current.lod,
435bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                        (RsAllocationCubemapFace)mtls->fep.current.face,
436bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                        mtls->fep.current.array[0], mtls->fep.current.array[1],
437bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                        mtls->fep.current.array[2], mtls->fep.current.array[3]);
438bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
439bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            RsExpandKernelParams kparams;
440bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            kparamSetup(&kparams, &mtls->fep);
441bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            fn(&kparams, mtls->start.x, mtls->end.x, mtls->fep.outStride[0]);
442bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        }
443bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    }
444bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
445bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams}
446c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams
447f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailesstatic void walk_2d(void *usr, uint32_t idx) {
448c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    MTLaunchStruct *mtls = (MTLaunchStruct *)usr;
449c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    RsExpandKernelDriverInfo fep = mtls->fep;
450c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    fep.lid = idx;
451bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    outer_foreach_t fn = (outer_foreach_t) mtls->kernel;
452709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
453c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    while (1) {
454c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        uint32_t slice  = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1);
455bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        uint32_t yStart = mtls->start.y + slice * mtls->mSliceSize;
456c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        uint32_t yEnd   = yStart + mtls->mSliceSize;
457709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
458bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        yEnd = rsMin(yEnd, mtls->end.y);
459709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
460c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        if (yEnd <= yStart) {
461c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            return;
462c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        }
463709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
464c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        for (fep.current.y = yStart; fep.current.y < yEnd; fep.current.y++) {
465bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            FepPtrSetup(mtls, &fep, mtls->start.x, fep.current.y);
466709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
467c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            RsExpandKernelParams kparams;
468c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            kparamSetup(&kparams, &fep);
469709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
470bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            fn(&kparams, mtls->start.x, mtls->end.x, fep.outStride[0]);
471709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        }
472c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    }
473f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes}
4744b2bea3dc20865f3a198797702e19912a6a2171cStephen Hines
475f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailesstatic void walk_1d(void *usr, uint32_t idx) {
476c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    MTLaunchStruct *mtls = (MTLaunchStruct *)usr;
477c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    RsExpandKernelDriverInfo fep = mtls->fep;
478c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    fep.lid = idx;
479bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    outer_foreach_t fn = (outer_foreach_t) mtls->kernel;
48080ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
481c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    while (1) {
482c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        uint32_t slice  = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1);
483bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        uint32_t xStart = mtls->start.x + slice * mtls->mSliceSize;
484c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        uint32_t xEnd   = xStart + mtls->mSliceSize;
48580ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
486bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        xEnd = rsMin(xEnd, mtls->end.x);
48780ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
488c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        if (xEnd <= xStart) {
489c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            return;
490c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        }
49180ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
492bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        FepPtrSetup(mtls, &fep, xStart, 0);
49380ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
494c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        RsExpandKernelParams kparams;
495c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        kparamSetup(&kparams, &fep);
496f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes
497c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams        fn(&kparams, xStart, xEnd, fep.outStride[0]);
498c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams    }
499709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
500709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
501f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailesvoid RsdCpuReferenceImpl::launchThreads(const Allocation ** ains,
502f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                                        uint32_t inLen,
503f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                                        Allocation* aout,
504f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                                        const RsScriptCall* sc,
505f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                                        MTLaunchStruct* mtls) {
5064b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5074b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes    //android::StopWatch kernel_time("kernel time");
5084b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
509bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams    bool outerDims = (mtls->start.z != mtls->end.z) ||
510bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.face != mtls->end.face) ||
511bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.lod != mtls->end.lod) ||
512bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.array[0] != mtls->end.array[0]) ||
513bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.array[1] != mtls->end.array[1]) ||
514bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.array[2] != mtls->end.array[2]) ||
515bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                     (mtls->start.array[3] != mtls->end.array[3]);
516bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams
5174b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes    if ((mWorkers.mCount >= 1) && mtls->isThreadable && !mInForEach) {
5184b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        const size_t targetByteChunk = 16 * 1024;
5194b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        mInForEach = true;
520f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes
521bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        if (outerDims) {
522bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            // No fancy logic for chunk size
523bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            mtls->mSliceSize = 1;
524bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            launchThreads(walk_general, mtls);
525bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        } else if (mtls->fep.dim.y > 1) {
526c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            uint32_t s1 = mtls->fep.dim.y / ((mWorkers.mCount + 1) * 4);
5274b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            uint32_t s2 = 0;
5284b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5294b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            // This chooses our slice size to rate limit atomic ops to
5304b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            // one per 16k bytes of reads/writes.
531c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            if ((mtls->aout[0] != nullptr) && mtls->aout[0]->mHal.drvState.lod[0].stride) {
532c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                s2 = targetByteChunk / mtls->aout[0]->mHal.drvState.lod[0].stride;
5334b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            } else {
534f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                // We know that there is either an output or an input.
535c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                s2 = targetByteChunk / mtls->ains[0]->mHal.drvState.lod[0].stride;
5364b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5374b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            mtls->mSliceSize = rsMin(s1, s2);
5384b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5394b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            if(mtls->mSliceSize < 1) {
5404b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes                mtls->mSliceSize = 1;
5414b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5424b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
543f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes            launchThreads(walk_2d, mtls);
5444b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        } else {
545c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            uint32_t s1 = mtls->fep.dim.x / ((mWorkers.mCount + 1) * 4);
5464b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            uint32_t s2 = 0;
5474b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5484b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            // This chooses our slice size to rate limit atomic ops to
5494b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            // one per 16k bytes of reads/writes.
550c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams            if ((mtls->aout[0] != nullptr) && mtls->aout[0]->getType()->getElementSizeBytes()) {
551c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                s2 = targetByteChunk / mtls->aout[0]->getType()->getElementSizeBytes();
5524b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            } else {
553f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes                // We know that there is either an output or an input.
554c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams                s2 = targetByteChunk / mtls->ains[0]->getType()->getElementSizeBytes();
5554b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5564b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            mtls->mSliceSize = rsMin(s1, s2);
5574b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5584b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            if (mtls->mSliceSize < 1) {
5594b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes                mtls->mSliceSize = 1;
5604b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5614b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
562f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes            launchThreads(walk_1d, mtls);
5634b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        }
5644b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        mInForEach = false;
5654b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
5664b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes    } else {
5674b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        outer_foreach_t fn = (outer_foreach_t) mtls->kernel;
568bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        uint32_t slice = 0;
56980ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes
5704b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
571bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams        while(SelectOuterSlice(mtls, slice++)) {
572bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams            for (mtls->fep.current.y = mtls->start.y;
573bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                 mtls->fep.current.y < mtls->end.y;
574bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                 mtls->fep.current.y++) {
5754b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
576bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                FepPtrSetup(mtls, &mtls->fep, mtls->start.x,
577bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                            mtls->fep.current.y, mtls->fep.current.z, mtls->fep.current.lod,
578bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                            (RsAllocationCubemapFace) mtls->fep.current.face,
579bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                            mtls->fep.current.array[0], mtls->fep.current.array[1],
580bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                            mtls->fep.current.array[2], mtls->fep.current.array[3]);
5814b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
582bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                RsExpandKernelParams kparams;
583bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                kparamSetup(&kparams, &mtls->fep);
584bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams                fn(&kparams, mtls->start.x, mtls->end.x, mtls->fep.outStride[0]);
5854b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes            }
5864b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes        }
5874b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes    }
5884b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes}
5894b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes
590709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuScriptImpl * RsdCpuReferenceImpl::setTLS(RsdCpuScriptImpl *sc) {
591709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    //ALOGE("setTls %p", sc);
592709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey);
593709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    rsAssert(tls);
594709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    RsdCpuScriptImpl *old = tls->mImpl;
595709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    tls->mImpl = sc;
596709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    tls->mContext = mRSC;
597709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    if (sc) {
598709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        tls->mScript = sc->getScript();
599709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    } else {
60044bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        tls->mScript = nullptr;
601709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
602709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return old;
603709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
604709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
605709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst RsdCpuReference::CpuSymbol * RsdCpuReferenceImpl::symLookup(const char *name) {
606709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return mSymLookupFn(mRSC, name);
607709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
608709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
609709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
610709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createScript(const ScriptC *s,
611709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                                    char const *resName, char const *cacheDir,
612709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                                    uint8_t const *bitcode, size_t bitcodeSize,
613709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                                    uint32_t flags) {
614709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
615709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    RsdCpuScriptImpl *i = new RsdCpuScriptImpl(this, s);
616005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines    if (!i->init(resName, cacheDir, bitcode, bitcodeSize, flags
617005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines        , getBccPluginName()
618005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines        )) {
619709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        delete i;
62044bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        return nullptr;
621709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
622709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return i;
623709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
624709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
6257c4b888f2147edf99690b6af75470774ff31c43bJason Samsextern RsdCpuScriptImpl * rsdIntrinsic_3DLUT(RsdCpuReferenceImpl *ctx,
6267c4b888f2147edf99690b6af75470774ff31c43bJason Sams                                             const Script *s, const Element *e);
627c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve3x3(RsdCpuReferenceImpl *ctx,
628c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                                   const Script *s, const Element *e);
629c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_ColorMatrix(RsdCpuReferenceImpl *ctx,
630c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                                   const Script *s, const Element *e);
631c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_LUT(RsdCpuReferenceImpl *ctx,
632c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                           const Script *s, const Element *e);
633c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve5x5(RsdCpuReferenceImpl *ctx,
634c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                                   const Script *s, const Element *e);
635c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blur(RsdCpuReferenceImpl *ctx,
636c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                            const Script *s, const Element *e);
637c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_YuvToRGB(RsdCpuReferenceImpl *ctx,
638c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                                const Script *s, const Element *e);
639c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blend(RsdCpuReferenceImpl *ctx,
640c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams                                             const Script *s, const Element *e);
6412282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Histogram(RsdCpuReferenceImpl *ctx,
6422282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams                                                 const Script *s, const Element *e);
64339ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Resize(RsdCpuReferenceImpl *ctx,
64439ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams                                              const Script *s, const Element *e);
645709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
646709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createIntrinsic(const Script *s,
647709a0978ae141198018ca9769f8d96292a8928e6Jason Sams                                    RsScriptIntrinsicID iid, Element *e) {
648709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
64944bef6fba6244292b751387f3d6c31cca96c28adChris Wailes    RsdCpuScriptImpl *i = nullptr;
650709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    switch (iid) {
6517c4b888f2147edf99690b6af75470774ff31c43bJason Sams    case RS_SCRIPT_INTRINSIC_ID_3DLUT:
6527c4b888f2147edf99690b6af75470774ff31c43bJason Sams        i = rsdIntrinsic_3DLUT(this, s, e);
6537c4b888f2147edf99690b6af75470774ff31c43bJason Sams        break;
654709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_3x3:
655c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_Convolve3x3(this, s, e);
656709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
657709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_COLOR_MATRIX:
658c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_ColorMatrix(this, s, e);
659709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
660709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_LUT:
661c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_LUT(this, s, e);
662709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
663709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_5x5:
664c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_Convolve5x5(this, s, e);
665709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
666709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_BLUR:
667c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_Blur(this, s, e);
668709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
669709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_YUV_TO_RGB:
670c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_YuvToRGB(this, s, e);
671709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
672709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    case RS_SCRIPT_INTRINSIC_ID_BLEND:
673c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams        i = rsdIntrinsic_Blend(this, s, e);
674709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        break;
6752282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams    case RS_SCRIPT_INTRINSIC_ID_HISTOGRAM:
6762282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams        i = rsdIntrinsic_Histogram(this, s, e);
6772282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams        break;
67839ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams    case RS_SCRIPT_INTRINSIC_ID_RESIZE:
67939ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams        i = rsdIntrinsic_Resize(this, s, e);
68039ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams        break;
681709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
682709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    default:
683709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        rsAssert(0);
684709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
685709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
686709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    return i;
687709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
688709a0978ae141198018ca9769f8d96292a8928e6Jason Sams
6891ffd86b448d78366190c540f98f8b6d641cdb6cfYang Nivoid* RsdCpuReferenceImpl::createScriptGroup(const ScriptGroupBase *sg) {
6901ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni  switch (sg->getApiVersion()) {
6911ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni    case ScriptGroupBase::SG_V1: {
6921ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      CpuScriptGroupImpl *sgi = new CpuScriptGroupImpl(this, sg);
6931ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      if (!sgi->init()) {
694709a0978ae141198018ca9769f8d96292a8928e6Jason Sams        delete sgi;
69544bef6fba6244292b751387f3d6c31cca96c28adChris Wailes        return nullptr;
6961ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      }
6971ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      return sgi;
698709a0978ae141198018ca9769f8d96292a8928e6Jason Sams    }
6991ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni    case ScriptGroupBase::SG_V2: {
7001ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni      return new CpuScriptGroup2Impl(this, sg);
7011ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni    }
7021ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni  }
7031ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni  return nullptr;
704709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}
705