device_api 20 python_bindings/python/Func_gpu.h FuncOrStage &func_gpu_threads0(FuncOrStage &that, hh::VarOrRVar thread_x, hh::DeviceAPI device_api) { device_api 21 python_bindings/python/Func_gpu.h return that.gpu_threads(thread_x, device_api); device_api 25 python_bindings/python/Func_gpu.h FuncOrStage &func_gpu_threads1(FuncOrStage &that, hh::VarOrRVar thread_x, hh::VarOrRVar thread_y, hh::DeviceAPI device_api) { device_api 26 python_bindings/python/Func_gpu.h return that.gpu_threads(thread_x, thread_y, device_api); device_api 30 python_bindings/python/Func_gpu.h FuncOrStage &func_gpu_threads2(FuncOrStage &that, hh::VarOrRVar thread_x, hh::VarOrRVar thread_y, hh::VarOrRVar thread_z, hh::DeviceAPI device_api) { device_api 31 python_bindings/python/Func_gpu.h return that.gpu_threads(thread_x, thread_y, thread_z, device_api); device_api 35 python_bindings/python/Func_gpu.h FuncOrStage &func_gpu_blocks0(FuncOrStage &that, hh::VarOrRVar block_x, hh::DeviceAPI device_api) { device_api 36 python_bindings/python/Func_gpu.h return that.gpu_blocks(block_x, device_api); device_api 40 python_bindings/python/Func_gpu.h FuncOrStage &func_gpu_blocks1(FuncOrStage &that, hh::VarOrRVar block_x, hh::VarOrRVar block_y, hh::DeviceAPI device_api) { device_api 41 python_bindings/python/Func_gpu.h return that.gpu_blocks(block_x, block_y, device_api); device_api 45 python_bindings/python/Func_gpu.h FuncOrStage &func_gpu_blocks2(FuncOrStage &that, hh::VarOrRVar block_x, hh::VarOrRVar block_y, hh::VarOrRVar block_z, hh::DeviceAPI device_api) { device_api 46 python_bindings/python/Func_gpu.h return that.gpu_blocks(block_x, block_y, block_z, device_api); device_api 51 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 52 python_bindings/python/Func_gpu.h return that.gpu(block_x, thread_x, device_api); device_api 58 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 59 python_bindings/python/Func_gpu.h return that.gpu(block_x, block_y, thread_x, thread_y, device_api); device_api 65 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 66 python_bindings/python/Func_gpu.h return that.gpu(block_x, block_y, block_z, thread_x, thread_y, thread_z, device_api); device_api 71 python_bindings/python/Func_gpu.h hh::Var tx, int x_size, hh::DeviceAPI device_api) { device_api 72 python_bindings/python/Func_gpu.h return that.gpu_tile(x, bx, tx, x_size, hh::TailStrategy::Auto, device_api); device_api 76 python_bindings/python/Func_gpu.h hh::RVar tx, int x_size, hh::DeviceAPI device_api) { device_api 77 python_bindings/python/Func_gpu.h return that.gpu_tile(x, bx, tx, x_size, hh::TailStrategy::Auto, device_api); device_api 82 python_bindings/python/Func_gpu.h int x_size, hh::DeviceAPI device_api) { device_api 83 python_bindings/python/Func_gpu.h return that.gpu_tile(x, tx, x_size, hh::TailStrategy::Auto, device_api); device_api 91 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 92 python_bindings/python/Func_gpu.h return that.gpu_tile(x, y, bx, by, tx, ty, x_size, y_size, hh::TailStrategy::Auto, device_api); device_api 99 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 100 python_bindings/python/Func_gpu.h return that.gpu_tile(x, y, tx, ty, x_size, y_size, hh::TailStrategy::Auto, device_api); device_api 106 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 107 python_bindings/python/Func_gpu.h return that.gpu_tile(x, y, tx, ty, x_size, y_size, hh::TailStrategy::Auto, device_api); device_api 115 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 116 python_bindings/python/Func_gpu.h return that.gpu_tile(x, y, z, bx, by, bz, tx, ty, tz, x_size, y_size, z_size, hh::TailStrategy::Auto, device_api); device_api 123 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 124 python_bindings/python/Func_gpu.h return that.gpu_tile(x, y, z, tx, ty, tx, x_size, y_size, z_size, hh::TailStrategy::Auto, device_api); device_api 130 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 131 python_bindings/python/Func_gpu.h return that.gpu_tile(x, x_size, hh::TailStrategy::Auto, device_api); device_api 136 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 137 python_bindings/python/Func_gpu.h return that.gpu_tile(x, y, x_size, y_size, hh::TailStrategy::Auto, device_api); device_api 142 python_bindings/python/Func_gpu.h hh::DeviceAPI device_api) { device_api 143 python_bindings/python/Func_gpu.h return that.gpu_tile(x, y, z, x_size, y_size, z_size, hh::TailStrategy::Auto, device_api); device_api 978 src/BoundsInference.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, body); device_api 154 src/CanonicalizeGPUVars.cpp stmt = For::make(name, min, extent, op->for_type, op->device_api, body); device_api 223 src/CodeGen_GPU_Host.cpp internal_assert(loop->device_api != DeviceAPI::Default_GPU) device_api 255 src/CodeGen_GPU_Host.cpp if (loop->device_api == DeviceAPI::GLSL) { device_api 284 src/CodeGen_GPU_Host.cpp if (loop->device_api == DeviceAPI::GLSL) { device_api 313 src/CodeGen_GPU_Host.cpp CodeGen_GPU_Dev *gpu_codegen = cgdev[loop->device_api]; device_api 315 src/CodeGen_GPU_Host.cpp << "Loop is scheduled on device " << loop->device_api device_api 27 src/ExternalCode.h ExternalCode(Kind kind, const Target &llvm_target, DeviceAPI device_api, const std::vector<uint8_t> &code, const std::string &name) device_api 28 src/ExternalCode.h : kind(kind), llvm_target(llvm_target), device_code_kind(device_api), code(code), nametag(name) { device_api 69 src/ExternalCode.h static ExternalCode device_code_wrapper(DeviceAPI device_api, const std::vector<uint8_t> &code, const std::string &name) { device_api 70 src/ExternalCode.h return ExternalCode(DeviceCode, Target(), device_api, code, name); device_api 322 src/Func.cpp void Stage::set_dim_device_api(VarOrRVar var, DeviceAPI device_api) { device_api 328 src/Func.cpp dims[i].device_api = device_api; device_api 336 src/Func.cpp << " to set to device API " << static_cast<int>(device_api) device_api 1517 src/Func.cpp Stage &Stage::gpu_threads(VarOrRVar tx, DeviceAPI device_api) { device_api 1518 src/Func.cpp set_dim_device_api(tx, device_api); device_api 1523 src/Func.cpp Stage &Stage::gpu_threads(VarOrRVar tx, VarOrRVar ty, DeviceAPI device_api) { device_api 1524 src/Func.cpp set_dim_device_api(tx, device_api); device_api 1525 src/Func.cpp set_dim_device_api(ty, device_api); device_api 1531 src/Func.cpp Stage &Stage::gpu_threads(VarOrRVar tx, VarOrRVar ty, VarOrRVar tz, DeviceAPI device_api) { device_api 1532 src/Func.cpp set_dim_device_api(tx, device_api); device_api 1533 src/Func.cpp set_dim_device_api(ty, device_api); device_api 1534 src/Func.cpp set_dim_device_api(tz, device_api); device_api 1541 src/Func.cpp Stage &Stage::gpu_blocks(VarOrRVar bx, DeviceAPI device_api) { device_api 1542 src/Func.cpp set_dim_device_api(bx, device_api); device_api 1547 src/Func.cpp Stage &Stage::gpu_blocks(VarOrRVar bx, VarOrRVar by, DeviceAPI device_api) { device_api 1548 src/Func.cpp set_dim_device_api(bx, device_api); device_api 1549 src/Func.cpp set_dim_device_api(by, device_api); device_api 1555 src/Func.cpp Stage &Stage::gpu_blocks(VarOrRVar bx, VarOrRVar by, VarOrRVar bz, DeviceAPI device_api) { device_api 1556 src/Func.cpp set_dim_device_api(bx, device_api); device_api 1557 src/Func.cpp set_dim_device_api(by, device_api); device_api 1558 src/Func.cpp set_dim_device_api(bz, device_api); device_api 1565 src/Func.cpp Stage &Stage::gpu_single_thread(DeviceAPI device_api) { device_api 1568 src/Func.cpp set_dim_device_api(block, device_api); device_api 1573 src/Func.cpp Stage &Stage::gpu(VarOrRVar bx, VarOrRVar tx, DeviceAPI device_api) { device_api 1578 src/Func.cpp VarOrRVar tx, VarOrRVar ty, DeviceAPI device_api) { device_api 1584 src/Func.cpp DeviceAPI device_api) { device_api 1589 src/Func.cpp TailStrategy tail, DeviceAPI device_api) { device_api 1591 src/Func.cpp set_dim_device_api(bx, device_api); device_api 1592 src/Func.cpp set_dim_device_api(tx, device_api); device_api 1599 src/Func.cpp TailStrategy tail, DeviceAPI device_api) { device_api 1601 src/Func.cpp set_dim_device_api(bx, device_api); device_api 1602 src/Func.cpp set_dim_device_api(tx, device_api); device_api 1609 src/Func.cpp TailStrategy tail, DeviceAPI device_api) { device_api 1611 src/Func.cpp set_dim_device_api(x, device_api); device_api 1612 src/Func.cpp set_dim_device_api(tx, device_api); device_api 1624 src/Func.cpp DeviceAPI device_api) { device_api 1626 src/Func.cpp set_dim_device_api(bx, device_api); device_api 1627 src/Func.cpp set_dim_device_api(by, device_api); device_api 1628 src/Func.cpp set_dim_device_api(tx, device_api); device_api 1629 src/Func.cpp set_dim_device_api(ty, device_api); device_api 1641 src/Func.cpp DeviceAPI device_api) { device_api 1642 src/Func.cpp return gpu_tile(x, y, x, y, tx, ty, x_size, y_size, tail, device_api); device_api 1649 src/Func.cpp DeviceAPI device_api) { device_api 1650 src/Func.cpp return gpu_tile(x, y, x, y, tx, ty, x_size, y_size, tail, device_api); device_api 1658 src/Func.cpp DeviceAPI device_api) { device_api 1670 src/Func.cpp set_dim_device_api(bx, device_api); device_api 1671 src/Func.cpp set_dim_device_api(by, device_api); device_api 1672 src/Func.cpp set_dim_device_api(bz, device_api); device_api 1673 src/Func.cpp set_dim_device_api(tx, device_api); device_api 1674 src/Func.cpp set_dim_device_api(ty, device_api); device_api 1675 src/Func.cpp set_dim_device_api(tz, device_api); device_api 1690 src/Func.cpp DeviceAPI device_api) { device_api 1691 src/Func.cpp return gpu_tile(x, y, z, x, y, z, tx, ty, tz, x_size, y_size, z_size, tail, device_api); device_api 1694 src/Func.cpp Stage &Stage::gpu_tile(VarOrRVar x, Expr x_size, TailStrategy tail, DeviceAPI device_api) { device_api 1698 src/Func.cpp set_dim_device_api(bx, device_api); device_api 1699 src/Func.cpp set_dim_device_api(tx, device_api); device_api 1709 src/Func.cpp DeviceAPI device_api) { device_api 1714 src/Func.cpp return gpu_tile(x, y, bx, by, tx, ty, x_size, y_size, tail, device_api); device_api 1720 src/Func.cpp DeviceAPI device_api) { device_api 1727 src/Func.cpp return gpu_tile(x, y, z, bx, by, bz, tx, ty, tz, x_size, y_size, z_size, tail, device_api); device_api 2024 src/Func.cpp Func &Func::gpu_threads(VarOrRVar tx, DeviceAPI device_api) { device_api 2026 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_threads(tx, device_api); device_api 2030 src/Func.cpp Func &Func::gpu_threads(VarOrRVar tx, VarOrRVar ty, DeviceAPI device_api) { device_api 2032 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_threads(tx, ty, device_api); device_api 2036 src/Func.cpp Func &Func::gpu_threads(VarOrRVar tx, VarOrRVar ty, VarOrRVar tz, DeviceAPI device_api) { device_api 2038 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_threads(tx, ty, tz, device_api); device_api 2042 src/Func.cpp Func &Func::gpu_blocks(VarOrRVar bx, DeviceAPI device_api) { device_api 2044 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_blocks(bx, device_api); device_api 2048 src/Func.cpp Func &Func::gpu_blocks(VarOrRVar bx, VarOrRVar by, DeviceAPI device_api) { device_api 2050 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_blocks(bx, by, device_api); device_api 2054 src/Func.cpp Func &Func::gpu_blocks(VarOrRVar bx, VarOrRVar by, VarOrRVar bz, DeviceAPI device_api) { device_api 2056 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_blocks(bx, by, bz, device_api); device_api 2060 src/Func.cpp Func &Func::gpu_single_thread(DeviceAPI device_api) { device_api 2062 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_single_thread(device_api); device_api 2066 src/Func.cpp Func &Func::gpu(VarOrRVar bx, VarOrRVar tx, DeviceAPI device_api) { device_api 2068 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu(bx, tx, device_api); device_api 2072 src/Func.cpp Func &Func::gpu(VarOrRVar bx, VarOrRVar by, VarOrRVar tx, VarOrRVar ty, DeviceAPI device_api) { device_api 2074 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu(bx, by, tx, ty, device_api); device_api 2078 src/Func.cpp Func &Func::gpu(VarOrRVar bx, VarOrRVar by, VarOrRVar bz, VarOrRVar tx, VarOrRVar ty, VarOrRVar tz, DeviceAPI device_api) { device_api 2080 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu(bx, by, bz, tx, ty, tz, device_api); device_api 2084 src/Func.cpp Func &Func::gpu_tile(VarOrRVar x, VarOrRVar bx, Var tx, Expr x_size, TailStrategy tail, DeviceAPI device_api) { device_api 2086 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_tile(x, bx, tx, x_size, tail, device_api); device_api 2090 src/Func.cpp Func &Func::gpu_tile(VarOrRVar x, VarOrRVar bx, RVar tx, Expr x_size, TailStrategy tail, DeviceAPI device_api) { device_api 2092 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_tile(x, bx, tx, x_size, tail, device_api); device_api 2096 src/Func.cpp Func &Func::gpu_tile(VarOrRVar x, VarOrRVar tx, Expr x_size, TailStrategy tail, DeviceAPI device_api) { device_api 2098 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_tile(x, tx, x_size, tail, device_api); device_api 2107 src/Func.cpp DeviceAPI device_api) { device_api 2110 src/Func.cpp .gpu_tile(x, y, bx, by, tx, ty, x_size, y_size, tail, device_api); device_api 2118 src/Func.cpp DeviceAPI device_api) { device_api 2121 src/Func.cpp .gpu_tile(x, y, tx, ty, x_size, y_size, tail, device_api); device_api 2129 src/Func.cpp DeviceAPI device_api) { device_api 2132 src/Func.cpp .gpu_tile(x, y, tx, ty, x_size, y_size, tail, device_api); device_api 2141 src/Func.cpp DeviceAPI device_api) { device_api 2144 src/Func.cpp .gpu_tile(x, y, z, bx, by, bz, tx, ty, tz, x_size, y_size, z_size, tail, device_api); device_api 2152 src/Func.cpp DeviceAPI device_api) { device_api 2155 src/Func.cpp .gpu_tile(x, y, z, tx, ty, tz, x_size, y_size, z_size, tail, device_api); device_api 2159 src/Func.cpp Func &Func::gpu_tile(VarOrRVar x, Expr x_size, TailStrategy tail, DeviceAPI device_api) { device_api 2161 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_tile(x, x_size, tail, device_api); device_api 2168 src/Func.cpp DeviceAPI device_api) { device_api 2170 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_tile(x, y, x_size, y_size, tail, device_api); device_api 2177 src/Func.cpp DeviceAPI device_api) { device_api 2179 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_tile(x, y, z, x_size, y_size, z_size, tail, device_api); device_api 2183 src/Func.cpp Func &Func::shader(Var x, Var y, Var c, DeviceAPI device_api) { device_api 2192 src/Func.cpp Stage(func.definition(), name(), args(), func.schedule().storage_dims()).gpu_blocks(x, y, device_api); device_api 59 src/Func.h void set_dim_device_api(VarOrRVar var, DeviceAPI device_api); device_api 206 src/Func.h EXPORT Stage &gpu_threads(VarOrRVar thread_x, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 207 src/Func.h EXPORT Stage &gpu_threads(VarOrRVar thread_x, VarOrRVar thread_y, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 208 src/Func.h EXPORT Stage &gpu_threads(VarOrRVar thread_x, VarOrRVar thread_y, VarOrRVar thread_z, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 209 src/Func.h EXPORT Stage &gpu_single_thread(DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 211 src/Func.h EXPORT Stage &gpu_blocks(VarOrRVar block_x, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 212 src/Func.h EXPORT Stage &gpu_blocks(VarOrRVar block_x, VarOrRVar block_y, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 213 src/Func.h EXPORT Stage &gpu_blocks(VarOrRVar block_x, VarOrRVar block_y, VarOrRVar block_z, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 215 src/Func.h EXPORT Stage &gpu(VarOrRVar block_x, VarOrRVar thread_x, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 218 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 221 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 229 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 232 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 236 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 242 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 248 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 253 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 260 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 265 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 273 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 278 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 283 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1439 src/Func.h EXPORT Func &gpu_threads(VarOrRVar thread_x, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1440 src/Func.h EXPORT Func &gpu_threads(VarOrRVar thread_x, VarOrRVar thread_y, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1441 src/Func.h EXPORT Func &gpu_threads(VarOrRVar thread_x, VarOrRVar thread_y, VarOrRVar thread_z, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1448 src/Func.h EXPORT Func &gpu_single_thread(DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1455 src/Func.h EXPORT Func &gpu_blocks(VarOrRVar block_x, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1456 src/Func.h EXPORT Func &gpu_blocks(VarOrRVar block_x, VarOrRVar block_y, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1457 src/Func.h EXPORT Func &gpu_blocks(VarOrRVar block_x, VarOrRVar block_y, VarOrRVar block_z, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1466 src/Func.h EXPORT Func &gpu(VarOrRVar block_x, VarOrRVar thread_x, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1468 src/Func.h VarOrRVar thread_x, VarOrRVar thread_y, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1470 src/Func.h VarOrRVar thread_x, VarOrRVar thread_y, VarOrRVar thread_z, DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1480 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1483 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1487 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1493 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1499 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1504 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1511 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1516 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1521 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1525 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1530 src/Func.h DeviceAPI device_api = DeviceAPI::Default_GPU); device_api 1538 src/Func.h EXPORT Func &shader(Var x, Var y, Var c, DeviceAPI device_api); device_api 131 src/FuseGPUThreadLoops.cpp const DeviceAPI device_api; device_api 147 src/FuseGPUThreadLoops.cpp s = For::make("." + name, 0, 1, ForType::GPUThread, device_api, s); device_api 183 src/FuseGPUThreadLoops.cpp NormalizeDimensionality(const ExtractBlockSize &e, DeviceAPI device_api) device_api 184 src/FuseGPUThreadLoops.cpp : block_size(e), device_api(device_api), depth(0), max_depth(0) {} device_api 271 src/FuseGPUThreadLoops.cpp const DeviceAPI device_api; device_api 306 src/FuseGPUThreadLoops.cpp stmt = For::make(op->name, mutate(op->min), mutate(op->extent), op->for_type, op->device_api, body); device_api 360 src/FuseGPUThreadLoops.cpp if (device_api == DeviceAPI::OpenGLCompute) { device_api 380 src/FuseGPUThreadLoops.cpp if (device_api == DeviceAPI::OpenGLCompute) { device_api 523 src/FuseGPUThreadLoops.cpp if (device_api == DeviceAPI::OpenGLCompute) { device_api 596 src/FuseGPUThreadLoops.cpp NormalizeDimensionality n(block_size, op->device_api); device_api 613 src/FuseGPUThreadLoops.cpp body = For::make("." + thread_names[i], 0, block_size.extent(i), ForType::GPUThread, op->device_api, body); device_api 618 src/FuseGPUThreadLoops.cpp body = For::make(".__thread_id_x", 0, 1, ForType::GPUThread, op->device_api, body); device_api 630 src/FuseGPUThreadLoops.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, body); device_api 649 src/FuseGPUThreadLoops.cpp if (op->device_api == DeviceAPI::GLSL) { device_api 668 src/FuseGPUThreadLoops.cpp ExtractSharedAllocations shared_mem(op->device_api); device_api 688 src/FuseGPUThreadLoops.cpp in_non_glsl_gpu = (in_non_glsl_gpu && op->device_api == DeviceAPI::None) || device_api 689 src/FuseGPUThreadLoops.cpp (op->device_api == DeviceAPI::CUDA) || (op->device_api == DeviceAPI::OpenCL) || device_api 690 src/FuseGPUThreadLoops.cpp (op->device_api == DeviceAPI::Metal); device_api 698 src/FuseGPUThreadLoops.cpp stmt = For::make(op->name, 0, op->extent, op->for_type, op->device_api, body); device_api 650 src/HexagonOffload.cpp if (loop->device_api != DeviceAPI::Hexagon) { device_api 328 src/IR.cpp Stmt For::make(const std::string &name, const Expr &min, const Expr &extent, ForType for_type, DeviceAPI device_api, const Stmt &body) { device_api 340 src/IR.cpp node->device_api = device_api; device_api 652 src/IR.h DeviceAPI device_api; device_api 655 src/IR.h EXPORT static Stmt make(const std::string &name, const Expr &min, const Expr &extent, ForType for_type, DeviceAPI device_api, const Stmt &body); device_api 213 src/IRMutator.cpp stmt = For::make(op->name, min, extent, op->for_type, op->device_api, body); device_api 564 src/IRPrinter.cpp stream << op->for_type << op->device_api << " (" << op->name << ", "; device_api 28 src/InjectHostDevBufferCopies.cpp DeviceAPI fixup_device_api(DeviceAPI device_api, const Target &target) { device_api 29 src/InjectHostDevBufferCopies.cpp if (device_api == DeviceAPI::Default_GPU) { device_api 42 src/InjectHostDevBufferCopies.cpp return device_api; device_api 45 src/InjectHostDevBufferCopies.cpp bool different_device_api(DeviceAPI device_api, DeviceAPI stmt_api, const Target &target) { device_api 46 src/InjectHostDevBufferCopies.cpp device_api = fixup_device_api(device_api, target); device_api 48 src/InjectHostDevBufferCopies.cpp return (stmt_api != DeviceAPI::None) && (device_api != stmt_api); device_api 59 src/InjectHostDevBufferCopies.cpp DeviceAPI device_api; device_api 64 src/InjectHostDevBufferCopies.cpp debug(2) << "Buffers to track: Setting Allocate for loop " << op->name << " to " << static_cast<int>(device_api) << "\n"; device_api 66 src/InjectHostDevBufferCopies.cpp auto it = internal.insert({ op->name, device_api }); device_api 72 src/InjectHostDevBufferCopies.cpp if (different_device_api(device_api, op->device_api, target)) { device_api 73 src/InjectHostDevBufferCopies.cpp debug(2) << "Buffers to track: switching from " << static_cast<int>(device_api) << device_api 74 src/InjectHostDevBufferCopies.cpp " to " << static_cast<int>(op->device_api) << " for loop " << op->name << "\n"; device_api 75 src/InjectHostDevBufferCopies.cpp DeviceAPI old_device_api = device_api; device_api 76 src/InjectHostDevBufferCopies.cpp device_api = fixup_device_api(op->device_api, target); device_api 77 src/InjectHostDevBufferCopies.cpp if (device_api == DeviceAPI::None) { device_api 78 src/InjectHostDevBufferCopies.cpp device_api = old_device_api; device_api 80 src/InjectHostDevBufferCopies.cpp internal_assert(device_api != DeviceAPI::None); device_api 82 src/InjectHostDevBufferCopies.cpp device_api = old_device_api; device_api 103 src/InjectHostDevBufferCopies.cpp different_device_api(device_api, internal[op->name], target)) { device_api 111 src/InjectHostDevBufferCopies.cpp different_device_api(device_api, internal[op->name], target)) { device_api 188 src/InjectHostDevBufferCopies.cpp DeviceAPI device_api; device_api 190 src/InjectHostDevBufferCopies.cpp Expr make_device_interface_call(DeviceAPI device_api) { device_api 192 src/InjectHostDevBufferCopies.cpp switch (device_api) { device_api 212 src/InjectHostDevBufferCopies.cpp internal_error << "Bad DeviceAPI " << static_cast<int>(device_api) << "\n"; device_api 257 src/InjectHostDevBufferCopies.cpp if (device_api != DeviceAPI::Host) { device_api 416 src/InjectHostDevBufferCopies.cpp debug(4) << "Device " << static_cast<int>(device_api) << " writes buffer " << op->name << "\n"; device_api 417 src/InjectHostDevBufferCopies.cpp state[op->name].devices_writing.insert(device_api); device_api 418 src/InjectHostDevBufferCopies.cpp state[op->name].devices_touched.insert(device_api); device_api 428 src/InjectHostDevBufferCopies.cpp debug(4) << "Device " << static_cast<int>(device_api) << " reads buffer " << op->name << "\n"; device_api 429 src/InjectHostDevBufferCopies.cpp state[op->name].devices_reading.insert(device_api); device_api 430 src/InjectHostDevBufferCopies.cpp state[op->name].devices_touched.insert(device_api); device_api 436 src/InjectHostDevBufferCopies.cpp internal_assert(device_api == DeviceAPI::GLSL); device_api 442 src/InjectHostDevBufferCopies.cpp state[buf_name].devices_reading.insert(device_api); device_api 443 src/InjectHostDevBufferCopies.cpp state[buf_name].devices_touched.insert(device_api); device_api 447 src/InjectHostDevBufferCopies.cpp internal_assert(device_api == DeviceAPI::GLSL); device_api 453 src/InjectHostDevBufferCopies.cpp state[buf_name].devices_writing.insert(device_api); device_api 454 src/InjectHostDevBufferCopies.cpp state[buf_name].devices_touched.insert(device_api); device_api 462 src/InjectHostDevBufferCopies.cpp if (device_api != DeviceAPI::Host) { device_api 523 src/InjectHostDevBufferCopies.cpp if (device_api != DeviceAPI::Host || device_api 618 src/InjectHostDevBufferCopies.cpp if (device_api != DeviceAPI::Host) { device_api 640 src/InjectHostDevBufferCopies.cpp if (device_api != DeviceAPI::Host) { device_api 711 src/InjectHostDevBufferCopies.cpp if (device_api != DeviceAPI::Host) { device_api 732 src/InjectHostDevBufferCopies.cpp if (different_device_api(device_api, op->device_api, target)) { device_api 733 src/InjectHostDevBufferCopies.cpp debug(4) << "Switching from device_api " << static_cast<int>(device_api) << " to op->device_api " << device_api 734 src/InjectHostDevBufferCopies.cpp static_cast<int>(op->device_api) << " in for loop " << op->name <<"\n"; device_api 735 src/InjectHostDevBufferCopies.cpp DeviceAPI old_device_api = device_api; device_api 736 src/InjectHostDevBufferCopies.cpp device_api = fixup_device_api(op->device_api, target); device_api 737 src/InjectHostDevBufferCopies.cpp if (device_api == DeviceAPI::None) { device_api 738 src/InjectHostDevBufferCopies.cpp device_api = old_device_api; device_api 740 src/InjectHostDevBufferCopies.cpp internal_assert(device_api != DeviceAPI::None); device_api 744 src/InjectHostDevBufferCopies.cpp device_api = old_device_api; device_api 147 src/InjectImageIntrinsics.cpp loop->device_api == DeviceAPI::GLSL) { device_api 523 src/LoopCarry.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, body); device_api 453 src/PartitionLoops.cpp if (op->device_api == DeviceAPI::GLSL) { device_api 635 src/PartitionLoops.cpp op->for_type, op->device_api, simpler_body); device_api 639 src/PartitionLoops.cpp op->for_type, op->device_api, prologue); device_api 644 src/PartitionLoops.cpp op->for_type, op->device_api, epilogue); device_api 663 src/PartitionLoops.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, stmt); device_api 726 src/PartitionLoops.cpp if (op->device_api == DeviceAPI::GLSL) { device_api 794 src/PartitionLoops.cpp inner = For::make(f->name, f->min, f->extent, f->for_type, f->device_api, inner); device_api 872 src/PartitionLoops.cpp inner = For::make(for_a->name, for_a->min, for_a->extent, for_a->for_type, for_a->device_api, inner); device_api 252 src/Prefetch.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, body); device_api 102 src/PrintLoopNest.cpp out << op->device_api; device_api 218 src/Profiling.cpp bool update_active_threads = (op->device_api == DeviceAPI::Hexagon || device_api 234 src/Profiling.cpp if (op->device_api == DeviceAPI::Hexagon) { device_api 250 src/Profiling.cpp } else if (op->device_api == DeviceAPI::None || device_api 251 src/Profiling.cpp op->device_api == DeviceAPI::Host) { device_api 257 src/Profiling.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, body); device_api 14 src/RemoveTrivialForLoops.cpp if (for_loop->device_api != DeviceAPI::None) { device_api 44 src/RemoveTrivialForLoops.cpp stmt = For::make(for_loop->name, for_loop->min, for_loop->extent, for_loop->for_type, for_loop->device_api, body); device_api 265 src/RemoveUndef.cpp stmt = For::make(op->name, min, extent, op->for_type, op->device_api, body); device_api 201 src/Schedule.h DeviceAPI device_api; device_api 213 src/ScheduleFunctions.cpp stmt = For::make(nest[i].name, min, extent, dim.for_type, dim.device_api, stmt); device_api 756 src/ScheduleFunctions.cpp for_loop->device_api, device_api 1031 src/ScheduleFunctions.cpp if (!target.supports_device_api(d.device_api)) { device_api 1033 src/ScheduleFunctions.cpp << " requires " << d.device_api device_api 1127 src/ScheduleFunctions.cpp op->device_api == DeviceAPI::None) { device_api 16 src/SelectGPUAPI.cpp DeviceAPI selected_api = op->device_api; device_api 17 src/SelectGPUAPI.cpp if (op->device_api == DeviceAPI::Default_GPU) { device_api 29 src/SelectGPUAPI.cpp if (op->device_api != selected_api) { device_api 4810 src/Simplify.cpp stmt = For::make(op->name, new_min, new_extent, op->for_type, op->device_api, new_body); device_api 289 src/SlidingWindow.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, l->body); device_api 348 src/SlidingWindow.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, new_body); device_api 226 src/StorageFolding.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, body); device_api 250 src/StorageFolding.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, body); device_api 80 src/Substitute.cpp stmt = For::make(op->name, new_min, new_extent, op->for_type, op->device_api, new_body); device_api 325 src/TrimNoOps.cpp stmt = For::make(op->name, min, extent, op->for_type, op->device_api, body); device_api 363 src/TrimNoOps.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, body); device_api 376 src/TrimNoOps.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, body); device_api 419 src/TrimNoOps.cpp stmt = For::make(op->name, new_min_var, new_extent, op->for_type, op->device_api, body); device_api 89 src/UniquifyVariableNames.cpp stmt = For::make(new_name, min, extent, op->for_type, op->device_api, body); device_api 115 src/VaryingAttributes.cpp bool kernel_loop = op->device_api == DeviceAPI::GLSL; device_api 139 src/VaryingAttributes.cpp stmt = For::make(op->name, op->min, op->extent, op->for_type, op->device_api, mutated_body); device_api 998 src/VaryingAttributes.cpp if (CodeGen_GPU_Dev::is_gpu_var(op->name) && op->device_api == DeviceAPI::GLSL) { device_api 1143 src/VaryingAttributes.cpp if (CodeGen_GPU_Dev::is_gpu_var(op->name) && op->device_api == DeviceAPI::GLSL) { device_api 783 src/VectorizeLoops.cpp Stmt transformed = For::make(op->name, 0, op->extent, for_type, op->device_api, body); device_api 806 src/VectorizeLoops.cpp stmt = For::make(op->name, min, extent, for_type, op->device_api, body); device_api 925 src/VectorizeLoops.cpp if (for_loop->device_api == DeviceAPI::Hexagon) { device_api 946 src/VectorizeLoops.cpp if (for_loop->device_api == DeviceAPI::Hexagon) {