taskflow

taskflow

现代C++并行和异构任务编程库Taskflow

Taskflow是一个C++任务编程库,专注于并行和异构计算。它通过简洁的API支持静态和动态任务图构建、条件执行和GPU加速。Taskflow提供可视化和分析工具,有助于程序优化。该库在性能和可扩展性方面表现优异,适用于各类并行计算场景,已在学术和工业项目中得到应用。

Taskflow并行编程任务图异步编程GPU加速Github开源项目

Taskflow <img align="right" width="10%" src="https://yellow-cdn.veclightyear.com/2b54e442/87170130-10b6-46d0-8dda-cbf48c1546dc.png">

<!--[![Linux 构建状态](https://travis-ci.com/taskflow/taskflow.svg?branch=master)](https://travis-ci.com/taskflow/taskflow)-->

Ubuntu macOS Windows 维基 TFProf 引用

Taskflow 帮助您使用现代 C++ 快速编写并行和异构任务程序

为什么选择 Taskflow?

在处理复杂的并行工作负载时,Taskflow 比许多现有的任务编程框架更快、更具表现力,并且更容易集成。

Taskflow 让您能够快速实现任务分解策略,结合了常规和不规则的计算模式,并配合高效的工作窃取调度器来优化您的多线程性能。

静态任务子流任务
<img align="right" src="https://yellow-cdn.veclightyear.com/2b54e442/6dbd64d7-b086-4d9b-9b96-4519b4274dde.svg" width="100%">

Taskflow 支持条件任务,让您能够在相互依赖的任务之间快速做出控制流决策,实现循环和条件,这在使用现有工具时往往很难实现。

条件任务

Taskflow 可组合。您可以通过组合模块化和可重用的块来创建大型并行图,这些块在单独的范围内更容易优化。

Taskflow 组合

Taskflow 支持异构任务,让您能够利用 CPU-GPU 协同计算的能力来加速各种科学计算应用。

并发 CPU-GPU 任务

Taskflow 提供了用于分析 Taskflow 程序所需的可视化和工具。

Taskflow 分析器

我们致力于为并行计算领域的学术和工业研究项目提供可信赖的开发支持。查看谁在使用 Taskflow以及用户对我们的评价:

查看快速演示并访问文档以了解更多关于 Taskflow 的信息。 技术细节可参考我们的 [IEEE TPDS 论文][TPDS21]。

开始您的第一个 Taskflow 程序

以下程序(simple.cpp)创建了四个任务 ABCD,其中 ABC 之前运行,DBC 之后运行。 当 A 完成时,BC 可以并行运行。 在 Compiler Explorer (godbolt) 上实时尝试!

#include <taskflow/taskflow.hpp> // Taskflow 是仅头文件的库 int main(){ tf::Executor executor; tf::Taskflow taskflow; auto [A, B, C, D] = taskflow.emplace( // 创建四个任务 [] () { std::cout << "TaskA\n"; }, [] () { std::cout << "TaskB\n"; }, [] () { std::cout << "TaskC\n"; }, [] () { std::cout << "TaskD\n"; } ); A.precede(B, C); // A 在 B 和 C 之前运行 D.succeed(B, C); // D 在 B 和 C 之后运行 executor.run(taskflow).wait(); return 0; }

Taskflow 是仅头文件的库,无需安装。 要编译程序,请克隆 Taskflow 项目并 告诉编译器包含 头文件

~$ git clone https://github.com/taskflow/taskflow.git # 只需克隆一次 ~$ g++ -std=c++20 examples/simple.cpp -I. -O2 -pthread -o simple ~$ ./simple TaskA TaskC TaskB TaskD

可视化您的第一个 Taskflow 程序

Taskflow 自带一个内置的分析器, TFProf, 让您能够在易于使用的基于 Web 的界面中分析和可视化 taskflow 程序。

# 运行程序时启用环境变量 TF_ENABLE_PROFILER ~$ TF_ENABLE_PROFILER=simple.json ./simple ~$ cat simple.json [ {"executor":"0","data":[{"worker":0,"level":0,"data":[{"span":[172,186],"name":"0_0","type":"static"},{"span":[187,189],"name":"0_1","type":"static"}]},{"worker":2,"level":0,"data":[{"span":[93,164],"name":"2_0","type":"static"},{"span":[170,179],"name":"2_1","type":"static"}]}]} ] # 将分析 json 数据粘贴到 https://taskflow.github.io/tfprof/

除了执行图外,您还可以将图导出为 DOT 格式, 并使用多个免费的 GraphViz 工具进行可视化。

// 通过 std::cout 将 taskflow 图导出为 DOT 格式
taskflow.dump(std::cout); 
<p align="center"><img src="https://yellow-cdn.veclightyear.com/2b54e442/963e5186-2725-4cd3-a965-0613aeb9fa35.svg"></p>

表达任务图并行性

Taskflow 为用户提供了静态和动态任务图构建能力, 使其能够在嵌入图内控制流的任务图中表达端到端的并行性。

  1. 创建子流图
  2. 在任务图中集成控制流
  3. 将任务卸载到 GPU
  4. 组合任务图
  5. 启动异步任务
  6. 执行 Taskflow
  7. 利用标准并行算法

创建子流图

Taskflow 支持动态任务,让您可以从任务的执行中创建子流图 以实现动态并行性。 以下程序在任务 B 中生成一个任务依赖图。

tf::Task A = taskflow.emplace([](){}).name("A"); tf::Task C = taskflow.emplace([](){}).name("C"); tf::Task D = taskflow.emplace([](){}).name("D"); tf::Task B = taskflow.emplace([] (tf::Subflow& subflow) { tf::Task B1 = subflow.emplace([](){}).name("B1"); tf::Task B2 = subflow.emplace([](){}).name("B2"); tf::Task B3 = subflow.emplace([](){}).name("B3"); B3.succeed(B1, B2); // B3 在 B1 和 B2 之后运行 }).name("B"); A.precede(B, C); // A 在 B 和 C 之前运行 D.succeed(B, C); // D 在 B 和 C 之后运行
<p align="center"><img src="https://yellow-cdn.veclightyear.com/2b54e442/a64857c1-9b3e-4b43-9522-367e41dd5bdd.svg"></p>

在任务图中集成控制流

Taskflow 支持条件任务,让您能够在相互依赖的任务之间 快速做出控制流决策,以在端到端任务图中实现循环 和条件。

tf::Task init = taskflow.emplace([](){}).name("init"); tf::Task stop = taskflow.emplace([](){}).name("stop"); // 创建一个返回随机二进制值的条件任务 tf::Task cond = taskflow.emplace( [](){ return std::rand() % 2; } ).name("cond"); init.precede(cond); // 创建一个反馈循环 {0: cond, 1: stop} cond.precede(cond, stop);
<p align="center"><img src="https://yellow-cdn.veclightyear.com/2b54e442/bcee6a89-ab4f-4257-89c1-c0bd4518c8ad.svg"></p>

将任务卸载到 GPU

Taskflow 支持 GPU 任务,让您能够利用 CPU-GPU 协同计算的能力,使用 CUDA 加速各种科学计算应用。

__global__ void saxpy(size_t N, float alpha, float* dx, float* dy) { int i = blockIdx.x*blockDim.x + threadIdx.x; if (i < n) { y[i] = a*x[i] + y[i]; } } tf::Task cudaflow = taskflow.emplace([&](tf::cudaFlow& cf) { // 数据复制任务 tf::cudaTask h2d_x = cf.copy(dx, hx.data(), N).name("h2d_x"); tf::cudaTask h2d_y = cf.copy(dy, hy.data(), N).name("h2d_y"); tf::cudaTask d2h_x = cf.copy(hx.data(), dx, N).name("d2h_x"); tf::cudaTask d2h_y = cf.copy(hy.data(), dy, N).name("d2h_y"); // 带有启动 saxpy 内核参数的内核任务 tf::cudaTask saxpy = cf.kernel( (N+255)/256, 256, 0, saxpy, N, 2.0f, dx, dy ).name("saxpy"); saxpy.succee <p align="center"><img src="https://yellow-cdn.veclightyear.com/2b54e442/776d5f51-defc-4154-bdfa-61c85b1c7379.svg"></p> ## 启动异步任务 Taskflow 支持*异步*任务。 您可以异步启动任务以动态探索任务图并行性。 ```cpp tf::Executor executor; // 直接从执行器创建异步任务 std::future<int> future = executor.async([](){ std::cout << "异步任务返回1\n"; return 1; }); executor.silent_async([](){ std::cout << "异步任务不返回\n"; }); // 创建具有动态依赖关系的异步任务 tf::AsyncTask A = executor.silent_dependent_async([](){ printf("A\n"); }); tf::AsyncTask B = executor.silent_dependent_async([](){ printf("B\n"); }, A); tf::AsyncTask C = executor.silent_dependent_async([](){ printf("C\n"); }, A); tf::AsyncTask D = executor.silent_dependent_async([](){ printf("D\n"); }, B, C); executor.wait_for_all();

执行 Taskflow

执行器提供了几种线程安全的方法来运行 taskflow。 您可以运行一次 taskflow、多次运行或直到满足停止条件为止。 这些方法是非阻塞的,返回 tf::Future<void> 让您查询执行状态。

// 运行 taskflow 一次 tf::Future<void> run_once = executor.run(taskflow); // 等待此次运行完成 run_once.get(); // 运行 taskflow 四次 executor.run_n(taskflow, 4); // 运行 taskflow 五次 executor.run_until(taskflow, [counter=5](){ return --counter == 0; }); // 阻塞执行器直到所有提交的 taskflow 完成 executor.wait_for_all();

利用标准并行算法

Taskflow 定义了算法,让您可以使用标准 C++ 语法快速表达常见的并行模式, 例如并行迭代、并行归约和并行排序。

// 标准并行 CPU 算法 tf::Task task1 = taskflow.for_each( // 并行将每个元素赋值为100 first, last, [] (auto& i) { i = 100; } ); tf::Task task2 = taskflow.reduce( // 并行归约一系列项 first, last, init, [] (auto a, auto b) { return a + b; } ); tf::Task task3 = taskflow.sort( // 并行排序一系列项 first, last, [] (auto a, auto b) { return a < b; } ); // 标准并行 GPU 算法 tf::cudaTask cuda1 = cudaflow.for_each( // 在 GPU 上将每个元素赋值为100 dfirst, dlast, [] __device__ (auto i) { i = 100; } ); tf::cudaTask cuda2 = cudaflow.reduce( // 在 GPU 上归约一系列项 dfirst, dlast, init, [] __device__ (auto a, auto b) { return a + b; } ); tf::cudaTask cuda3 = cudaflow.sort( // 在 GPU 上排序一系列项 dfirst, dlast, [] __device__ (auto a, auto b) { return a < b; } );

此外,Taskflow 提供了可组合的图形构建块,让您能够 高效实现常见的并行算法,如并行流水线。

// 创建一个流水线,通过三个串行阶段传播五个令牌 tf::Pipeline pl(num_parallel_lines, tf::Pipe{tf::PipeType::SERIAL, [](tf::Pipeflow& pf) { if(pf.token() == 5) { pf.stop(); } }}, tf::Pipe{tf::PipeType::SERIAL, [](tf::Pipeflow& pf) { printf("阶段 2: 输入缓冲区[%zu] = %d\n", pf.line(), buffer[pf.line()]); }}, tf::Pipe{tf::PipeType::SERIAL, [](tf::Pipeflow& pf) { printf("阶段 3: 输入缓冲区[%zu] = %d\n", pf.line(), buffer[pf.line()]); }} ); taskflow.composed_of(pl) executor.run(taskflow).wait();

支持的编译器

要使用 Taskflow,您只需要一个支持 C++17 的编译器:

  • GNU C++ 编译器至少 v8.4 版本,使用 -std=c++17
  • Clang C++ 编译器至少 v6.0 版本,使用 -std=c++17
  • Microsoft Visual Studio 至少 v19.27 版本,使用 /std:c++17
  • AppleClang Xcode 版本至少 v12.0,使用 -std=c++17
  • Nvidia CUDA 工具包和编译器 (nvcc) 至少 v11.1 版本,使用 -std=c++17
  • Intel C++ 编译器至少 v19.0.1 版本,使用 -std=c++17
  • Intel DPC++ Clang 编译器至少 v13.0.0 版本,使用 -std=c++17 和 SYCL20

Taskflow 适用于 Linux、Windows 和 Mac OS X。

尽管 %Taskflow 主要支持 C++17,但您可以通过 -std=c++20 启用 C++20 编译 以实现更好的性能,这得益于新的 C++20 特性。

了解更多关于 Taskflow 的信息

访问我们的项目网站文档 以了解更多关于 Taskflow 的信息。要参与其中:

CppCon20 技术讲座MUC++ 技术讲座
<img align="right" src="https://yellow-cdn.veclightyear.com/2b54e442/75ee52b4-6640-441a-9b17-b690f4203911.jpg" width="100%">

我们致力于支持并行和异构计算领域的学术和工业研究项目的可信开发。 如果您正在使用 Taskflow,请引用我们在 2021 年 IEEE TPDS 上发表的以下论文:

更重要的是,我们感谢所有 Taskflow 贡献者以及 以下赞助 Taskflow 项目的组织!

<!-- --><!-- --><!-- --><!-- -->
<img src="https://yellow-cdn.veclightyear.com/2b54e442/46d9eeb8-f420-4a13-9cf0-6e26e2256585.png"><img src="https://yellow-cdn.veclightyear.com/2b54e442/61e07628-cd93-42ae-b0ba-c31ac4945d3a.png"><img src="https://yellow-cdn.veclightyear.com/2b54e442/6bb06c17-3f9c-463c-b18b-59f4e35d7b7b.png"><img src="https://yellow-cdn.veclightyear.com/2b54e442/f3dd4b2d-fa29-4644-85af-170a25f8d4f2.png">
<img src="https://yellow-cdn.veclightyear.com/2b54e442/47e071eb-80c4-4179-998b-a974cc8600f8.png">

许可证

Taskflow 采用 MIT 许可证。 您完全可以自由地重新分发您从 Taskflow 衍生的作品。


编辑推荐精选

咔片PPT

咔片PPT

AI助力,做PPT更简单!

咔片是一款轻量化在线演示设计工具,借助 AI 技术,实现从内容生成到智能设计的一站式 PPT 制作服务。支持多种文档格式导入生成 PPT,提供海量模板、智能美化、素材替换等功能,适用于销售、教师、学生等各类人群,能高效制作出高品质 PPT,满足不同场景演示需求。

讯飞绘文

讯飞绘文

选题、配图、成文,一站式创作,让内容运营更高效

讯飞绘文,一个AI集成平台,支持写作、选题、配图、排版和发布。高效生成适用于各类媒体的定制内容,加速品牌传播,提升内容营销效果。

AI助手热门AI工具AI创作AI辅助写作讯飞绘文内容运营个性化文章多平台分发
材料星

材料星

专业的AI公文写作平台,公文写作神器

AI 材料星,专业的 AI 公文写作辅助平台,为体制内工作人员提供高效的公文写作解决方案。拥有海量公文文库、9 大核心 AI 功能,支持 30 + 文稿类型生成,助力快速完成领导讲话、工作总结、述职报告等材料,提升办公效率,是体制打工人的得力写作神器。

openai-agents-python

openai-agents-python

OpenAI Agents SDK,助力开发者便捷使用 OpenAI 相关功能。

openai-agents-python 是 OpenAI 推出的一款强大 Python SDK,它为开发者提供了与 OpenAI 模型交互的高效工具,支持工具调用、结果处理、追踪等功能,涵盖多种应用场景,如研究助手、财务研究等,能显著提升开发效率,让开发者更轻松地利用 OpenAI 的技术优势。

Hunyuan3D-2

Hunyuan3D-2

高分辨率纹理 3D 资产生成

Hunyuan3D-2 是腾讯开发的用于 3D 资产生成的强大工具,支持从文本描述、单张图片或多视角图片生成 3D 模型,具备快速形状生成能力,可生成带纹理的高质量 3D 模型,适用于多个领域,为 3D 创作提供了高效解决方案。

3FS

3FS

一个具备存储、管理和客户端操作等多种功能的分布式文件系统相关项目。

3FS 是一个功能强大的分布式文件系统项目,涵盖了存储引擎、元数据管理、客户端工具等多个模块。它支持多种文件操作,如创建文件和目录、设置布局等,同时具备高效的事件循环、节点选择和协程池管理等特性。适用于需要大规模数据存储和管理的场景,能够提高系统的性能和可靠性,是分布式存储领域的优质解决方案。

TRELLIS

TRELLIS

用于可扩展和多功能 3D 生成的结构化 3D 潜在表示

TRELLIS 是一个专注于 3D 生成的项目,它利用结构化 3D 潜在表示技术,实现了可扩展且多功能的 3D 生成。项目提供了多种 3D 生成的方法和工具,包括文本到 3D、图像到 3D 等,并且支持多种输出格式,如 3D 高斯、辐射场和网格等。通过 TRELLIS,用户可以根据文本描述或图像输入快速生成高质量的 3D 资产,适用于游戏开发、动画制作、虚拟现实等多个领域。

ai-agents-for-beginners

ai-agents-for-beginners

10 节课教你开启构建 AI 代理所需的一切知识

AI Agents for Beginners 是一个专为初学者打造的课程项目,提供 10 节课程,涵盖构建 AI 代理的必备知识,支持多种语言,包含规划设计、工具使用、多代理等丰富内容,助您快速入门 AI 代理领域。

AEE

AEE

AI Excel全自动制表工具

AEE 在线 AI 全自动 Excel 编辑器,提供智能录入、自动公式、数据整理、图表生成等功能,高效处理 Excel 任务,提升办公效率。支持自动高亮数据、批量计算、不规则数据录入,适用于企业、教育、金融等多场景。

UI-TARS-desktop

UI-TARS-desktop

基于 UI-TARS 视觉语言模型的桌面应用,可通过自然语言控制计算机进行多模态操作。

UI-TARS-desktop 是一款功能强大的桌面应用,基于 UI-TARS(视觉语言模型)构建。它具备自然语言控制、截图与视觉识别、精确的鼠标键盘控制等功能,支持跨平台使用(Windows/MacOS),能提供实时反馈和状态显示,且数据完全本地处理,保障隐私安全。该应用集成了多种大语言模型和搜索方式,还可进行文件系统操作。适用于需要智能交互和自动化任务的场景,如信息检索、文件管理等。其提供了详细的文档,包括快速启动、部署、贡献指南和 SDK 使用说明等,方便开发者使用和扩展。

下拉加载更多