rsCpuCore.cpp revision 0b575de8ed0b628d84d256f5846500b0385979bd
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" 20709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 21709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <malloc.h> 22709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsContext.h" 23709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 24709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/types.h> 25709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/resource.h> 26709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sched.h> 27709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/syscall.h> 28709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <string.h> 290b575de8ed0b628d84d256f5846500b0385979bdTim Murray 300b575de8ed0b628d84d256f5846500b0385979bdTim Murray#ifndef RS_SERVER 310b575de8ed0b628d84d256f5846500b0385979bdTim Murray#include <cutils/properties.h> 32709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "utils/StopWatch.h" 330b575de8ed0b628d84d256f5846500b0385979bdTim Murray#endif 340b575de8ed0b628d84d256f5846500b0385979bdTim Murray 350b575de8ed0b628d84d256f5846500b0385979bdTim Murray#ifdef RS_SERVER 360b575de8ed0b628d84d256f5846500b0385979bdTim Murray// Android exposes gettid(), standard Linux does not 370b575de8ed0b628d84d256f5846500b0385979bdTim Murraystatic pid_t gettid() { 380b575de8ed0b628d84d256f5846500b0385979bdTim Murray return syscall(SYS_gettid); 390b575de8ed0b628d84d256f5846500b0385979bdTim Murray} 400b575de8ed0b628d84d256f5846500b0385979bdTim Murray#endif 41709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 42709a0978ae141198018ca9769f8d96292a8928e6Jason Samsusing namespace android; 43709a0978ae141198018ca9769f8d96292a8928e6Jason Samsusing namespace android::renderscript; 44709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 45709a0978ae141198018ca9769f8d96292a8928e6Jason Samstypedef void (*outer_foreach_t)( 46709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const android::renderscript::RsForEachStubParamStruct *, 47709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t x1, uint32_t x2, 48709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t instep, uint32_t outstep); 49709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 50709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 51709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_key_t gThreadTLSKey = 0; 52709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic uint32_t gThreadTLSKeyCount = 0; 53709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_mutex_t gInitMutex = PTHREAD_MUTEX_INITIALIZER; 54709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 55709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::~RsdCpuReference() { 56709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 57709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 58709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference * RsdCpuReference::create(Context *rsc, uint32_t version_major, 59cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn 60cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#ifndef RS_COMPATIBILITY_LIB 61cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams , bcc::RSLinkRuntimeCallback pLinkRuntimeCallback 62cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#endif 63cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams ) { 64709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 65709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *cpu = new RsdCpuReferenceImpl(rsc); 66709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!cpu) { 67709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 68709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 69709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!cpu->init(version_major, version_minor, lfn, slfn)) { 70709a0978ae141198018ca9769f8d96292a8928e6Jason Sams delete cpu; 71709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 72709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 73f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 74cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#ifndef RS_COMPATIBILITY_LIB 75f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines cpu->setLinkRuntimeCallback(pLinkRuntimeCallback); 76cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#endif 77f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 78709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return cpu; 79709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 80709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 81709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 82709a0978ae141198018ca9769f8d96292a8928e6Jason SamsContext * RsdCpuReference::getTlsContext() { 83709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 84709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return tls->mContext; 85709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 86709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 87709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst Script * RsdCpuReference::getTlsScript() { 88709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 89709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return tls->mScript; 90709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 91709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 92f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hinespthread_key_t RsdCpuReference::getThreadTLSKey(){ return gThreadTLSKey; } 93709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 94709a0978ae141198018ca9769f8d96292a8928e6Jason Sams//////////////////////////////////////////////////////////// 95709a0978ae141198018ca9769f8d96292a8928e6Jason Sams/// 96709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 97709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::RsdCpuReferenceImpl(Context *rsc) { 98709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mRSC = rsc; 99709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 100709a0978ae141198018ca9769f8d96292a8928e6Jason Sams version_major = 0; 101709a0978ae141198018ca9769f8d96292a8928e6Jason Sams version_minor = 0; 102709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mInForEach = false; 103709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&mWorkers, 0, sizeof(mWorkers)); 104709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&mTlsStruct, 0, sizeof(mTlsStruct)); 105709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mExit = false; 106cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#ifndef RS_COMPATIBILITY_LIB 107f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines mLinkRuntimeCallback = NULL; 108cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#endif 109709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 111709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 112709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid * RsdCpuReferenceImpl::helperThreadProc(void *vrsc) { 113709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *dc = (RsdCpuReferenceImpl *)vrsc; 114709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1150b575de8ed0b628d84d256f5846500b0385979bdTim Murray uint32_t idx = __sync_fetch_and_add(&dc->mWorkers.mLaunchCount, 1); 116709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 117709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGV("RS helperThread starting %p idx=%i", dc, idx); 118709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 119709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mLaunchSignals[idx].init(); 120709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mNativeThreadId[idx] = gettid(); 121709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 122709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&dc->mTlsStruct, 0, sizeof(dc->mTlsStruct)); 123709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int status = pthread_setspecific(gThreadTLSKey, &dc->mTlsStruct); 124709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 125709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("pthread_setspecific %i", status); 126709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 127709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 128709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#if 0 129709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef struct {uint64_t bits[1024 / 64]; } cpu_set_t; 130709a0978ae141198018ca9769f8d96292a8928e6Jason Sams cpu_set_t cpuset; 131709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&cpuset, 0, sizeof(cpuset)); 132709a0978ae141198018ca9769f8d96292a8928e6Jason Sams cpuset.bits[idx / 64] |= 1ULL << (idx % 64); 133709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int ret = syscall(241, rsc->mWorkers.mNativeThreadId[idx], 134709a0978ae141198018ca9769f8d96292a8928e6Jason Sams sizeof(cpuset), &cpuset); 135709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("SETAFFINITY ret = %i %s", ret, EGLUtils::strerror(ret)); 136709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 137709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 138709a0978ae141198018ca9769f8d96292a8928e6Jason Sams while (!dc->mExit) { 139709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mLaunchSignals[idx].wait(); 140709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (dc->mWorkers.mLaunchCallback) { 141709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // idx +1 is used because the calling thread is always worker 0. 142709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mLaunchCallback(dc->mWorkers.mLaunchData, idx+1); 143709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 1440b575de8ed0b628d84d256f5846500b0385979bdTim Murray __sync_fetch_and_sub(&dc->mWorkers.mRunningCount, 1); 145709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mCompleteSignal.set(); 146709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 147709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 148709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGV("RS helperThread exited %p idx=%i", dc, idx); 149709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 150709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 151709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 152709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::launchThreads(WorkerCallback_t cbk, void *data) { 153709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchData = data; 154709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchCallback = cbk; 1554d252d6e807b89764dad123ac845df298c52ca97Tim Murray 1564d252d6e807b89764dad123ac845df298c52ca97Tim Murray // fast path for very small launches 1574d252d6e807b89764dad123ac845df298c52ca97Tim Murray MTLaunchStruct *mtls = (MTLaunchStruct *)data; 1584d252d6e807b89764dad123ac845df298c52ca97Tim Murray if (mtls && mtls->fep.dimY <= 1 && mtls->xEnd <= mtls->xStart + mtls->mSliceSize) { 1594d252d6e807b89764dad123ac845df298c52ca97Tim Murray if (mWorkers.mLaunchCallback) { 1604d252d6e807b89764dad123ac845df298c52ca97Tim Murray mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0); 1614d252d6e807b89764dad123ac845df298c52ca97Tim Murray } 1624d252d6e807b89764dad123ac845df298c52ca97Tim Murray return; 1634d252d6e807b89764dad123ac845df298c52ca97Tim Murray } 1644d252d6e807b89764dad123ac845df298c52ca97Tim Murray 1650b575de8ed0b628d84d256f5846500b0385979bdTim Murray mWorkers.mRunningCount = mWorkers.mCount; 1660b575de8ed0b628d84d256f5846500b0385979bdTim Murray __sync_synchronize(); 1670b575de8ed0b628d84d256f5846500b0385979bdTim Murray 168709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 169709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchSignals[ct].set(); 170709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 171709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 172709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // We use the calling thread as one of the workers so we can start without 173709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // the delay of the thread wakeup. 174709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (mWorkers.mLaunchCallback) { 1754d252d6e807b89764dad123ac845df298c52ca97Tim Murray mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0); 176709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 177709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1780b575de8ed0b628d84d256f5846500b0385979bdTim Murray while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) { 179709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCompleteSignal.wait(); 180709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 181709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 182709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 183709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 184709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::lockMutex() { 185709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_mutex_lock(&gInitMutex); 186709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 187709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 188709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::unlockMutex() { 189709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_mutex_unlock(&gInitMutex); 190709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 191709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 192709a0978ae141198018ca9769f8d96292a8928e6Jason Samsbool RsdCpuReferenceImpl::init(uint32_t version_major, uint32_t version_minor, 193709a0978ae141198018ca9769f8d96292a8928e6Jason Sams sym_lookup_t lfn, script_lookup_t slfn) { 194709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 195709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mSymLookupFn = lfn; 196709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mScriptLookupFn = slfn; 197709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 198709a0978ae141198018ca9769f8d96292a8928e6Jason Sams lockMutex(); 199709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!gThreadTLSKeyCount) { 200709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int status = pthread_key_create(&gThreadTLSKey, NULL); 201709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 202709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("Failed to init thread tls key."); 203709a0978ae141198018ca9769f8d96292a8928e6Jason Sams unlockMutex(); 204709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return false; 205709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 206709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 207709a0978ae141198018ca9769f8d96292a8928e6Jason Sams gThreadTLSKeyCount++; 208709a0978ae141198018ca9769f8d96292a8928e6Jason Sams unlockMutex(); 209709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 210709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mTlsStruct.mContext = mRSC; 211709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mTlsStruct.mScript = NULL; 212709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int status = pthread_setspecific(gThreadTLSKey, &mTlsStruct); 213709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 214709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("pthread_setspecific %i", status); 215709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 216709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 217709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int cpu = sysconf(_SC_NPROCESSORS_ONLN); 218709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if(mRSC->props.mDebugMaxThreads) { 219709a0978ae141198018ca9769f8d96292a8928e6Jason Sams cpu = mRSC->props.mDebugMaxThreads; 220709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 221709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (cpu < 2) { 222709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCount = 0; 223709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return true; 224709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 225709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 226709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // Subtract one from the cpu count because we also use the command thread as a worker. 227709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCount = (uint32_t)(cpu - 1); 228709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 229709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGV("%p Launching thread(s), CPUs %i", mRSC, mWorkers.mCount); 230709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 231709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mThreadId = (pthread_t *) calloc(mWorkers.mCount, sizeof(pthread_t)); 232709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mNativeThreadId = (pid_t *) calloc(mWorkers.mCount, sizeof(pid_t)); 233709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchSignals = new Signal[mWorkers.mCount]; 234709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchCallback = NULL; 235709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 236709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCompleteSignal.init(); 237709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 2380b575de8ed0b628d84d256f5846500b0385979bdTim Murray mWorkers.mRunningCount = mWorkers.mCount; 2390b575de8ed0b628d84d256f5846500b0385979bdTim Murray mWorkers.mLaunchCount = 0; 2400b575de8ed0b628d84d256f5846500b0385979bdTim Murray __sync_synchronize(); 241709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 242709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_attr_t threadAttr; 243709a0978ae141198018ca9769f8d96292a8928e6Jason Sams status = pthread_attr_init(&threadAttr); 244709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 245709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("Failed to init thread attribute."); 246709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return false; 247709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 248709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 249709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct=0; ct < mWorkers.mCount; ct++) { 250709a0978ae141198018ca9769f8d96292a8928e6Jason Sams status = pthread_create(&mWorkers.mThreadId[ct], &threadAttr, helperThreadProc, this); 251709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 252709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCount = ct; 253709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("Created fewer than expected number of RS threads."); 254709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 255709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 256709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 2570b575de8ed0b628d84d256f5846500b0385979bdTim Murray while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) { 258709a0978ae141198018ca9769f8d96292a8928e6Jason Sams usleep(100); 259709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 260709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 261709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_attr_destroy(&threadAttr); 262709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return true; 263709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 264709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 265709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 266709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::setPriority(int32_t priority) { 267709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct=0; ct < mWorkers.mCount; ct++) { 268709a0978ae141198018ca9769f8d96292a8928e6Jason Sams setpriority(PRIO_PROCESS, mWorkers.mNativeThreadId[ct], priority); 269709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 270709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 271709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 272709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::~RsdCpuReferenceImpl() { 273709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mExit = true; 274709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchData = NULL; 275709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchCallback = NULL; 2760b575de8ed0b628d84d256f5846500b0385979bdTim Murray mWorkers.mRunningCount = mWorkers.mCount; 2770b575de8ed0b628d84d256f5846500b0385979bdTim Murray __sync_synchronize(); 278709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 279709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchSignals[ct].set(); 280709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 281709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void *res; 282709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 283709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_join(mWorkers.mThreadId[ct], &res); 284709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 2850b575de8ed0b628d84d256f5846500b0385979bdTim Murray rsAssert(__sync_fetch_and_or(&mWorkers.mRunningCount, 0) == 0); 286709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 287709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // Global structure cleanup. 288709a0978ae141198018ca9769f8d96292a8928e6Jason Sams lockMutex(); 289709a0978ae141198018ca9769f8d96292a8928e6Jason Sams --gThreadTLSKeyCount; 290709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!gThreadTLSKeyCount) { 291709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_key_delete(gThreadTLSKey); 292709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 293709a0978ae141198018ca9769f8d96292a8928e6Jason Sams unlockMutex(); 294709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 295709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 296709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 297709a0978ae141198018ca9769f8d96292a8928e6Jason Samstypedef void (*rs_t)(const void *, void *, const void *, uint32_t, uint32_t, uint32_t, uint32_t); 298709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 299709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic void wc_xy(void *usr, uint32_t idx) { 300709a0978ae141198018ca9769f8d96292a8928e6Jason Sams MTLaunchStruct *mtls = (MTLaunchStruct *)usr; 301709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsForEachStubParamStruct p; 302709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memcpy(&p, &mtls->fep, sizeof(p)); 303709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.lid = idx; 304709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t sig = mtls->sig; 305709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 306709a0978ae141198018ca9769f8d96292a8928e6Jason Sams outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 307709a0978ae141198018ca9769f8d96292a8928e6Jason Sams while (1) { 3080b575de8ed0b628d84d256f5846500b0385979bdTim Murray uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 309709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t yStart = mtls->yStart + slice * mtls->mSliceSize; 310709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t yEnd = yStart + mtls->mSliceSize; 311709a0978ae141198018ca9769f8d96292a8928e6Jason Sams yEnd = rsMin(yEnd, mtls->yEnd); 312709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (yEnd <= yStart) { 313709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return; 314709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 315709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 316709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("usr idx %i, x %i,%i y %i,%i", idx, mtls->xStart, mtls->xEnd, yStart, yEnd); 317709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("usr ptr in %p, out %p", mtls->fep.ptrIn, mtls->fep.ptrOut); 318709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 319709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (p.y = yStart; p.y < yEnd; p.y++) { 320709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.out = mtls->fep.ptrOut + (mtls->fep.yStrideOut * p.y) + 321709a0978ae141198018ca9769f8d96292a8928e6Jason Sams (mtls->fep.eStrideOut * mtls->xStart); 322709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.in = mtls->fep.ptrIn + (mtls->fep.yStrideIn * p.y) + 323709a0978ae141198018ca9769f8d96292a8928e6Jason Sams (mtls->fep.eStrideIn * mtls->xStart); 324709a0978ae141198018ca9769f8d96292a8928e6Jason Sams fn(&p, mtls->xStart, mtls->xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut); 325709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 326709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 327709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 328709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 329709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic void wc_x(void *usr, uint32_t idx) { 330709a0978ae141198018ca9769f8d96292a8928e6Jason Sams MTLaunchStruct *mtls = (MTLaunchStruct *)usr; 331709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsForEachStubParamStruct p; 332709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memcpy(&p, &mtls->fep, sizeof(p)); 333709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.lid = idx; 334709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t sig = mtls->sig; 335709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 336709a0978ae141198018ca9769f8d96292a8928e6Jason Sams outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 337709a0978ae141198018ca9769f8d96292a8928e6Jason Sams while (1) { 3380b575de8ed0b628d84d256f5846500b0385979bdTim Murray uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 339709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t xStart = mtls->xStart + slice * mtls->mSliceSize; 340709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t xEnd = xStart + mtls->mSliceSize; 341709a0978ae141198018ca9769f8d96292a8928e6Jason Sams xEnd = rsMin(xEnd, mtls->xEnd); 342709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (xEnd <= xStart) { 343709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return; 344709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 345709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 346709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("usr slice %i idx %i, x %i,%i", slice, idx, xStart, xEnd); 347709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("usr ptr in %p, out %p", mtls->fep.ptrIn, mtls->fep.ptrOut); 348709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 349709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.out = mtls->fep.ptrOut + (mtls->fep.eStrideOut * xStart); 350709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.in = mtls->fep.ptrIn + (mtls->fep.eStrideIn * xStart); 351709a0978ae141198018ca9769f8d96292a8928e6Jason Sams fn(&p, xStart, xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut); 352709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 353709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 354709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 355709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::launchThreads(const Allocation * ain, Allocation * aout, 356709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const RsScriptCall *sc, MTLaunchStruct *mtls) { 357709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 358709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //android::StopWatch kernel_time("kernel time"); 359709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 360709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if ((mWorkers.mCount >= 1) && mtls->isThreadable && !mInForEach) { 361709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const size_t targetByteChunk = 16 * 1024; 362709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mInForEach = true; 363709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (mtls->fep.dimY > 1) { 364709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t s1 = mtls->fep.dimY / ((mWorkers.mCount + 1) * 4); 365709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t s2 = 0; 366709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 367709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // This chooses our slice size to rate limit atomic ops to 368709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // one per 16k bytes of reads/writes. 369709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (mtls->fep.yStrideOut) { 370709a0978ae141198018ca9769f8d96292a8928e6Jason Sams s2 = targetByteChunk / mtls->fep.yStrideOut; 371709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 372709a0978ae141198018ca9769f8d96292a8928e6Jason Sams s2 = targetByteChunk / mtls->fep.yStrideIn; 373709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 374709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->mSliceSize = rsMin(s1, s2); 375709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 376709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if(mtls->mSliceSize < 1) { 377709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->mSliceSize = 1; 378709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 379709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 380709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // mtls->mSliceSize = 2; 381709a0978ae141198018ca9769f8d96292a8928e6Jason Sams launchThreads(wc_xy, mtls); 382709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 383709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t s1 = mtls->fep.dimX / ((mWorkers.mCount + 1) * 4); 384709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t s2 = 0; 385709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 386709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // This chooses our slice size to rate limit atomic ops to 387709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // one per 16k bytes of reads/writes. 388709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (mtls->fep.eStrideOut) { 389709a0978ae141198018ca9769f8d96292a8928e6Jason Sams s2 = targetByteChunk / mtls->fep.eStrideOut; 390709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 391709a0978ae141198018ca9769f8d96292a8928e6Jason Sams s2 = targetByteChunk / mtls->fep.eStrideIn; 392709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 393709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->mSliceSize = rsMin(s1, s2); 394709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 395709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if(mtls->mSliceSize < 1) { 396709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->mSliceSize = 1; 397709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 398709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 399709a0978ae141198018ca9769f8d96292a8928e6Jason Sams launchThreads(wc_x, mtls); 400709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 401709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mInForEach = false; 402709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 403709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("launch 1"); 404709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 405709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsForEachStubParamStruct p; 406709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memcpy(&p, &mtls->fep, sizeof(p)); 407709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t sig = mtls->sig; 408709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 409709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("launch 3"); 410709a0978ae141198018ca9769f8d96292a8928e6Jason Sams outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 411709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (p.ar[0] = mtls->arrayStart; p.ar[0] < mtls->arrayEnd; p.ar[0]++) { 412709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (p.z = mtls->zStart; p.z < mtls->zEnd; p.z++) { 413709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (p.y = mtls->yStart; p.y < mtls->yEnd; p.y++) { 414709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t offset = mtls->fep.dimY * mtls->fep.dimZ * p.ar[0] + 415709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->fep.dimY * p.z + p.y; 416709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.out = mtls->fep.ptrOut + (mtls->fep.yStrideOut * offset) + 417709a0978ae141198018ca9769f8d96292a8928e6Jason Sams (mtls->fep.eStrideOut * mtls->xStart); 418709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.in = mtls->fep.ptrIn + (mtls->fep.yStrideIn * offset) + 419709a0978ae141198018ca9769f8d96292a8928e6Jason Sams (mtls->fep.eStrideIn * mtls->xStart); 420709a0978ae141198018ca9769f8d96292a8928e6Jason Sams fn(&p, mtls->xStart, mtls->xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut); 421709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 422709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 423709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 424709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 425709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 426709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 427709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuScriptImpl * RsdCpuReferenceImpl::setTLS(RsdCpuScriptImpl *sc) { 428709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("setTls %p", sc); 429709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 430709a0978ae141198018ca9769f8d96292a8928e6Jason Sams rsAssert(tls); 431709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl *old = tls->mImpl; 432709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mImpl = sc; 433709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mContext = mRSC; 434709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (sc) { 435709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mScript = sc->getScript(); 436709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 437709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mScript = NULL; 438709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 439709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return old; 440709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 441709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 442709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst RsdCpuReference::CpuSymbol * RsdCpuReferenceImpl::symLookup(const char *name) { 443709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return mSymLookupFn(mRSC, name); 444709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 445709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 446709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 447709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createScript(const ScriptC *s, 448709a0978ae141198018ca9769f8d96292a8928e6Jason Sams char const *resName, char const *cacheDir, 449709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint8_t const *bitcode, size_t bitcodeSize, 450709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t flags) { 451709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 452709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl *i = new RsdCpuScriptImpl(this, s); 453709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!i->init(resName, cacheDir, bitcode, bitcodeSize, flags)) { 454709a0978ae141198018ca9769f8d96292a8928e6Jason Sams delete i; 455709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 456709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 457709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return i; 458709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 459709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 4607c4b888f2147edf99690b6af75470774ff31c43bJason Samsextern RsdCpuScriptImpl * rsdIntrinsic_3DLUT(RsdCpuReferenceImpl *ctx, 4617c4b888f2147edf99690b6af75470774ff31c43bJason Sams const Script *s, const Element *e); 462c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve3x3(RsdCpuReferenceImpl *ctx, 463c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 464c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_ColorMatrix(RsdCpuReferenceImpl *ctx, 465c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 466c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_LUT(RsdCpuReferenceImpl *ctx, 467c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 468c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve5x5(RsdCpuReferenceImpl *ctx, 469c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 470c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blur(RsdCpuReferenceImpl *ctx, 471c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 472c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_YuvToRGB(RsdCpuReferenceImpl *ctx, 473c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 474c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blend(RsdCpuReferenceImpl *ctx, 475c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 476709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 477709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createIntrinsic(const Script *s, 478709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsScriptIntrinsicID iid, Element *e) { 479709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 480709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl *i = NULL; 481709a0978ae141198018ca9769f8d96292a8928e6Jason Sams switch (iid) { 4827c4b888f2147edf99690b6af75470774ff31c43bJason Sams case RS_SCRIPT_INTRINSIC_ID_3DLUT: 4837c4b888f2147edf99690b6af75470774ff31c43bJason Sams i = rsdIntrinsic_3DLUT(this, s, e); 4847c4b888f2147edf99690b6af75470774ff31c43bJason Sams break; 485709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_3x3: 486c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Convolve3x3(this, s, e); 487709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 488709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_COLOR_MATRIX: 489c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_ColorMatrix(this, s, e); 490709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 491709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_LUT: 492c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_LUT(this, s, e); 493709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 494709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_5x5: 495c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Convolve5x5(this, s, e); 496709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 497709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_BLUR: 498c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Blur(this, s, e); 499709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 500709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_YUV_TO_RGB: 501c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_YuvToRGB(this, s, e); 502709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 503709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_BLEND: 504c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Blend(this, s, e); 505709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 506709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 507709a0978ae141198018ca9769f8d96292a8928e6Jason Sams default: 508709a0978ae141198018ca9769f8d96292a8928e6Jason Sams rsAssert(0); 509709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 510709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 511709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return i; 512709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 513709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 514709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScriptGroup * RsdCpuReferenceImpl::createScriptGroup(const ScriptGroup *sg) { 515709a0978ae141198018ca9769f8d96292a8928e6Jason Sams CpuScriptGroupImpl *sgi = new CpuScriptGroupImpl(this, sg); 516709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!sgi->init()) { 517709a0978ae141198018ca9769f8d96292a8928e6Jason Sams delete sgi; 518709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 519709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 520709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return sgi; 521709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 522709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 523709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 524