rsCpuCore.cpp revision 64c682b65cd04ac83b51251b40dca14423df351a
1/* 2 * Copyright (C) 2012 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#include "rsCpuCore.h" 18#include "rsCpuScript.h" 19#include "rsCpuScriptGroup.h" 20#include "rsCpuScriptGroup2.h" 21 22#include <malloc.h> 23#include "rsContext.h" 24 25#include <sys/types.h> 26#include <sys/resource.h> 27#include <sched.h> 28#include <sys/syscall.h> 29#include <string.h> 30#include <unistd.h> 31 32#include <stdio.h> 33#include <stdlib.h> 34#include <fcntl.h> 35 36#if !defined(RS_SERVER) && !defined(RS_COMPATIBILITY_LIB) 37#include <cutils/properties.h> 38#include "utils/StopWatch.h" 39#endif 40 41#ifdef RS_SERVER 42// Android exposes gettid(), standard Linux does not 43static pid_t gettid() { 44 return syscall(SYS_gettid); 45} 46#endif 47 48using namespace android; 49using namespace android::renderscript; 50 51typedef void (*outer_foreach_t)( 52 const android::renderscript::RsExpandKernelParams *, 53 uint32_t x1, uint32_t x2, uint32_t outstep); 54 55 56static pthread_key_t gThreadTLSKey = 0; 57static uint32_t gThreadTLSKeyCount = 0; 58static pthread_mutex_t gInitMutex = PTHREAD_MUTEX_INITIALIZER; 59 60bool android::renderscript::gArchUseSIMD = false; 61 62RsdCpuReference::~RsdCpuReference() { 63} 64 65RsdCpuReference * RsdCpuReference::create(Context *rsc, uint32_t version_major, 66 uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn 67 , bcc::RSLinkRuntimeCallback pLinkRuntimeCallback, 68 RSSelectRTCallback pSelectRTCallback, 69 const char *pBccPluginName 70 ) { 71 72 RsdCpuReferenceImpl *cpu = new RsdCpuReferenceImpl(rsc); 73 if (!cpu) { 74 return nullptr; 75 } 76 if (!cpu->init(version_major, version_minor, lfn, slfn)) { 77 delete cpu; 78 return nullptr; 79 } 80 81 cpu->setLinkRuntimeCallback(pLinkRuntimeCallback); 82 cpu->setSelectRTCallback(pSelectRTCallback); 83 if (pBccPluginName) { 84 cpu->setBccPluginName(pBccPluginName); 85 } 86 87 return cpu; 88} 89 90 91Context * RsdCpuReference::getTlsContext() { 92 ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 93 return tls->mContext; 94} 95 96const Script * RsdCpuReference::getTlsScript() { 97 ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 98 return tls->mScript; 99} 100 101pthread_key_t RsdCpuReference::getThreadTLSKey(){ return gThreadTLSKey; } 102 103//////////////////////////////////////////////////////////// 104/// 105 106RsdCpuReferenceImpl::RsdCpuReferenceImpl(Context *rsc) { 107 mRSC = rsc; 108 109 version_major = 0; 110 version_minor = 0; 111 mInForEach = false; 112 memset(&mWorkers, 0, sizeof(mWorkers)); 113 memset(&mTlsStruct, 0, sizeof(mTlsStruct)); 114 mExit = false; 115 mLinkRuntimeCallback = nullptr; 116 mSelectRTCallback = nullptr; 117 mSetupCompilerCallback = nullptr; 118} 119 120 121void * RsdCpuReferenceImpl::helperThreadProc(void *vrsc) { 122 RsdCpuReferenceImpl *dc = (RsdCpuReferenceImpl *)vrsc; 123 124 uint32_t idx = __sync_fetch_and_add(&dc->mWorkers.mLaunchCount, 1); 125 126 //ALOGV("RS helperThread starting %p idx=%i", dc, idx); 127 128 dc->mWorkers.mLaunchSignals[idx].init(); 129 dc->mWorkers.mNativeThreadId[idx] = gettid(); 130 131 memset(&dc->mTlsStruct, 0, sizeof(dc->mTlsStruct)); 132 int status = pthread_setspecific(gThreadTLSKey, &dc->mTlsStruct); 133 if (status) { 134 ALOGE("pthread_setspecific %i", status); 135 } 136 137#if 0 138 typedef struct {uint64_t bits[1024 / 64]; } cpu_set_t; 139 cpu_set_t cpuset; 140 memset(&cpuset, 0, sizeof(cpuset)); 141 cpuset.bits[idx / 64] |= 1ULL << (idx % 64); 142 int ret = syscall(241, rsc->mWorkers.mNativeThreadId[idx], 143 sizeof(cpuset), &cpuset); 144 ALOGE("SETAFFINITY ret = %i %s", ret, EGLUtils::strerror(ret)); 145#endif 146 147 while (!dc->mExit) { 148 dc->mWorkers.mLaunchSignals[idx].wait(); 149 if (dc->mWorkers.mLaunchCallback) { 150 // idx +1 is used because the calling thread is always worker 0. 151 dc->mWorkers.mLaunchCallback(dc->mWorkers.mLaunchData, idx+1); 152 } 153 __sync_fetch_and_sub(&dc->mWorkers.mRunningCount, 1); 154 dc->mWorkers.mCompleteSignal.set(); 155 } 156 157 //ALOGV("RS helperThread exited %p idx=%i", dc, idx); 158 return nullptr; 159} 160 161void RsdCpuReferenceImpl::launchThreads(WorkerCallback_t cbk, void *data) { 162 mWorkers.mLaunchData = data; 163 mWorkers.mLaunchCallback = cbk; 164 165 // fast path for very small launches 166 MTLaunchStruct *mtls = (MTLaunchStruct *)data; 167 if (mtls && mtls->fep.dim.y <= 1 && mtls->end.x <= mtls->start.x + mtls->mSliceSize) { 168 if (mWorkers.mLaunchCallback) { 169 mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0); 170 } 171 return; 172 } 173 174 mWorkers.mRunningCount = mWorkers.mCount; 175 __sync_synchronize(); 176 177 for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 178 mWorkers.mLaunchSignals[ct].set(); 179 } 180 181 // We use the calling thread as one of the workers so we can start without 182 // the delay of the thread wakeup. 183 if (mWorkers.mLaunchCallback) { 184 mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0); 185 } 186 187 while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) { 188 mWorkers.mCompleteSignal.wait(); 189 } 190} 191 192 193void RsdCpuReferenceImpl::lockMutex() { 194 pthread_mutex_lock(&gInitMutex); 195} 196 197void RsdCpuReferenceImpl::unlockMutex() { 198 pthread_mutex_unlock(&gInitMutex); 199} 200 201static int 202read_file(const char* pathname, char* buffer, size_t buffsize) 203{ 204 int fd, len; 205 206 fd = open(pathname, O_RDONLY); 207 if (fd < 0) 208 return -1; 209 210 do { 211 len = read(fd, buffer, buffsize); 212 } while (len < 0 && errno == EINTR); 213 214 close(fd); 215 216 return len; 217} 218 219static void GetCpuInfo() { 220 char cpuinfo[4096]; 221 int cpuinfo_len; 222 223 cpuinfo_len = read_file("/proc/cpuinfo", cpuinfo, sizeof cpuinfo); 224 if (cpuinfo_len < 0) /* should not happen */ { 225 return; 226 } 227 228#if defined(ARCH_ARM_HAVE_VFP) || defined(ARCH_ARM_USE_INTRINSICS) 229 gArchUseSIMD = (!!strstr(cpuinfo, " neon")) || 230 (!!strstr(cpuinfo, " asimd")); 231#elif defined(ARCH_X86_HAVE_SSSE3) 232 gArchUseSIMD = !!strstr(cpuinfo, " ssse3"); 233#endif 234} 235 236bool RsdCpuReferenceImpl::init(uint32_t version_major, uint32_t version_minor, 237 sym_lookup_t lfn, script_lookup_t slfn) { 238 239 mSymLookupFn = lfn; 240 mScriptLookupFn = slfn; 241 242 lockMutex(); 243 if (!gThreadTLSKeyCount) { 244 int status = pthread_key_create(&gThreadTLSKey, nullptr); 245 if (status) { 246 ALOGE("Failed to init thread tls key."); 247 unlockMutex(); 248 return false; 249 } 250 } 251 gThreadTLSKeyCount++; 252 unlockMutex(); 253 254 mTlsStruct.mContext = mRSC; 255 mTlsStruct.mScript = nullptr; 256 int status = pthread_setspecific(gThreadTLSKey, &mTlsStruct); 257 if (status) { 258 ALOGE("pthread_setspecific %i", status); 259 } 260 261 GetCpuInfo(); 262 263 int cpu = sysconf(_SC_NPROCESSORS_CONF); 264 if(mRSC->props.mDebugMaxThreads) { 265 cpu = mRSC->props.mDebugMaxThreads; 266 } 267 if (cpu < 2) { 268 mWorkers.mCount = 0; 269 return true; 270 } 271 272 // Subtract one from the cpu count because we also use the command thread as a worker. 273 mWorkers.mCount = (uint32_t)(cpu - 1); 274 275 ALOGV("%p Launching thread(s), CPUs %i", mRSC, mWorkers.mCount + 1); 276 277 mWorkers.mThreadId = (pthread_t *) calloc(mWorkers.mCount, sizeof(pthread_t)); 278 mWorkers.mNativeThreadId = (pid_t *) calloc(mWorkers.mCount, sizeof(pid_t)); 279 mWorkers.mLaunchSignals = new Signal[mWorkers.mCount]; 280 mWorkers.mLaunchCallback = nullptr; 281 282 mWorkers.mCompleteSignal.init(); 283 284 mWorkers.mRunningCount = mWorkers.mCount; 285 mWorkers.mLaunchCount = 0; 286 __sync_synchronize(); 287 288 pthread_attr_t threadAttr; 289 status = pthread_attr_init(&threadAttr); 290 if (status) { 291 ALOGE("Failed to init thread attribute."); 292 return false; 293 } 294 295 for (uint32_t ct=0; ct < mWorkers.mCount; ct++) { 296 status = pthread_create(&mWorkers.mThreadId[ct], &threadAttr, helperThreadProc, this); 297 if (status) { 298 mWorkers.mCount = ct; 299 ALOGE("Created fewer than expected number of RS threads."); 300 break; 301 } 302 } 303 while (__sync_fetch_and_or(&mWorkers.mRunningCount, 0) != 0) { 304 usleep(100); 305 } 306 307 pthread_attr_destroy(&threadAttr); 308 return true; 309} 310 311 312void RsdCpuReferenceImpl::setPriority(int32_t priority) { 313 for (uint32_t ct=0; ct < mWorkers.mCount; ct++) { 314 setpriority(PRIO_PROCESS, mWorkers.mNativeThreadId[ct], priority); 315 } 316} 317 318RsdCpuReferenceImpl::~RsdCpuReferenceImpl() { 319 mExit = true; 320 mWorkers.mLaunchData = nullptr; 321 mWorkers.mLaunchCallback = nullptr; 322 mWorkers.mRunningCount = mWorkers.mCount; 323 __sync_synchronize(); 324 for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 325 mWorkers.mLaunchSignals[ct].set(); 326 } 327 void *res; 328 for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 329 pthread_join(mWorkers.mThreadId[ct], &res); 330 } 331 rsAssert(__sync_fetch_and_or(&mWorkers.mRunningCount, 0) == 0); 332 free(mWorkers.mThreadId); 333 free(mWorkers.mNativeThreadId); 334 delete[] mWorkers.mLaunchSignals; 335 336 // Global structure cleanup. 337 lockMutex(); 338 --gThreadTLSKeyCount; 339 if (!gThreadTLSKeyCount) { 340 pthread_key_delete(gThreadTLSKey); 341 } 342 unlockMutex(); 343 344} 345 346typedef void (*rs_t)(const void *, void *, const void *, uint32_t, uint32_t, uint32_t, uint32_t); 347typedef void (*walk_loop_t)(const MTLaunchStruct*, 348 RsExpandKernelDriverInfo, 349 outer_foreach_t); 350 351static void kparamSetup(RsExpandKernelParams *kparams, const RsExpandKernelDriverInfo *fep) { 352 //ALOGE("kp usr %p", fep->usr); 353 //ALOGE("kp slot %i", fep->slot); 354 //ALOGE("kp dim %i %i %i", fep->dim.x, fep->dim.y, fep->dim.z); 355 //ALOGE("kp lid %i", fep->lid); 356 //ALOGE("kp in[0] stide %i ptr %p", fep->inStride[0], fep->inPtr[0]); 357 //ALOGE("kp out[0] ptr %p", fep->outPtr[0]); 358 //ALOGE("kp loc %i %i %i", fep->current.x, fep->current.y, fep->current.z); 359 360 kparams->usr = fep->usr; 361 kparams->slot = fep->slot; 362 kparams->dimX = fep->dim.x; 363 kparams->dimY = fep->dim.y; 364 kparams->dimZ = fep->dim.z; 365 kparams->lid = fep->lid; 366 kparams->inEStrides = (uint32_t *)&fep->inStride[0]; 367 kparams->ins = (const void **)&fep->inPtr[0]; 368 kparams->out = fep->outPtr[0]; 369 kparams->y = fep->current.y; 370 kparams->z = fep->current.z; 371} 372 373static inline void FepPtrSetup(const MTLaunchStruct *mtls, RsExpandKernelDriverInfo *fep, 374 uint32_t x, uint32_t y, 375 uint32_t z = 0, uint32_t lod = 0, 376 RsAllocationCubemapFace face = RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X, 377 uint32_t a1 = 0, uint32_t a2 = 0, uint32_t a3 = 0, uint32_t a4 = 0) { 378 379 for (uint32_t i = 0; i < fep->inLen; i++) { 380 fep->inPtr[i] = (const uint8_t *)mtls->ains[i]->getPointerUnchecked(x, y, z, lod, face, a1, a2, a3, a4); 381 } 382 383 if (mtls->aout[0] != nullptr) { 384 fep->outPtr[0] = (uint8_t *)mtls->aout[0]->getPointerUnchecked(x, y, z, lod, face, a1, a2, a3, a4); 385 } 386} 387 388static uint32_t sliceInt(uint32_t *p, uint32_t val, uint32_t start, uint32_t end) { 389 if (start >= end) { 390 *p = start; 391 return val; 392 } 393 394 uint32_t div = end - start; 395 396 uint32_t n = val / div; 397 *p = (val - (n * div)) + start; 398 return n; 399} 400 401static bool SelectOuterSlice(MTLaunchStruct* mtls, uint32_t sliceNum) { 402 403 uint32_t r = sliceNum; 404 r = sliceInt(&mtls->fep.current.z, r, mtls->start.z, mtls->end.z); 405 r = sliceInt(&mtls->fep.current.lod, r, mtls->start.lod, mtls->end.lod); 406 r = sliceInt(&mtls->fep.current.face, r, mtls->start.face, mtls->end.face); 407 r = sliceInt(&mtls->fep.current.array[0], r, mtls->start.array[0], mtls->end.array[0]); 408 r = sliceInt(&mtls->fep.current.array[1], r, mtls->start.array[1], mtls->end.array[1]); 409 r = sliceInt(&mtls->fep.current.array[2], r, mtls->start.array[2], mtls->end.array[2]); 410 r = sliceInt(&mtls->fep.current.array[3], r, mtls->start.array[3], mtls->end.array[3]); 411 return r == 0; 412} 413 414 415static void walk_general(void *usr, uint32_t idx) { 416 MTLaunchStruct *mtls = (MTLaunchStruct *)usr; 417 RsExpandKernelDriverInfo fep = mtls->fep; 418 fep.lid = idx; 419 outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 420 421 422 while(1) { 423 uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 424 425 if (!SelectOuterSlice(mtls, slice)) { 426 return; 427 } 428 429 for (mtls->fep.current.y = mtls->start.y; 430 mtls->fep.current.y < mtls->end.y; 431 mtls->fep.current.y++) { 432 433 FepPtrSetup(mtls, &mtls->fep, mtls->start.x, 434 mtls->fep.current.y, mtls->fep.current.z, mtls->fep.current.lod, 435 (RsAllocationCubemapFace)mtls->fep.current.face, 436 mtls->fep.current.array[0], mtls->fep.current.array[1], 437 mtls->fep.current.array[2], mtls->fep.current.array[3]); 438 439 RsExpandKernelParams kparams; 440 kparamSetup(&kparams, &mtls->fep); 441 fn(&kparams, mtls->start.x, mtls->end.x, mtls->fep.outStride[0]); 442 } 443 } 444 445} 446 447static void walk_2d(void *usr, uint32_t idx) { 448 MTLaunchStruct *mtls = (MTLaunchStruct *)usr; 449 RsExpandKernelDriverInfo fep = mtls->fep; 450 fep.lid = idx; 451 outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 452 453 while (1) { 454 uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 455 uint32_t yStart = mtls->start.y + slice * mtls->mSliceSize; 456 uint32_t yEnd = yStart + mtls->mSliceSize; 457 458 yEnd = rsMin(yEnd, mtls->end.y); 459 460 if (yEnd <= yStart) { 461 return; 462 } 463 464 for (fep.current.y = yStart; fep.current.y < yEnd; fep.current.y++) { 465 FepPtrSetup(mtls, &fep, mtls->start.x, fep.current.y); 466 467 RsExpandKernelParams kparams; 468 kparamSetup(&kparams, &fep); 469 470 fn(&kparams, mtls->start.x, mtls->end.x, fep.outStride[0]); 471 } 472 } 473} 474 475static void walk_1d(void *usr, uint32_t idx) { 476 MTLaunchStruct *mtls = (MTLaunchStruct *)usr; 477 RsExpandKernelDriverInfo fep = mtls->fep; 478 fep.lid = idx; 479 outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 480 481 while (1) { 482 uint32_t slice = (uint32_t)__sync_fetch_and_add(&mtls->mSliceNum, 1); 483 uint32_t xStart = mtls->start.x + slice * mtls->mSliceSize; 484 uint32_t xEnd = xStart + mtls->mSliceSize; 485 486 xEnd = rsMin(xEnd, mtls->end.x); 487 488 if (xEnd <= xStart) { 489 return; 490 } 491 492 FepPtrSetup(mtls, &fep, xStart, 0); 493 494 RsExpandKernelParams kparams; 495 kparamSetup(&kparams, &fep); 496 497 fn(&kparams, xStart, xEnd, fep.outStride[0]); 498 } 499} 500 501void RsdCpuReferenceImpl::launchThreads(const Allocation ** ains, 502 uint32_t inLen, 503 Allocation* aout, 504 const RsScriptCall* sc, 505 MTLaunchStruct* mtls) { 506 507 //android::StopWatch kernel_time("kernel time"); 508 509 bool outerDims = (mtls->start.z != mtls->end.z) || 510 (mtls->start.face != mtls->end.face) || 511 (mtls->start.lod != mtls->end.lod) || 512 (mtls->start.array[0] != mtls->end.array[0]) || 513 (mtls->start.array[1] != mtls->end.array[1]) || 514 (mtls->start.array[2] != mtls->end.array[2]) || 515 (mtls->start.array[3] != mtls->end.array[3]); 516 517 if ((mWorkers.mCount >= 1) && mtls->isThreadable && !mInForEach) { 518 const size_t targetByteChunk = 16 * 1024; 519 mInForEach = true; 520 521 if (outerDims) { 522 // No fancy logic for chunk size 523 mtls->mSliceSize = 1; 524 launchThreads(walk_general, mtls); 525 } else if (mtls->fep.dim.y > 1) { 526 uint32_t s1 = mtls->fep.dim.y / ((mWorkers.mCount + 1) * 4); 527 uint32_t s2 = 0; 528 529 // This chooses our slice size to rate limit atomic ops to 530 // one per 16k bytes of reads/writes. 531 if ((mtls->aout[0] != nullptr) && mtls->aout[0]->mHal.drvState.lod[0].stride) { 532 s2 = targetByteChunk / mtls->aout[0]->mHal.drvState.lod[0].stride; 533 } else { 534 // We know that there is either an output or an input. 535 s2 = targetByteChunk / mtls->ains[0]->mHal.drvState.lod[0].stride; 536 } 537 mtls->mSliceSize = rsMin(s1, s2); 538 539 if(mtls->mSliceSize < 1) { 540 mtls->mSliceSize = 1; 541 } 542 543 launchThreads(walk_2d, mtls); 544 } else { 545 uint32_t s1 = mtls->fep.dim.x / ((mWorkers.mCount + 1) * 4); 546 uint32_t s2 = 0; 547 548 // This chooses our slice size to rate limit atomic ops to 549 // one per 16k bytes of reads/writes. 550 if ((mtls->aout[0] != nullptr) && mtls->aout[0]->getType()->getElementSizeBytes()) { 551 s2 = targetByteChunk / mtls->aout[0]->getType()->getElementSizeBytes(); 552 } else { 553 // We know that there is either an output or an input. 554 s2 = targetByteChunk / mtls->ains[0]->getType()->getElementSizeBytes(); 555 } 556 mtls->mSliceSize = rsMin(s1, s2); 557 558 if (mtls->mSliceSize < 1) { 559 mtls->mSliceSize = 1; 560 } 561 562 launchThreads(walk_1d, mtls); 563 } 564 mInForEach = false; 565 566 } else { 567 outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 568 uint32_t slice = 0; 569 570 571 while(SelectOuterSlice(mtls, slice++)) { 572 for (mtls->fep.current.y = mtls->start.y; 573 mtls->fep.current.y < mtls->end.y; 574 mtls->fep.current.y++) { 575 576 FepPtrSetup(mtls, &mtls->fep, mtls->start.x, 577 mtls->fep.current.y, mtls->fep.current.z, mtls->fep.current.lod, 578 (RsAllocationCubemapFace) mtls->fep.current.face, 579 mtls->fep.current.array[0], mtls->fep.current.array[1], 580 mtls->fep.current.array[2], mtls->fep.current.array[3]); 581 582 RsExpandKernelParams kparams; 583 kparamSetup(&kparams, &mtls->fep); 584 fn(&kparams, mtls->start.x, mtls->end.x, mtls->fep.outStride[0]); 585 } 586 } 587 } 588} 589 590RsdCpuScriptImpl * RsdCpuReferenceImpl::setTLS(RsdCpuScriptImpl *sc) { 591 //ALOGE("setTls %p", sc); 592 ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 593 rsAssert(tls); 594 RsdCpuScriptImpl *old = tls->mImpl; 595 tls->mImpl = sc; 596 tls->mContext = mRSC; 597 if (sc) { 598 tls->mScript = sc->getScript(); 599 } else { 600 tls->mScript = nullptr; 601 } 602 return old; 603} 604 605const RsdCpuReference::CpuSymbol * RsdCpuReferenceImpl::symLookup(const char *name) { 606 return mSymLookupFn(mRSC, name); 607} 608 609 610RsdCpuReference::CpuScript * RsdCpuReferenceImpl::createScript(const ScriptC *s, 611 char const *resName, char const *cacheDir, 612 uint8_t const *bitcode, size_t bitcodeSize, 613 uint32_t flags) { 614 615 RsdCpuScriptImpl *i = new RsdCpuScriptImpl(this, s); 616 if (!i->init(resName, cacheDir, bitcode, bitcodeSize, flags 617 , getBccPluginName() 618 )) { 619 delete i; 620 return nullptr; 621 } 622 return i; 623} 624 625extern RsdCpuScriptImpl * rsdIntrinsic_3DLUT(RsdCpuReferenceImpl *ctx, 626 const Script *s, const Element *e); 627extern RsdCpuScriptImpl * rsdIntrinsic_Convolve3x3(RsdCpuReferenceImpl *ctx, 628 const Script *s, const Element *e); 629extern RsdCpuScriptImpl * rsdIntrinsic_ColorMatrix(RsdCpuReferenceImpl *ctx, 630 const Script *s, const Element *e); 631extern RsdCpuScriptImpl * rsdIntrinsic_LUT(RsdCpuReferenceImpl *ctx, 632 const Script *s, const Element *e); 633extern RsdCpuScriptImpl * rsdIntrinsic_Convolve5x5(RsdCpuReferenceImpl *ctx, 634 const Script *s, const Element *e); 635extern RsdCpuScriptImpl * rsdIntrinsic_Blur(RsdCpuReferenceImpl *ctx, 636 const Script *s, const Element *e); 637extern RsdCpuScriptImpl * rsdIntrinsic_YuvToRGB(RsdCpuReferenceImpl *ctx, 638 const Script *s, const Element *e); 639extern RsdCpuScriptImpl * rsdIntrinsic_Blend(RsdCpuReferenceImpl *ctx, 640 const Script *s, const Element *e); 641extern RsdCpuScriptImpl * rsdIntrinsic_Histogram(RsdCpuReferenceImpl *ctx, 642 const Script *s, const Element *e); 643extern RsdCpuScriptImpl * rsdIntrinsic_Resize(RsdCpuReferenceImpl *ctx, 644 const Script *s, const Element *e); 645extern RsdCpuScriptImpl * rsdIntrinsic_BLAS(RsdCpuReferenceImpl *ctx, 646 const Script *s, const Element *e); 647 648RsdCpuReference::CpuScript * RsdCpuReferenceImpl::createIntrinsic(const Script *s, 649 RsScriptIntrinsicID iid, Element *e) { 650 651 RsdCpuScriptImpl *i = nullptr; 652 switch (iid) { 653 case RS_SCRIPT_INTRINSIC_ID_3DLUT: 654 i = rsdIntrinsic_3DLUT(this, s, e); 655 break; 656 case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_3x3: 657 i = rsdIntrinsic_Convolve3x3(this, s, e); 658 break; 659 case RS_SCRIPT_INTRINSIC_ID_COLOR_MATRIX: 660 i = rsdIntrinsic_ColorMatrix(this, s, e); 661 break; 662 case RS_SCRIPT_INTRINSIC_ID_LUT: 663 i = rsdIntrinsic_LUT(this, s, e); 664 break; 665 case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_5x5: 666 i = rsdIntrinsic_Convolve5x5(this, s, e); 667 break; 668 case RS_SCRIPT_INTRINSIC_ID_BLUR: 669 i = rsdIntrinsic_Blur(this, s, e); 670 break; 671 case RS_SCRIPT_INTRINSIC_ID_YUV_TO_RGB: 672 i = rsdIntrinsic_YuvToRGB(this, s, e); 673 break; 674 case RS_SCRIPT_INTRINSIC_ID_BLEND: 675 i = rsdIntrinsic_Blend(this, s, e); 676 break; 677 case RS_SCRIPT_INTRINSIC_ID_HISTOGRAM: 678 i = rsdIntrinsic_Histogram(this, s, e); 679 break; 680 case RS_SCRIPT_INTRINSIC_ID_RESIZE: 681 i = rsdIntrinsic_Resize(this, s, e); 682 break; 683#if !defined(RS_COMPATIBILITY_LIB) 684 case RS_SCRIPT_INTRINSIC_ID_BLAS: 685 i = rsdIntrinsic_BLAS(this, s, e); 686 break; 687#endif 688 689 default: 690 rsAssert(0); 691 } 692 693 return i; 694} 695 696void* RsdCpuReferenceImpl::createScriptGroup(const ScriptGroupBase *sg) { 697 switch (sg->getApiVersion()) { 698 case ScriptGroupBase::SG_V1: { 699 CpuScriptGroupImpl *sgi = new CpuScriptGroupImpl(this, sg); 700 if (!sgi->init()) { 701 delete sgi; 702 return nullptr; 703 } 704 return sgi; 705 } 706 case ScriptGroupBase::SG_V2: { 707 return new CpuScriptGroup2Impl(this, sg); 708 } 709 } 710 return nullptr; 711} 712