1709a0978ae141198018ca9769f8d96292a8928e6Jason Sams/* 2709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Copyright (C) 2012 The Android Open Source Project 3709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * 4709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 5709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * you may not use this file except in compliance with the License. 6709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * You may obtain a copy of the License at 7709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * 8709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * http://www.apache.org/licenses/LICENSE-2.0 9709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * 10709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Unless required by applicable law or agreed to in writing, software 11709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * distributed under the License is distributed on an "AS IS" BASIS, 12709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * See the License for the specific language governing permissions and 14709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * limitations under the License. 15709a0978ae141198018ca9769f8d96292a8928e6Jason Sams */ 16709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 17709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuCore.h" 18709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuScript.h" 19709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuScriptGroup.h" 201ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni#include "rsCpuScriptGroup2.h" 21709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 22709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <malloc.h> 23709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsContext.h" 24709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 25709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/types.h> 26709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/resource.h> 27709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sched.h> 28709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/syscall.h> 2911fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala#include <stdio.h> 30709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <string.h> 31b0934b67b95cc27e2358c2aa4db5f7c1067c8f9bStephen Hines#include <unistd.h> 320b575de8ed0b628d84d256f5846500b0385979bdTim Murray 33ae2ec3febedfc29376b9104413fb4042028f1265David Gross#define REDUCE_ALOGV(mtls, level, ...) do { if ((mtls)->logReduce >= (level)) ALOGV(__VA_ARGS__); } while(0) 3435dbc8c20814997a00e06d7798a5633c37706978David Gross 35709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_key_t gThreadTLSKey = 0; 36709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic uint32_t gThreadTLSKeyCount = 0; 37709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_mutex_t gInitMutex = PTHREAD_MUTEX_INITIALIZER; 38709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 39462de21ac2e1773b99aedee012adb374e476ae36Chih-Hung Hsiehnamespace android { 40462de21ac2e1773b99aedee012adb374e476ae36Chih-Hung Hsiehnamespace renderscript { 41462de21ac2e1773b99aedee012adb374e476ae36Chih-Hung Hsieh 42462de21ac2e1773b99aedee012adb374e476ae36Chih-Hung Hsiehbool gArchUseSIMD = false; 43f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams 44709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::~RsdCpuReference() { 45709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 46709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 47709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference * RsdCpuReference::create(Context *rsc, uint32_t version_major, 48cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn 49b043df0676fef226336deb3a00ead2f31e02343fDavid Gross , RSSelectRTCallback pSelectRTCallback, 50005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines const char *pBccPluginName 51cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams ) { 52709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 53709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *cpu = new RsdCpuReferenceImpl(rsc); 54709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!cpu) { 5544bef6fba6244292b751387f3d6c31cca96c28adChris Wailes return nullptr; 56709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 57709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!cpu->init(version_major, version_minor, lfn, slfn)) { 58709a0978ae141198018ca9769f8d96292a8928e6Jason Sams delete cpu; 5944bef6fba6244292b751387f3d6c31cca96c28adChris Wailes return nullptr; 60709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 61f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 621d476620399d54774e4fd386c1d23cc583d49522Stephen Hines cpu->setSelectRTCallback(pSelectRTCallback); 63005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines if (pBccPluginName) { 64005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines cpu->setBccPluginName(pBccPluginName); 65005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines } 66f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 67709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return cpu; 68709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 69709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 70709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 71709a0978ae141198018ca9769f8d96292a8928e6Jason SamsContext * RsdCpuReference::getTlsContext() { 72709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 73709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return tls->mContext; 74709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 75709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 76709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst Script * RsdCpuReference::getTlsScript() { 77709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 78709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return tls->mScript; 79709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 80709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 81f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hinespthread_key_t RsdCpuReference::getThreadTLSKey(){ return gThreadTLSKey; } 82709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 83709a0978ae141198018ca9769f8d96292a8928e6Jason Sams//////////////////////////////////////////////////////////// 84709a0978ae141198018ca9769f8d96292a8928e6Jason Sams/// 85709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 86709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::RsdCpuReferenceImpl(Context *rsc) { 87709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mRSC = rsc; 88709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 89709a0978ae141198018ca9769f8d96292a8928e6Jason Sams version_major = 0; 90709a0978ae141198018ca9769f8d96292a8928e6Jason Sams version_minor = 0; 9135dbc8c20814997a00e06d7798a5633c37706978David Gross mInKernel = false; 92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&mWorkers, 0, sizeof(mWorkers)); 93709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&mTlsStruct, 0, sizeof(mTlsStruct)); 94709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mExit = false; 9544bef6fba6244292b751387f3d6c31cca96c28adChris Wailes mSelectRTCallback = nullptr; 968409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines mEmbedGlobalInfo = true; 978409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines mEmbedGlobalInfoSkipConstant = true; 98709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 99709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 100709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 101709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid * RsdCpuReferenceImpl::helperThreadProc(void *vrsc) { 102709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *dc = (RsdCpuReferenceImpl *)vrsc; 103709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1040b575de8ed0b628d84d256f5846500b0385979bdTim Murray uint32_t idx = __sync_fetch_and_add(&dc->mWorkers.mLaunchCount, 1); 105709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 106709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGV("RS helperThread starting %p idx=%i", dc, idx); 107709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 108709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mLaunchSignals[idx].init(); 109709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mNativeThreadId[idx] = gettid(); 110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 111709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&dc->mTlsStruct, 0, sizeof(dc->mTlsStruct)); 112709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int status = pthread_setspecific(gThreadTLSKey, &dc->mTlsStruct); 113709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 114709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("pthread_setspecific %i", status); 115709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 116709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 117709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#if 0 118709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef struct {uint64_t bits[1024 / 64]; } cpu_set_t; 119709a0978ae141198018ca9769f8d96292a8928e6Jason Sams cpu_set_t cpuset; 120709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&cpuset, 0, sizeof(cpuset)); 121709a0978ae141198018ca9769f8d96292a8928e6Jason Sams cpuset.bits[idx / 64] |= 1ULL << (idx % 64); 122709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int ret = syscall(241, rsc->mWorkers.mNativeThreadId[idx], 123709a0978ae141198018ca9769f8d96292a8928e6Jason Sams sizeof(cpuset), &cpuset); 124709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("SETAFFINITY ret = %i %s", ret, EGLUtils::strerror(ret)); 125709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 126709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 127709a0978ae141198018ca9769f8d96292a8928e6Jason Sams while (!dc->mExit) { 128709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mLaunchSignals[idx].wait(); 129709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (dc->mWorkers.mLaunchCallback) { 130709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // idx +1 is used because the calling thread is always worker 0. 131709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mLaunchCallback(dc->mWorkers.mLaunchData, idx+1); 132709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 1330b575de8ed0b628d84d256f5846500b0385979bdTim Murray __sync_fetch_and_sub(&dc->mWorkers.mRunningCount, 1); 134709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mCompleteSignal.set(); 135709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 136709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 137709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGV("RS helperThread exited %p idx=%i", dc, idx); 13844bef6fba6244292b751387f3d6c31cca96c28adChris Wailes return nullptr; 139709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 140709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 14114ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// Launch a kernel. 14214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// The callback function is called to execute the kernel. 143709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::launchThreads(WorkerCallback_t cbk, void *data) { 144709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchData = data; 145709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchCallback = cbk; 1464d252d6e807b89764dad123ac845df298c52ca97Tim Murray 1474d252d6e807b89764dad123ac845df298c52ca97Tim Murray // fast path for very small launches 14814ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala MTLaunchStructCommon *mtls = (MTLaunchStructCommon *)data; 14914ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala if (mtls && mtls->dimPtr->y <= 1 && mtls->end.x <= mtls->start.x + mtls->mSliceSize) { 1504d252d6e807b89764dad123ac845df298c52ca97Tim Murray if (mWorkers.mLaunchCallback) { 1514d252d6e807b89764dad123ac845df298c52ca97Tim Murray mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0); 1524d252d6e807b89764dad123ac845df298c52ca97Tim Murray } 1534d252d6e807b89764dad123ac845df298c52ca97Tim Murray return; 1544d252d6e807b89764dad123ac845df298c52ca97Tim Murray } 1554d252d6e807b89764dad123ac845df298c52ca97Tim Murray 1560b575de8ed0b628d84d256f5846500b0385979bdTim Murray mWorkers.mRunningCount = mWorkers.mCount; 1570b575de8ed0b628d84d256f5846500b0385979bdTim Murray __sync_synchronize(); 1580b575de8ed0b628d84d256f5846500b0385979bdTim Murray 159709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 160709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchSignals[ct].set(); 161709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 162709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 163709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // We use the calling thread as one of the workers so we can start without 164709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // the delay of the thread wakeup. 165709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (mWorkers.mLaunchCallback) { 1664d252d6e807b89764dad123ac845df298c52ca97Tim Murray mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0); 167709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 168709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1690b575de8ed0b628d84d256f5846500b0385979bdTim Murray while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) { 170709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCompleteSignal.wait(); 171709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 172709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 173709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 174709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 175709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::lockMutex() { 176709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_mutex_lock(&gInitMutex); 177709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 178709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 179709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::unlockMutex() { 180709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_mutex_unlock(&gInitMutex); 181709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 182709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 18311fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala// Determine if the CPU we're running on supports SIMD instructions. 184f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Samsstatic void GetCpuInfo() { 18511fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala // Read the CPU flags from /proc/cpuinfo. 18611fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala FILE *cpuinfo = fopen("/proc/cpuinfo", "r"); 187f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams 18811fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala if (!cpuinfo) { 189f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams return; 190f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams } 191f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams 19211fd9ec1ab8dfa7ae45c6edeea48dddc4633efeaMatt Wala char cpuinfostr[4096]; 1935d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang // fgets() ends with newline or EOF, need to check the whole 1945d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang // "cpuinfo" file to make sure we can use SIMD or not. 1955d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang while (fgets(cpuinfostr, sizeof(cpuinfostr), cpuinfo)) { 196074424a4ac5b093331df2c92e7a5bcbfff136b71Jason Sams#if defined(ARCH_ARM_HAVE_VFP) || defined(ARCH_ARM_USE_INTRINSICS) 1975d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang gArchUseSIMD = strstr(cpuinfostr, " neon") || strstr(cpuinfostr, " asimd"); 1987b7060c61e4182b29186849c5a857ea5f0898e56Rose, James#elif defined(ARCH_X86_HAVE_SSSE3) 1995d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang gArchUseSIMD = strstr(cpuinfostr, " ssse3"); 2007b7060c61e4182b29186849c5a857ea5f0898e56Rose, James#endif 2015d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang if (gArchUseSIMD) { 2025d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang break; 2035d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang } 2045d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang } 2055d70cb591d78d62d10839a52302ec9087c6f3350Miao Wang fclose(cpuinfo); 206f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams} 207f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams 208709a0978ae141198018ca9769f8d96292a8928e6Jason Samsbool RsdCpuReferenceImpl::init(uint32_t version_major, uint32_t version_minor, 209709a0978ae141198018ca9769f8d96292a8928e6Jason Sams sym_lookup_t lfn, script_lookup_t slfn) { 210709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mSymLookupFn = lfn; 211709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mScriptLookupFn = slfn; 212709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 213709a0978ae141198018ca9769f8d96292a8928e6Jason Sams lockMutex(); 214709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!gThreadTLSKeyCount) { 21544bef6fba6244292b751387f3d6c31cca96c28adChris Wailes int status = pthread_key_create(&gThreadTLSKey, nullptr); 216709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 217709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("Failed to init thread tls key."); 218709a0978ae141198018ca9769f8d96292a8928e6Jason Sams unlockMutex(); 219709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return false; 220709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 221709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 222709a0978ae141198018ca9769f8d96292a8928e6Jason Sams gThreadTLSKeyCount++; 223709a0978ae141198018ca9769f8d96292a8928e6Jason Sams unlockMutex(); 224709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 225709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mTlsStruct.mContext = mRSC; 22644bef6fba6244292b751387f3d6c31cca96c28adChris Wailes mTlsStruct.mScript = nullptr; 227709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int status = pthread_setspecific(gThreadTLSKey, &mTlsStruct); 228709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 229709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("pthread_setspecific %i", status); 230709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 231709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 23235dbc8c20814997a00e06d7798a5633c37706978David Gross mPageSize = sysconf(_SC_PAGE_SIZE); 233013ff53dc8d6c2464e1b5e1ea8c50b68f7b4aa73David Gross // ALOGV("page size = %ld", mPageSize); 23435dbc8c20814997a00e06d7798a5633c37706978David Gross 235f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams GetCpuInfo(); 236f5ef8df639ba6363aa5d546e57ce872d04144cb6Jason Sams 23777d57a305f4134e78ebc91869011c4009988104eJason Sams int cpu = sysconf(_SC_NPROCESSORS_CONF); 238709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if(mRSC->props.mDebugMaxThreads) { 239709a0978ae141198018ca9769f8d96292a8928e6Jason Sams cpu = mRSC->props.mDebugMaxThreads; 240709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 241709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (cpu < 2) { 242709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCount = 0; 243709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return true; 244709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 245709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 246709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // Subtract one from the cpu count because we also use the command thread as a worker. 247709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCount = (uint32_t)(cpu - 1); 248709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 249554054c2317cc38df1e57186caf171f1b07ec6e3Yang Ni if (mRSC->props.mLogScripts) { 250554054c2317cc38df1e57186caf171f1b07ec6e3Yang Ni ALOGV("%p Launching thread(s), CPUs %i", mRSC, mWorkers.mCount + 1); 251554054c2317cc38df1e57186caf171f1b07ec6e3Yang Ni } 252709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 253709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mThreadId = (pthread_t *) calloc(mWorkers.mCount, sizeof(pthread_t)); 254709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mNativeThreadId = (pid_t *) calloc(mWorkers.mCount, sizeof(pid_t)); 255709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchSignals = new Signal[mWorkers.mCount]; 25644bef6fba6244292b751387f3d6c31cca96c28adChris Wailes mWorkers.mLaunchCallback = nullptr; 257709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 258709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCompleteSignal.init(); 259709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 2600b575de8ed0b628d84d256f5846500b0385979bdTim Murray mWorkers.mRunningCount = mWorkers.mCount; 2610b575de8ed0b628d84d256f5846500b0385979bdTim Murray mWorkers.mLaunchCount = 0; 2620b575de8ed0b628d84d256f5846500b0385979bdTim Murray __sync_synchronize(); 263709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 264709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_attr_t threadAttr; 265709a0978ae141198018ca9769f8d96292a8928e6Jason Sams status = pthread_attr_init(&threadAttr); 266709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 267709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("Failed to init thread attribute."); 268709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return false; 269709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 270709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 271709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct=0; ct < mWorkers.mCount; ct++) { 272709a0978ae141198018ca9769f8d96292a8928e6Jason Sams status = pthread_create(&mWorkers.mThreadId[ct], &threadAttr, helperThreadProc, this); 273709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 274709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCount = ct; 275709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("Created fewer than expected number of RS threads."); 276709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 277709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 278709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 2790b575de8ed0b628d84d256f5846500b0385979bdTim Murray while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) { 280709a0978ae141198018ca9769f8d96292a8928e6Jason Sams usleep(100); 281709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 282709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 283709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_attr_destroy(&threadAttr); 284709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return true; 285709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 286709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 287709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 288709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::setPriority(int32_t priority) { 289709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct=0; ct < mWorkers.mCount; ct++) { 290709a0978ae141198018ca9769f8d96292a8928e6Jason Sams setpriority(PRIO_PROCESS, mWorkers.mNativeThreadId[ct], priority); 291709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 292709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 293709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 294709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::~RsdCpuReferenceImpl() { 295709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mExit = true; 29644bef6fba6244292b751387f3d6c31cca96c28adChris Wailes mWorkers.mLaunchData = nullptr; 29744bef6fba6244292b751387f3d6c31cca96c28adChris Wailes mWorkers.mLaunchCallback = nullptr; 2980b575de8ed0b628d84d256f5846500b0385979bdTim Murray mWorkers.mRunningCount = mWorkers.mCount; 2990b575de8ed0b628d84d256f5846500b0385979bdTim Murray __sync_synchronize(); 300709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 301709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchSignals[ct].set(); 302709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 303709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void *res; 304709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 305709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_join(mWorkers.mThreadId[ct], &res); 306709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 30719601aa29ec9dc950fc25d4473c2b006670f7b3aMiao Wang // b/23109602 30819601aa29ec9dc950fc25d4473c2b006670f7b3aMiao Wang // TODO: Refactor the implementation with threadpool to 30919601aa29ec9dc950fc25d4473c2b006670f7b3aMiao Wang // fix the race condition in the destuctor. 31019601aa29ec9dc950fc25d4473c2b006670f7b3aMiao Wang // rsAssert(__sync_fetch_and_or(&mWorkers.mRunningCount, 0) == 0); 31107ef704308b514272ed2f5c3e6a2f4c055550158Jens Gulin free(mWorkers.mThreadId); 31207ef704308b514272ed2f5c3e6a2f4c055550158Jens Gulin free(mWorkers.mNativeThreadId); 31307ef704308b514272ed2f5c3e6a2f4c055550158Jens Gulin delete[] mWorkers.mLaunchSignals; 314709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 315709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // Global structure cleanup. 316709a0978ae141198018ca9769f8d96292a8928e6Jason Sams lockMutex(); 317709a0978ae141198018ca9769f8d96292a8928e6Jason Sams --gThreadTLSKeyCount; 318709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!gThreadTLSKeyCount) { 319709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_key_delete(gThreadTLSKey); 320709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 321709a0978ae141198018ca9769f8d96292a8928e6Jason Sams unlockMutex(); 322709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 323709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 324709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 32514ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// Set up the appropriate input and output pointers to the kernel driver info structure. 32614ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// Inputs: 32714ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// mtls - The MTLaunchStruct holding information about the kernel launch 32814ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// fep - The forEach parameters (driver info structure) 32914ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala// x, y, z, lod, face, a1, a2, a3, a4 - The start offsets into each dimension 33014ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Walastatic inline void FepPtrSetup(const MTLaunchStructForEach *mtls, RsExpandKernelDriverInfo *fep, 331c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams uint32_t x, uint32_t y, 332c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams uint32_t z = 0, uint32_t lod = 0, 333c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams RsAllocationCubemapFace face = RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X, 334c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams uint32_t a1 = 0, uint32_t a2 = 0, uint32_t a3 = 0, uint32_t a4 = 0) { 335efc33a921195c313007d9471efcd00cc825b0055I-Jui (Ray) Sung // When rsForEach passes a null input allocation (as opposed to no input), 336efc33a921195c313007d9471efcd00cc825b0055I-Jui (Ray) Sung // fep->inLen can be 1 with mtls->ains[0] being null. 337efc33a921195c313007d9471efcd00cc825b0055I-Jui (Ray) Sung // This should only happen on old style kernels. 338c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams for (uint32_t i = 0; i < fep->inLen; i++) { 339efc33a921195c313007d9471efcd00cc825b0055I-Jui (Ray) Sung if (mtls->ains[i] == nullptr) { 340efc33a921195c313007d9471efcd00cc825b0055I-Jui (Ray) Sung rsAssert(fep->inLen == 1); 341efc33a921195c313007d9471efcd00cc825b0055I-Jui (Ray) Sung continue; 342efc33a921195c313007d9471efcd00cc825b0055I-Jui (Ray) Sung } 343c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams fep->inPtr[i] = (const uint8_t *)mtls->ains[i]->getPointerUnchecked(x, y, z, lod, face, a1, a2, a3, a4); 344709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 345c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams if (mtls->aout[0] != nullptr) { 346c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams fep->outPtr[0] = (uint8_t *)mtls->aout[0]->getPointerUnchecked(x, y, z, lod, face, a1, a2, a3, a4); 347c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams } 348709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 349709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 3506c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// Set up the appropriate input and output pointers to the kernel driver info structure. 3516c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// Inputs: 3526c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// mtls - The MTLaunchStruct holding information about the kernel launch 3536c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// redp - The reduce parameters (driver info structure) 3546c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// x, y, z - The start offsets into each dimension 355ae2ec3febedfc29376b9104413fb4042028f1265David Grossstatic inline void RedpPtrSetup(const MTLaunchStructReduce *mtls, RsExpandKernelDriverInfo *redp, 3566c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross uint32_t x, uint32_t y, uint32_t z) { 3576c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross for (uint32_t i = 0; i < redp->inLen; i++) { 3586c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross redp->inPtr[i] = (const uint8_t *)mtls->ains[i]->getPointerUnchecked(x, y, z); 3596c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross } 3606c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross} 3616c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross 362bf2111d3b3de310932099514f06924e48fa1d7b2Jason Samsstatic uint32_t sliceInt(uint32_t *p, uint32_t val, uint32_t start, uint32_t end) { 363bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams if (start >= end) { 364bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams *p = start; 365bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams return val; 366bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams } 367bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 368bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams uint32_t div = end - start; 369bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 370bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams uint32_t n = val / div; 371bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams *p = (val - (n * div)) + start; 372bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams return n; 373bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams} 374bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 3756c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Grossstatic bool SelectOuterSlice(const MTLaunchStructCommon *mtls, RsExpandKernelDriverInfo* info, uint32_t sliceNum) { 376bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams uint32_t r = sliceNum; 3776c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross r = sliceInt(&info->current.z, r, mtls->start.z, mtls->end.z); 3786c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross r = sliceInt(&info->current.lod, r, mtls->start.lod, mtls->end.lod); 3796c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross r = sliceInt(&info->current.face, r, mtls->start.face, mtls->end.face); 3806c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross r = sliceInt(&info->current.array[0], r, mtls->start.array[0], mtls->end.array[0]); 3816c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross r = sliceInt(&info->current.array[1], r, mtls->start.array[1], mtls->end.array[1]); 3826c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross r = sliceInt(&info->current.array[2], r, mtls->start.array[2], mtls->end.array[2]); 3836c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross r = sliceInt(&info->current.array[3], r, mtls->start.array[3], mtls->end.array[3]); 384bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams return r == 0; 385bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams} 386bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 387f9bd1f2afed17da60e4e10277ace806f0d6fead8David Grossstatic bool SelectZSlice(const MTLaunchStructCommon *mtls, RsExpandKernelDriverInfo* info, uint32_t sliceNum) { 388f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross return sliceInt(&info->current.z, sliceNum, mtls->start.z, mtls->end.z) == 0; 389f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross} 390bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 391f9bd1f2afed17da60e4e10277ace806f0d6fead8David Grossstatic void walk_general_foreach(void *usr, uint32_t idx) { 39214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala MTLaunchStructForEach *mtls = (MTLaunchStructForEach *)usr; 393bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams RsExpandKernelDriverInfo fep = mtls->fep; 394bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams fep.lid = idx; 39514ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala ForEachFunc_t fn = mtls->kernel; 396bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 397bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams while(1) { 398bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 399bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 40059b35f29cc576243a322bf88bc16063a9810da55Jason Sams if (!SelectOuterSlice(mtls, &fep, slice)) { 401bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams return; 402bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams } 403bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 40459b35f29cc576243a322bf88bc16063a9810da55Jason Sams for (fep.current.y = mtls->start.y; fep.current.y < mtls->end.y; 40559b35f29cc576243a322bf88bc16063a9810da55Jason Sams fep.current.y++) { 406bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 40759b35f29cc576243a322bf88bc16063a9810da55Jason Sams FepPtrSetup(mtls, &fep, mtls->start.x, 40859b35f29cc576243a322bf88bc16063a9810da55Jason Sams fep.current.y, fep.current.z, fep.current.lod, 40959b35f29cc576243a322bf88bc16063a9810da55Jason Sams (RsAllocationCubemapFace)fep.current.face, 41059b35f29cc576243a322bf88bc16063a9810da55Jason Sams fep.current.array[0], fep.current.array[1], 41159b35f29cc576243a322bf88bc16063a9810da55Jason Sams fep.current.array[2], fep.current.array[3]); 412bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 41359b35f29cc576243a322bf88bc16063a9810da55Jason Sams fn(&fep, mtls->start.x, mtls->end.x, mtls->fep.outStride[0]); 414bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams } 415bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams } 416bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams} 417c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams 418f9bd1f2afed17da60e4e10277ace806f0d6fead8David Grossstatic void walk_2d_foreach(void *usr, uint32_t idx) { 41914ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala MTLaunchStructForEach *mtls = (MTLaunchStructForEach *)usr; 420c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams RsExpandKernelDriverInfo fep = mtls->fep; 421c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams fep.lid = idx; 42214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala ForEachFunc_t fn = mtls->kernel; 423709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 424c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams while (1) { 425c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 426bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams uint32_t yStart = mtls->start.y + slice * mtls->mSliceSize; 427c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams uint32_t yEnd = yStart + mtls->mSliceSize; 428709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 429bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams yEnd = rsMin(yEnd, mtls->end.y); 430709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 431c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams if (yEnd <= yStart) { 432c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams return; 433c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams } 434709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 435c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams for (fep.current.y = yStart; fep.current.y < yEnd; fep.current.y++) { 436bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams FepPtrSetup(mtls, &fep, mtls->start.x, fep.current.y); 437709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 438b0abb140ac51b93d1a85aadaa63fe057f2d29850David Gross fn(&fep, mtls->start.x, mtls->end.x, fep.outStride[0]); 439709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 440c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams } 441f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes} 4424b2bea3dc20865f3a198797702e19912a6a2171cStephen Hines 44335dbc8c20814997a00e06d7798a5633c37706978David Grossstatic void walk_1d_foreach(void *usr, uint32_t idx) { 44414ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala MTLaunchStructForEach *mtls = (MTLaunchStructForEach *)usr; 445c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams RsExpandKernelDriverInfo fep = mtls->fep; 446c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams fep.lid = idx; 44714ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala ForEachFunc_t fn = mtls->kernel; 44880ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes 449c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams while (1) { 450c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 451bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams uint32_t xStart = mtls->start.x + slice * mtls->mSliceSize; 452c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams uint32_t xEnd = xStart + mtls->mSliceSize; 45380ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes 454bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams xEnd = rsMin(xEnd, mtls->end.x); 45580ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes 456c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams if (xEnd <= xStart) { 457c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams return; 458c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams } 45980ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes 460bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams FepPtrSetup(mtls, &fep, xStart, 0); 46180ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes 462b0abb140ac51b93d1a85aadaa63fe057f2d29850David Gross fn(&fep, xStart, xEnd, fep.outStride[0]); 463c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams } 464709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 465709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 46635dbc8c20814997a00e06d7798a5633c37706978David Gross// The function format_bytes() is an auxiliary function to assist in logging. 46735dbc8c20814997a00e06d7798a5633c37706978David Gross// 46835dbc8c20814997a00e06d7798a5633c37706978David Gross// Bytes are read from an input (inBuf) and written (as pairs of hex digits) 46935dbc8c20814997a00e06d7798a5633c37706978David Gross// to an output (outBuf). 47035dbc8c20814997a00e06d7798a5633c37706978David Gross// 47135dbc8c20814997a00e06d7798a5633c37706978David Gross// Output format: 47235dbc8c20814997a00e06d7798a5633c37706978David Gross// - starts with ": " 47335dbc8c20814997a00e06d7798a5633c37706978David Gross// - each input byte is translated to a pair of hex digits 47435dbc8c20814997a00e06d7798a5633c37706978David Gross// - bytes are separated by "." except that every fourth separator is "|" 47535dbc8c20814997a00e06d7798a5633c37706978David Gross// - if the input is sufficiently long, the output is truncated and terminated with "..." 47635dbc8c20814997a00e06d7798a5633c37706978David Gross// 47735dbc8c20814997a00e06d7798a5633c37706978David Gross// Arguments: 47835dbc8c20814997a00e06d7798a5633c37706978David Gross// - outBuf -- Pointer to buffer of type "FormatBuf" into which output is written 47935dbc8c20814997a00e06d7798a5633c37706978David Gross// - inBuf -- Pointer to bytes which are to be formatted into outBuf 48035dbc8c20814997a00e06d7798a5633c37706978David Gross// - inBytes -- Number of bytes in inBuf 48135dbc8c20814997a00e06d7798a5633c37706978David Gross// 48235dbc8c20814997a00e06d7798a5633c37706978David Gross// Constant: 48335dbc8c20814997a00e06d7798a5633c37706978David Gross// - kFormatInBytesMax -- Only min(kFormatInBytesMax, inBytes) bytes will be read 48435dbc8c20814997a00e06d7798a5633c37706978David Gross// from inBuf 48535dbc8c20814997a00e06d7798a5633c37706978David Gross// 48635dbc8c20814997a00e06d7798a5633c37706978David Gross// Return value: 48735dbc8c20814997a00e06d7798a5633c37706978David Gross// - pointer (const char *) to output (which is part of outBuf) 48835dbc8c20814997a00e06d7798a5633c37706978David Gross// 48935dbc8c20814997a00e06d7798a5633c37706978David Grossstatic const int kFormatInBytesMax = 16; 49035dbc8c20814997a00e06d7798a5633c37706978David Gross// ": " + 2 digits per byte + 1 separator between bytes + "..." + null 49135dbc8c20814997a00e06d7798a5633c37706978David Grosstypedef char FormatBuf[2 + kFormatInBytesMax*2 + (kFormatInBytesMax - 1) + 3 + 1]; 49235dbc8c20814997a00e06d7798a5633c37706978David Grossstatic const char *format_bytes(FormatBuf *outBuf, const uint8_t *inBuf, const int inBytes) { 493a25174fddaed975e63cddc8d05207d2943cdeee6Michael Butler strlcpy(*outBuf, ": ", sizeof(*outBuf)); 49435dbc8c20814997a00e06d7798a5633c37706978David Gross int pos = 2; 49535dbc8c20814997a00e06d7798a5633c37706978David Gross const int lim = std::min(kFormatInBytesMax, inBytes); 49635dbc8c20814997a00e06d7798a5633c37706978David Gross for (int i = 0; i < lim; ++i) { 49735dbc8c20814997a00e06d7798a5633c37706978David Gross if (i) { 49835dbc8c20814997a00e06d7798a5633c37706978David Gross sprintf(*outBuf + pos, (i % 4 ? "." : "|")); 49935dbc8c20814997a00e06d7798a5633c37706978David Gross ++pos; 50035dbc8c20814997a00e06d7798a5633c37706978David Gross } 50135dbc8c20814997a00e06d7798a5633c37706978David Gross sprintf(*outBuf + pos, "%02x", inBuf[i]); 50235dbc8c20814997a00e06d7798a5633c37706978David Gross pos += 2; 50335dbc8c20814997a00e06d7798a5633c37706978David Gross } 50435dbc8c20814997a00e06d7798a5633c37706978David Gross if (kFormatInBytesMax < inBytes) 505ca451c3280b6265a9b79273b4bf89e121a050cabMichael Butler strlcpy(*outBuf + pos, "...", sizeof(FormatBuf) - pos); 50635dbc8c20814997a00e06d7798a5633c37706978David Gross return *outBuf; 50735dbc8c20814997a00e06d7798a5633c37706978David Gross} 50835dbc8c20814997a00e06d7798a5633c37706978David Gross 509ae2ec3febedfc29376b9104413fb4042028f1265David Grossstatic void reduce_get_accumulator(uint8_t *&accumPtr, const MTLaunchStructReduce *mtls, 510ae2ec3febedfc29376b9104413fb4042028f1265David Gross const char *walkerName, uint32_t threadIdx) { 511f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross rsAssert(!accumPtr); 512f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 513f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross uint32_t accumIdx = (uint32_t)__sync_fetch_and_add(&mtls->accumCount, 1); 514f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (mtls->outFunc) { 515f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross accumPtr = mtls->accumAlloc + mtls->accumStride * accumIdx; 516f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } else { 517f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (accumIdx == 0) { 518f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross accumPtr = mtls->redp.outPtr[0]; 519f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } else { 520f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross accumPtr = mtls->accumAlloc + mtls->accumStride * (accumIdx - 1); 521f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 522f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 523ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 2, "%s(%p): idx = %u got accumCount %u and accumPtr %p", 524ae2ec3febedfc29376b9104413fb4042028f1265David Gross walkerName, mtls->accumFunc, threadIdx, accumIdx, accumPtr); 525f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross // initialize accumulator 526f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (mtls->initFunc) { 527f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross mtls->initFunc(accumPtr); 528f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } else { 529f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross memset(accumPtr, 0, mtls->accumSize); 530f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 531f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross} 532f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 533ae2ec3febedfc29376b9104413fb4042028f1265David Grossstatic void walk_1d_reduce(void *usr, uint32_t idx) { 534ae2ec3febedfc29376b9104413fb4042028f1265David Gross const MTLaunchStructReduce *mtls = (const MTLaunchStructReduce *)usr; 53535dbc8c20814997a00e06d7798a5633c37706978David Gross RsExpandKernelDriverInfo redp = mtls->redp; 53635dbc8c20814997a00e06d7798a5633c37706978David Gross 53735dbc8c20814997a00e06d7798a5633c37706978David Gross // find accumulator 53835dbc8c20814997a00e06d7798a5633c37706978David Gross uint8_t *&accumPtr = mtls->accumPtr[idx]; 53935dbc8c20814997a00e06d7798a5633c37706978David Gross if (!accumPtr) { 540ae2ec3febedfc29376b9104413fb4042028f1265David Gross reduce_get_accumulator(accumPtr, mtls, __func__, idx); 54135dbc8c20814997a00e06d7798a5633c37706978David Gross } 54235dbc8c20814997a00e06d7798a5633c37706978David Gross 54335dbc8c20814997a00e06d7798a5633c37706978David Gross // accumulate 544ae2ec3febedfc29376b9104413fb4042028f1265David Gross const ReduceAccumulatorFunc_t fn = mtls->accumFunc; 54535dbc8c20814997a00e06d7798a5633c37706978David Gross while (1) { 54635dbc8c20814997a00e06d7798a5633c37706978David Gross uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 54735dbc8c20814997a00e06d7798a5633c37706978David Gross uint32_t xStart = mtls->start.x + slice * mtls->mSliceSize; 54835dbc8c20814997a00e06d7798a5633c37706978David Gross uint32_t xEnd = xStart + mtls->mSliceSize; 54935dbc8c20814997a00e06d7798a5633c37706978David Gross 55035dbc8c20814997a00e06d7798a5633c37706978David Gross xEnd = rsMin(xEnd, mtls->end.x); 55135dbc8c20814997a00e06d7798a5633c37706978David Gross 55235dbc8c20814997a00e06d7798a5633c37706978David Gross if (xEnd <= xStart) { 55335dbc8c20814997a00e06d7798a5633c37706978David Gross return; 55435dbc8c20814997a00e06d7798a5633c37706978David Gross } 55535dbc8c20814997a00e06d7798a5633c37706978David Gross 55635dbc8c20814997a00e06d7798a5633c37706978David Gross RedpPtrSetup(mtls, &redp, xStart, 0, 0); 55735dbc8c20814997a00e06d7798a5633c37706978David Gross fn(&redp, xStart, xEnd, accumPtr); 55835dbc8c20814997a00e06d7798a5633c37706978David Gross 559f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross // Emit log line after slice has been run, so that we can include 560f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross // the results of the run on that line. 56135dbc8c20814997a00e06d7798a5633c37706978David Gross FormatBuf fmt; 562013ff53dc8d6c2464e1b5e1ea8c50b68f7b4aa73David Gross if (mtls->logReduce >= 3) { 56335dbc8c20814997a00e06d7798a5633c37706978David Gross format_bytes(&fmt, accumPtr, mtls->accumSize); 56435dbc8c20814997a00e06d7798a5633c37706978David Gross } else { 56535dbc8c20814997a00e06d7798a5633c37706978David Gross fmt[0] = 0; 56635dbc8c20814997a00e06d7798a5633c37706978David Gross } 567ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 2, "walk_1d_reduce(%p): idx = %u, x in [%u, %u)%s", 568ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->accumFunc, idx, xStart, xEnd, fmt); 56935dbc8c20814997a00e06d7798a5633c37706978David Gross } 57035dbc8c20814997a00e06d7798a5633c37706978David Gross} 57135dbc8c20814997a00e06d7798a5633c37706978David Gross 572ae2ec3febedfc29376b9104413fb4042028f1265David Grossstatic void walk_2d_reduce(void *usr, uint32_t idx) { 573ae2ec3febedfc29376b9104413fb4042028f1265David Gross const MTLaunchStructReduce *mtls = (const MTLaunchStructReduce *)usr; 574f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross RsExpandKernelDriverInfo redp = mtls->redp; 575f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 576f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross // find accumulator 577f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross uint8_t *&accumPtr = mtls->accumPtr[idx]; 578f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (!accumPtr) { 579ae2ec3febedfc29376b9104413fb4042028f1265David Gross reduce_get_accumulator(accumPtr, mtls, __func__, idx); 580f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 581f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 582f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross // accumulate 583ae2ec3febedfc29376b9104413fb4042028f1265David Gross const ReduceAccumulatorFunc_t fn = mtls->accumFunc; 584f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross while (1) { 585f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 586f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross uint32_t yStart = mtls->start.y + slice * mtls->mSliceSize; 587f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross uint32_t yEnd = yStart + mtls->mSliceSize; 588f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 589f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross yEnd = rsMin(yEnd, mtls->end.y); 590f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 591f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (yEnd <= yStart) { 592f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross return; 593f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 594f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 595f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross for (redp.current.y = yStart; redp.current.y < yEnd; redp.current.y++) { 596f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross RedpPtrSetup(mtls, &redp, mtls->start.x, redp.current.y, 0); 597f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross fn(&redp, mtls->start.x, mtls->end.x, accumPtr); 598f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 599f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 600f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross FormatBuf fmt; 601f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (mtls->logReduce >= 3) { 602f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross format_bytes(&fmt, accumPtr, mtls->accumSize); 603f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } else { 604f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross fmt[0] = 0; 605f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 606ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 2, "walk_2d_reduce(%p): idx = %u, y in [%u, %u)%s", 607ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->accumFunc, idx, yStart, yEnd, fmt); 608f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 609f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross} 610f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 611ae2ec3febedfc29376b9104413fb4042028f1265David Grossstatic void walk_3d_reduce(void *usr, uint32_t idx) { 612ae2ec3febedfc29376b9104413fb4042028f1265David Gross const MTLaunchStructReduce *mtls = (const MTLaunchStructReduce *)usr; 613f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross RsExpandKernelDriverInfo redp = mtls->redp; 614f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 615f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross // find accumulator 616f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross uint8_t *&accumPtr = mtls->accumPtr[idx]; 617f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (!accumPtr) { 618ae2ec3febedfc29376b9104413fb4042028f1265David Gross reduce_get_accumulator(accumPtr, mtls, __func__, idx); 619f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 620f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 621f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross // accumulate 622ae2ec3febedfc29376b9104413fb4042028f1265David Gross const ReduceAccumulatorFunc_t fn = mtls->accumFunc; 623f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross while (1) { 624f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 625f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 626f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (!SelectZSlice(mtls, &redp, slice)) { 627f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross return; 628f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 629f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 630f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross for (redp.current.y = mtls->start.y; redp.current.y < mtls->end.y; redp.current.y++) { 631f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross RedpPtrSetup(mtls, &redp, mtls->start.x, redp.current.y, redp.current.z); 632f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross fn(&redp, mtls->start.x, mtls->end.x, accumPtr); 633f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 634f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 635f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross FormatBuf fmt; 636f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (mtls->logReduce >= 3) { 637f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross format_bytes(&fmt, accumPtr, mtls->accumSize); 638f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } else { 639f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross fmt[0] = 0; 640f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 641ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 2, "walk_3d_reduce(%p): idx = %u, z = %u%s", 642ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->accumFunc, idx, redp.current.z, fmt); 643f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 644f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross} 645f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross 6466c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// Launch a general reduce-style kernel. 6476c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// Inputs: 6486c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// ains[0..inLen-1]: Array of allocations that contain the inputs 6496c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// aout: The allocation that will hold the output 6506c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross// mtls: Holds launch parameters 651ae2ec3febedfc29376b9104413fb4042028f1265David Grossvoid RsdCpuReferenceImpl::launchReduce(const Allocation ** ains, 652ae2ec3febedfc29376b9104413fb4042028f1265David Gross uint32_t inLen, 653ae2ec3febedfc29376b9104413fb4042028f1265David Gross Allocation * aout, 654ae2ec3febedfc29376b9104413fb4042028f1265David Gross MTLaunchStructReduce *mtls) { 655013ff53dc8d6c2464e1b5e1ea8c50b68f7b4aa73David Gross mtls->logReduce = mRSC->props.mLogReduce; 65635dbc8c20814997a00e06d7798a5633c37706978David Gross if ((mWorkers.mCount >= 1) && mtls->isThreadable && !mInKernel) { 657ae2ec3febedfc29376b9104413fb4042028f1265David Gross launchReduceParallel(ains, inLen, aout, mtls); 65835dbc8c20814997a00e06d7798a5633c37706978David Gross } else { 659ae2ec3febedfc29376b9104413fb4042028f1265David Gross launchReduceSerial(ains, inLen, aout, mtls); 66035dbc8c20814997a00e06d7798a5633c37706978David Gross } 66135dbc8c20814997a00e06d7798a5633c37706978David Gross} 66235dbc8c20814997a00e06d7798a5633c37706978David Gross 66335dbc8c20814997a00e06d7798a5633c37706978David Gross// Launch a general reduce-style kernel, single-threaded. 66435dbc8c20814997a00e06d7798a5633c37706978David Gross// Inputs: 66535dbc8c20814997a00e06d7798a5633c37706978David Gross// ains[0..inLen-1]: Array of allocations that contain the inputs 66635dbc8c20814997a00e06d7798a5633c37706978David Gross// aout: The allocation that will hold the output 66735dbc8c20814997a00e06d7798a5633c37706978David Gross// mtls: Holds launch parameters 668ae2ec3febedfc29376b9104413fb4042028f1265David Grossvoid RsdCpuReferenceImpl::launchReduceSerial(const Allocation ** ains, 669ae2ec3febedfc29376b9104413fb4042028f1265David Gross uint32_t inLen, 670ae2ec3febedfc29376b9104413fb4042028f1265David Gross Allocation * aout, 671ae2ec3febedfc29376b9104413fb4042028f1265David Gross MTLaunchStructReduce *mtls) { 672ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 1, "launchReduceSerial(%p): %u x %u x %u", mtls->accumFunc, 673ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->redp.dim.x, mtls->redp.dim.y, mtls->redp.dim.z); 67435dbc8c20814997a00e06d7798a5633c37706978David Gross 6756c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross // In the presence of outconverter, we allocate temporary memory for 6766c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross // the accumulator. 6776c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross // 6786c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross // In the absence of outconverter, we use the output allocation as the 6796c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross // accumulator. 6806c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross uint8_t *const accumPtr = (mtls->outFunc 6816c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross ? static_cast<uint8_t *>(malloc(mtls->accumSize)) 6826c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross : mtls->redp.outPtr[0]); 6836c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross 6846c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross // initialize 6856c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross if (mtls->initFunc) { 6866c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross mtls->initFunc(accumPtr); 6876c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross } else { 6886c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross memset(accumPtr, 0, mtls->accumSize); 6896c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross } 6906c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross 6916c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross // accumulate 692ae2ec3febedfc29376b9104413fb4042028f1265David Gross const ReduceAccumulatorFunc_t fn = mtls->accumFunc; 6936c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross uint32_t slice = 0; 6946c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross while (SelectOuterSlice(mtls, &mtls->redp, slice++)) { 6956c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross for (mtls->redp.current.y = mtls->start.y; 6966c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross mtls->redp.current.y < mtls->end.y; 6976c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross mtls->redp.current.y++) { 6986c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross RedpPtrSetup(mtls, &mtls->redp, mtls->start.x, mtls->redp.current.y, mtls->redp.current.z); 6996c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross fn(&mtls->redp, mtls->start.x, mtls->end.x, accumPtr); 7006c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross } 7016c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross } 7026c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross 7036c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross // outconvert 7046c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross if (mtls->outFunc) { 7056c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross mtls->outFunc(mtls->redp.outPtr[0], accumPtr); 7066c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross free(accumPtr); 7076c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross } 7086c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross} 7096c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross 71035dbc8c20814997a00e06d7798a5633c37706978David Gross// Launch a general reduce-style kernel, multi-threaded. 71135dbc8c20814997a00e06d7798a5633c37706978David Gross// Inputs: 71235dbc8c20814997a00e06d7798a5633c37706978David Gross// ains[0..inLen-1]: Array of allocations that contain the inputs 71335dbc8c20814997a00e06d7798a5633c37706978David Gross// aout: The allocation that will hold the output 71435dbc8c20814997a00e06d7798a5633c37706978David Gross// mtls: Holds launch parameters 715ae2ec3febedfc29376b9104413fb4042028f1265David Grossvoid RsdCpuReferenceImpl::launchReduceParallel(const Allocation ** ains, 716ae2ec3febedfc29376b9104413fb4042028f1265David Gross uint32_t inLen, 717ae2ec3febedfc29376b9104413fb4042028f1265David Gross Allocation * aout, 718ae2ec3febedfc29376b9104413fb4042028f1265David Gross MTLaunchStructReduce *mtls) { 719f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross // For now, we don't know how to go parallel in the absence of a combiner. 720f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (!mtls->combFunc) { 721ae2ec3febedfc29376b9104413fb4042028f1265David Gross launchReduceSerial(ains, inLen, aout, mtls); 72235dbc8c20814997a00e06d7798a5633c37706978David Gross return; 72335dbc8c20814997a00e06d7798a5633c37706978David Gross } 72435dbc8c20814997a00e06d7798a5633c37706978David Gross 72535dbc8c20814997a00e06d7798a5633c37706978David Gross // Number of threads = "main thread" + number of other (worker) threads 72635dbc8c20814997a00e06d7798a5633c37706978David Gross const uint32_t numThreads = mWorkers.mCount + 1; 72735dbc8c20814997a00e06d7798a5633c37706978David Gross 72835dbc8c20814997a00e06d7798a5633c37706978David Gross // In the absence of outconverter, we use the output allocation as 72935dbc8c20814997a00e06d7798a5633c37706978David Gross // an accumulator, and therefore need to allocate one fewer accumulator. 73035dbc8c20814997a00e06d7798a5633c37706978David Gross const uint32_t numAllocAccum = numThreads - (mtls->outFunc == nullptr); 73135dbc8c20814997a00e06d7798a5633c37706978David Gross 73235dbc8c20814997a00e06d7798a5633c37706978David Gross // If mDebugReduceSplitAccum, then we want each accumulator to start 73335dbc8c20814997a00e06d7798a5633c37706978David Gross // on a page boundary. (TODO: Would some unit smaller than a page 73435dbc8c20814997a00e06d7798a5633c37706978David Gross // be sufficient to avoid false sharing?) 73535dbc8c20814997a00e06d7798a5633c37706978David Gross if (mRSC->props.mDebugReduceSplitAccum) { 73635dbc8c20814997a00e06d7798a5633c37706978David Gross // Round up accumulator size to an integral number of pages 73735dbc8c20814997a00e06d7798a5633c37706978David Gross mtls->accumStride = 73835dbc8c20814997a00e06d7798a5633c37706978David Gross (unsigned(mtls->accumSize) + unsigned(mPageSize)-1) & 73935dbc8c20814997a00e06d7798a5633c37706978David Gross ~(unsigned(mPageSize)-1); 74035dbc8c20814997a00e06d7798a5633c37706978David Gross // Each accumulator gets its own page. Alternatively, if we just 74135dbc8c20814997a00e06d7798a5633c37706978David Gross // wanted to make sure no two accumulators are on the same page, 74235dbc8c20814997a00e06d7798a5633c37706978David Gross // we could instead do 74335dbc8c20814997a00e06d7798a5633c37706978David Gross // allocSize = mtls->accumStride * (numAllocation - 1) + mtls->accumSize 74435dbc8c20814997a00e06d7798a5633c37706978David Gross const size_t allocSize = mtls->accumStride * numAllocAccum; 74535dbc8c20814997a00e06d7798a5633c37706978David Gross mtls->accumAlloc = static_cast<uint8_t *>(memalign(mPageSize, allocSize)); 74635dbc8c20814997a00e06d7798a5633c37706978David Gross } else { 74735dbc8c20814997a00e06d7798a5633c37706978David Gross mtls->accumStride = mtls->accumSize; 74835dbc8c20814997a00e06d7798a5633c37706978David Gross mtls->accumAlloc = static_cast<uint8_t *>(malloc(mtls->accumStride * numAllocAccum)); 74935dbc8c20814997a00e06d7798a5633c37706978David Gross } 75035dbc8c20814997a00e06d7798a5633c37706978David Gross 75135dbc8c20814997a00e06d7798a5633c37706978David Gross const size_t accumPtrArrayBytes = sizeof(uint8_t *) * numThreads; 75235dbc8c20814997a00e06d7798a5633c37706978David Gross mtls->accumPtr = static_cast<uint8_t **>(malloc(accumPtrArrayBytes)); 75335dbc8c20814997a00e06d7798a5633c37706978David Gross memset(mtls->accumPtr, 0, accumPtrArrayBytes); 75435dbc8c20814997a00e06d7798a5633c37706978David Gross 75535dbc8c20814997a00e06d7798a5633c37706978David Gross mtls->accumCount = 0; 75635dbc8c20814997a00e06d7798a5633c37706978David Gross 75735dbc8c20814997a00e06d7798a5633c37706978David Gross rsAssert(!mInKernel); 75835dbc8c20814997a00e06d7798a5633c37706978David Gross mInKernel = true; 759ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 1, "launchReduceParallel(%p): %u x %u x %u, %u threads, accumAlloc = %p", 760ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->accumFunc, 761ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->redp.dim.x, mtls->redp.dim.y, mtls->redp.dim.z, 762ae2ec3febedfc29376b9104413fb4042028f1265David Gross numThreads, mtls->accumAlloc); 763f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross if (mtls->redp.dim.z > 1) { 764f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross mtls->mSliceSize = 1; 765ae2ec3febedfc29376b9104413fb4042028f1265David Gross launchThreads(walk_3d_reduce, mtls); 766f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } else if (mtls->redp.dim.y > 1) { 767f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross mtls->mSliceSize = rsMax(1U, mtls->redp.dim.y / (numThreads * 4)); 768ae2ec3febedfc29376b9104413fb4042028f1265David Gross launchThreads(walk_2d_reduce, mtls); 769f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } else { 770f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross mtls->mSliceSize = rsMax(1U, mtls->redp.dim.x / (numThreads * 4)); 771ae2ec3febedfc29376b9104413fb4042028f1265David Gross launchThreads(walk_1d_reduce, mtls); 772f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross } 77335dbc8c20814997a00e06d7798a5633c37706978David Gross mInKernel = false; 77435dbc8c20814997a00e06d7798a5633c37706978David Gross 77535dbc8c20814997a00e06d7798a5633c37706978David Gross // Combine accumulators and identify final accumulator 77635dbc8c20814997a00e06d7798a5633c37706978David Gross uint8_t *finalAccumPtr = (mtls->outFunc ? nullptr : mtls->redp.outPtr[0]); 77735dbc8c20814997a00e06d7798a5633c37706978David Gross // Loop over accumulators, combining into finalAccumPtr. If finalAccumPtr 77835dbc8c20814997a00e06d7798a5633c37706978David Gross // is null, then the first accumulator I find becomes finalAccumPtr. 77935dbc8c20814997a00e06d7798a5633c37706978David Gross for (unsigned idx = 0; idx < mtls->accumCount; ++idx) { 78035dbc8c20814997a00e06d7798a5633c37706978David Gross uint8_t *const thisAccumPtr = mtls->accumPtr[idx]; 78135dbc8c20814997a00e06d7798a5633c37706978David Gross if (finalAccumPtr) { 78235dbc8c20814997a00e06d7798a5633c37706978David Gross if (finalAccumPtr != thisAccumPtr) { 78335dbc8c20814997a00e06d7798a5633c37706978David Gross if (mtls->combFunc) { 784013ff53dc8d6c2464e1b5e1ea8c50b68f7b4aa73David Gross if (mtls->logReduce >= 3) { 78535dbc8c20814997a00e06d7798a5633c37706978David Gross FormatBuf fmt; 786ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 3, "launchReduceParallel(%p): accumulating into%s", 787ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->accumFunc, 788ae2ec3febedfc29376b9104413fb4042028f1265David Gross format_bytes(&fmt, finalAccumPtr, mtls->accumSize)); 789ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 3, "launchReduceParallel(%p): accumulator[%d]%s", 790ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->accumFunc, idx, 791ae2ec3febedfc29376b9104413fb4042028f1265David Gross format_bytes(&fmt, thisAccumPtr, mtls->accumSize)); 79235dbc8c20814997a00e06d7798a5633c37706978David Gross } 79335dbc8c20814997a00e06d7798a5633c37706978David Gross mtls->combFunc(finalAccumPtr, thisAccumPtr); 79435dbc8c20814997a00e06d7798a5633c37706978David Gross } else { 79535dbc8c20814997a00e06d7798a5633c37706978David Gross rsAssert(!"expected combiner"); 79635dbc8c20814997a00e06d7798a5633c37706978David Gross } 79735dbc8c20814997a00e06d7798a5633c37706978David Gross } 79835dbc8c20814997a00e06d7798a5633c37706978David Gross } else { 79935dbc8c20814997a00e06d7798a5633c37706978David Gross finalAccumPtr = thisAccumPtr; 80035dbc8c20814997a00e06d7798a5633c37706978David Gross } 80135dbc8c20814997a00e06d7798a5633c37706978David Gross } 80235dbc8c20814997a00e06d7798a5633c37706978David Gross rsAssert(finalAccumPtr != nullptr); 803013ff53dc8d6c2464e1b5e1ea8c50b68f7b4aa73David Gross if (mtls->logReduce >= 3) { 80435dbc8c20814997a00e06d7798a5633c37706978David Gross FormatBuf fmt; 805ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 3, "launchReduceParallel(%p): final accumulator%s", 806ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->accumFunc, format_bytes(&fmt, finalAccumPtr, mtls->accumSize)); 80735dbc8c20814997a00e06d7798a5633c37706978David Gross } 80835dbc8c20814997a00e06d7798a5633c37706978David Gross 80935dbc8c20814997a00e06d7798a5633c37706978David Gross // Outconvert 81035dbc8c20814997a00e06d7798a5633c37706978David Gross if (mtls->outFunc) { 81135dbc8c20814997a00e06d7798a5633c37706978David Gross mtls->outFunc(mtls->redp.outPtr[0], finalAccumPtr); 812013ff53dc8d6c2464e1b5e1ea8c50b68f7b4aa73David Gross if (mtls->logReduce >= 3) { 81335dbc8c20814997a00e06d7798a5633c37706978David Gross FormatBuf fmt; 814ae2ec3febedfc29376b9104413fb4042028f1265David Gross REDUCE_ALOGV(mtls, 3, "launchReduceParallel(%p): final outconverted result%s", 815ae2ec3febedfc29376b9104413fb4042028f1265David Gross mtls->accumFunc, 816ae2ec3febedfc29376b9104413fb4042028f1265David Gross format_bytes(&fmt, mtls->redp.outPtr[0], mtls->redp.outStride[0])); 81735dbc8c20814997a00e06d7798a5633c37706978David Gross } 81835dbc8c20814997a00e06d7798a5633c37706978David Gross } 81935dbc8c20814997a00e06d7798a5633c37706978David Gross 82035dbc8c20814997a00e06d7798a5633c37706978David Gross // Clean up 82135dbc8c20814997a00e06d7798a5633c37706978David Gross free(mtls->accumPtr); 82235dbc8c20814997a00e06d7798a5633c37706978David Gross free(mtls->accumAlloc); 82335dbc8c20814997a00e06d7798a5633c37706978David Gross} 82435dbc8c20814997a00e06d7798a5633c37706978David Gross 82535dbc8c20814997a00e06d7798a5633c37706978David Gross 82614ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Walavoid RsdCpuReferenceImpl::launchForEach(const Allocation ** ains, 827f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, 828f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes Allocation* aout, 829f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall* sc, 83014ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala MTLaunchStructForEach* mtls) { 8314b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 8324b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes //android::StopWatch kernel_time("kernel time"); 8334b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 834bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams bool outerDims = (mtls->start.z != mtls->end.z) || 835bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams (mtls->start.face != mtls->end.face) || 836bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams (mtls->start.lod != mtls->end.lod) || 837bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams (mtls->start.array[0] != mtls->end.array[0]) || 838bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams (mtls->start.array[1] != mtls->end.array[1]) || 839bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams (mtls->start.array[2] != mtls->end.array[2]) || 840bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams (mtls->start.array[3] != mtls->end.array[3]); 841bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams 84235dbc8c20814997a00e06d7798a5633c37706978David Gross if ((mWorkers.mCount >= 1) && mtls->isThreadable && !mInKernel) { 8434b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes const size_t targetByteChunk = 16 * 1024; 84435dbc8c20814997a00e06d7798a5633c37706978David Gross mInKernel = true; // NOTE: The guard immediately above ensures this was !mInKernel 845f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 846bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams if (outerDims) { 847bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams // No fancy logic for chunk size 848bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams mtls->mSliceSize = 1; 849f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross launchThreads(walk_general_foreach, mtls); 850bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams } else if (mtls->fep.dim.y > 1) { 851c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams uint32_t s1 = mtls->fep.dim.y / ((mWorkers.mCount + 1) * 4); 8524b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes uint32_t s2 = 0; 8534b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 8544b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes // This chooses our slice size to rate limit atomic ops to 8554b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes // one per 16k bytes of reads/writes. 856c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams if ((mtls->aout[0] != nullptr) && mtls->aout[0]->mHal.drvState.lod[0].stride) { 857c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams s2 = targetByteChunk / mtls->aout[0]->mHal.drvState.lod[0].stride; 858a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams } else if (mtls->ains[0]) { 859c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams s2 = targetByteChunk / mtls->ains[0]->mHal.drvState.lod[0].stride; 860a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams } else { 861a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams // Launch option only case 862a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams // Use s1 based only on the dimensions 863a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams s2 = s1; 8644b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } 8654b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes mtls->mSliceSize = rsMin(s1, s2); 8664b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 8674b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes if(mtls->mSliceSize < 1) { 8684b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes mtls->mSliceSize = 1; 8694b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } 8704b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 871f9bd1f2afed17da60e4e10277ace806f0d6fead8David Gross launchThreads(walk_2d_foreach, mtls); 8724b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } else { 873c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams uint32_t s1 = mtls->fep.dim.x / ((mWorkers.mCount + 1) * 4); 8744b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes uint32_t s2 = 0; 8754b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 8764b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes // This chooses our slice size to rate limit atomic ops to 8774b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes // one per 16k bytes of reads/writes. 878c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams if ((mtls->aout[0] != nullptr) && mtls->aout[0]->getType()->getElementSizeBytes()) { 879c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams s2 = targetByteChunk / mtls->aout[0]->getType()->getElementSizeBytes(); 880a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams } else if (mtls->ains[0]) { 881c0d68470b978a79ce024fde56f23ea3690603ccdJason Sams s2 = targetByteChunk / mtls->ains[0]->getType()->getElementSizeBytes(); 882a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams } else { 883a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams // Launch option only case 884a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams // Use s1 based only on the dimensions 885a9139c724f8312b3634d213599f2d6b3b2505db2Jason Sams s2 = s1; 8864b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } 8874b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes mtls->mSliceSize = rsMin(s1, s2); 8884b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 8894b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes if (mtls->mSliceSize < 1) { 8904b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes mtls->mSliceSize = 1; 8914b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } 8924b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 89335dbc8c20814997a00e06d7798a5633c37706978David Gross launchThreads(walk_1d_foreach, mtls); 8944b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } 89535dbc8c20814997a00e06d7798a5633c37706978David Gross mInKernel = false; 8964b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 8974b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } else { 89814ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala ForEachFunc_t fn = mtls->kernel; 899bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams uint32_t slice = 0; 90080ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes 9014b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 90259b35f29cc576243a322bf88bc16063a9810da55Jason Sams while(SelectOuterSlice(mtls, &mtls->fep, slice++)) { 903bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams for (mtls->fep.current.y = mtls->start.y; 904bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams mtls->fep.current.y < mtls->end.y; 905bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams mtls->fep.current.y++) { 9064b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 907bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams FepPtrSetup(mtls, &mtls->fep, mtls->start.x, 908bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams mtls->fep.current.y, mtls->fep.current.z, mtls->fep.current.lod, 909bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams (RsAllocationCubemapFace) mtls->fep.current.face, 910bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams mtls->fep.current.array[0], mtls->fep.current.array[1], 911bf2111d3b3de310932099514f06924e48fa1d7b2Jason Sams mtls->fep.current.array[2], mtls->fep.current.array[3]); 9124b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 913b0abb140ac51b93d1a85aadaa63fe057f2d29850David Gross fn(&mtls->fep, mtls->start.x, mtls->end.x, mtls->fep.outStride[0]); 9144b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } 9154b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } 9164b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes } 9174b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes} 9184b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 919709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuScriptImpl * RsdCpuReferenceImpl::setTLS(RsdCpuScriptImpl *sc) { 920709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("setTls %p", sc); 921709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 922709a0978ae141198018ca9769f8d96292a8928e6Jason Sams rsAssert(tls); 923709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl *old = tls->mImpl; 924709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mImpl = sc; 925709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mContext = mRSC; 926709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (sc) { 927709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mScript = sc->getScript(); 928709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 92944bef6fba6244292b751387f3d6c31cca96c28adChris Wailes tls->mScript = nullptr; 930709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 931709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return old; 932709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 933709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 934709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst RsdCpuReference::CpuSymbol * RsdCpuReferenceImpl::symLookup(const char *name) { 935709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return mSymLookupFn(mRSC, name); 936709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 937709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 938709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 939709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createScript(const ScriptC *s, 940709a0978ae141198018ca9769f8d96292a8928e6Jason Sams char const *resName, char const *cacheDir, 941709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint8_t const *bitcode, size_t bitcodeSize, 942709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t flags) { 943709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 944709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl *i = new RsdCpuScriptImpl(this, s); 945005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines if (!i->init(resName, cacheDir, bitcode, bitcodeSize, flags 946005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines , getBccPluginName() 947005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines )) { 948709a0978ae141198018ca9769f8d96292a8928e6Jason Sams delete i; 94944bef6fba6244292b751387f3d6c31cca96c28adChris Wailes return nullptr; 950709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 951709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return i; 952709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 953709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 9547c4b888f2147edf99690b6af75470774ff31c43bJason Samsextern RsdCpuScriptImpl * rsdIntrinsic_3DLUT(RsdCpuReferenceImpl *ctx, 9557c4b888f2147edf99690b6af75470774ff31c43bJason Sams const Script *s, const Element *e); 956c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve3x3(RsdCpuReferenceImpl *ctx, 957c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 958c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_ColorMatrix(RsdCpuReferenceImpl *ctx, 959c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 960c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_LUT(RsdCpuReferenceImpl *ctx, 961c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 962c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve5x5(RsdCpuReferenceImpl *ctx, 963c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 964c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blur(RsdCpuReferenceImpl *ctx, 965c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 966c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_YuvToRGB(RsdCpuReferenceImpl *ctx, 967c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 968c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blend(RsdCpuReferenceImpl *ctx, 969c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 9702282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Histogram(RsdCpuReferenceImpl *ctx, 9712282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams const Script *s, const Element *e); 97239ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Resize(RsdCpuReferenceImpl *ctx, 97339ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams const Script *s, const Element *e); 97464c682b65cd04ac83b51251b40dca14423df351aTim Murrayextern RsdCpuScriptImpl * rsdIntrinsic_BLAS(RsdCpuReferenceImpl *ctx, 97564c682b65cd04ac83b51251b40dca14423df351aTim Murray const Script *s, const Element *e); 976709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 977709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createIntrinsic(const Script *s, 978709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsScriptIntrinsicID iid, Element *e) { 979709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 98044bef6fba6244292b751387f3d6c31cca96c28adChris Wailes RsdCpuScriptImpl *i = nullptr; 981709a0978ae141198018ca9769f8d96292a8928e6Jason Sams switch (iid) { 9827c4b888f2147edf99690b6af75470774ff31c43bJason Sams case RS_SCRIPT_INTRINSIC_ID_3DLUT: 9837c4b888f2147edf99690b6af75470774ff31c43bJason Sams i = rsdIntrinsic_3DLUT(this, s, e); 9847c4b888f2147edf99690b6af75470774ff31c43bJason Sams break; 985709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_3x3: 986c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Convolve3x3(this, s, e); 987709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 988709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_COLOR_MATRIX: 989c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_ColorMatrix(this, s, e); 990709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 991709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_LUT: 992c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_LUT(this, s, e); 993709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 994709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_5x5: 995c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Convolve5x5(this, s, e); 996709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 997709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_BLUR: 998c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Blur(this, s, e); 999709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 1000709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_YUV_TO_RGB: 1001c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_YuvToRGB(this, s, e); 1002709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 1003709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_BLEND: 1004c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Blend(this, s, e); 1005709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 10062282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams case RS_SCRIPT_INTRINSIC_ID_HISTOGRAM: 10072282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams i = rsdIntrinsic_Histogram(this, s, e); 10082282e2816ac5f5de53f9bd4f3ecbdfd6d756d120Jason Sams break; 100939ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams case RS_SCRIPT_INTRINSIC_ID_RESIZE: 101039ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams i = rsdIntrinsic_Resize(this, s, e); 101139ab94aafb7f0916a7f6e345ee1fa0f5ff3bbacdJason Sams break; 101264c682b65cd04ac83b51251b40dca14423df351aTim Murray case RS_SCRIPT_INTRINSIC_ID_BLAS: 101364c682b65cd04ac83b51251b40dca14423df351aTim Murray i = rsdIntrinsic_BLAS(this, s, e); 101464c682b65cd04ac83b51251b40dca14423df351aTim Murray break; 1015709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1016709a0978ae141198018ca9769f8d96292a8928e6Jason Sams default: 1017709a0978ae141198018ca9769f8d96292a8928e6Jason Sams rsAssert(0); 1018709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 1019709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1020709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return i; 1021709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 1022709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 10231ffd86b448d78366190c540f98f8b6d641cdb6cfYang Nivoid* RsdCpuReferenceImpl::createScriptGroup(const ScriptGroupBase *sg) { 10241ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni switch (sg->getApiVersion()) { 10251ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni case ScriptGroupBase::SG_V1: { 10261ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni CpuScriptGroupImpl *sgi = new CpuScriptGroupImpl(this, sg); 10271ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni if (!sgi->init()) { 1028709a0978ae141198018ca9769f8d96292a8928e6Jason Sams delete sgi; 102944bef6fba6244292b751387f3d6c31cca96c28adChris Wailes return nullptr; 10301ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni } 10311ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni return sgi; 1032709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 10331ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni case ScriptGroupBase::SG_V2: { 10341ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni return new CpuScriptGroup2Impl(this, sg); 10351ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni } 10361ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni } 10371ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni return nullptr; 1038709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 1039462de21ac2e1773b99aedee012adb374e476ae36Chih-Hung Hsieh 1040462de21ac2e1773b99aedee012adb374e476ae36Chih-Hung Hsieh} // namespace renderscript 1041462de21ac2e1773b99aedee012adb374e476ae36Chih-Hung Hsieh} // namespace android 1042