1.[转]Megatron-LM源码系列(八): Context Parallel并行
2.Go 组件:context 学习笔记
3.Context 使用场景&&源码解读
4.SpringBoot源码 | refreshContext方法解析
5.Spring Boot源码解析(四)ApplicationContext准备阶段
6.React中不常用的编编程功能——Context
[转]Megatron-LM源码系列(八): Context Parallel并行
原文链接: Megatron-LM源码系列(八): Context Parallel并行
Context Parallel并行(CP)与sequence并行(SP)相比,核心差异在于SP只针对Layernorm和Dropout输出的程源activation在sequence维度进行切分,而CP则进一步扩展,编编程对所有input输入和所有输出activation在sequence维度上进行切分,程源形成更高效的编编程并行处理策略。除了Attention模块外,程源小说捕快程序源码其他如Layernorm、编编程Dropout等模块在CP并行中无需任何修改,程源因为它们在处理过程中没有涉及多token间的编编程交互。
Attention模块之所以特殊,程源是编编程因为在计算过程中,每个token的程源查询(query)需要与同一sequence中其他token的键(key)和值(value)进行交互计算,存在内在依赖性。编编程xpshop 源码因此,程源在进行CP并行时,编编程计算开始前需要通过allgather通信手段获取所有token的KV向量,反向计算时则通过reduce_scatter分发gradient梯度。
为了降低显存使用,前向计算阶段每个GPU仅保存部分KV块,反向阶段则通过allgather通信获取全部KV数据。这些通信操作在特定的rank位置(相同TP组内)进行,底层通过send和recv等操作实现allgather和reduce_scatter。
以TP2-CP2的transformer网络为例,CP并行的通信操作在Attention之前执行,其他则为TP通信。doubango源码AG表示allgather,RS表示reduce_scatter,AG/RS表示前向allgather反向reduce_scatter,RS/AG表示前向reduce_scatter反向allgather。
TP2对应为[GPU0, GPU1], [GPU2, GPU3],CP2指的就是TP组相同位置的rank号,即[GPU0, GPU2], [GPU1, GPU3]。CP并行类似于Ring Attention,但提供了OSS与FlashAttention版本,并去除了冗余的low-triangle causal masking计算。
LLM常因序列长度过长而导致显存耗尽(OOM)。传统解决方法包括重计算或扩大TP(tensor parallel)大小,workerman源码但各自存在计算代价增加或线性fc计算时间减少与通信难以掩盖的问题。CP则能更高效地解决这一问题,每个GPU处理一部分序列,同时减少CP倍的通信和计算量,同时保持TP不变,使得activation量也减少CP倍。性能优化结果展示于图表中,用户可通过指定--context-parallel-size在Megatron中实现CP。
具体源码实现以Megatron-Core 0.5.0版本为例进行说明。
参考资料:
Go 组件:context 学习笔记
作者:@神奇的君,腾讯CSIG后台开发工程师
在追求Go语言强大并发能力的过程中,Context成为了不可或缺的pidof 源码工具。它使得协程间的协作和管理更加便捷,尤其是在需要控制协程状态和同步操作时。
想象没有Context的世界,每个编程语言都有自己的并发控制方式,比如Java通过join进行同步。在Go中,channel和sync包常用于协程间通信,channel用于通知协程执行特定操作,而sync则用于加锁和同步。
以往,若要实现协程的终止,可能会通过channel来控制,因为Go协程不支持直接外部退出。例如,通过设置关闭条件,每个协程需要判断channel的状态,这在处理全局时间限制时显得复杂。这时,Context的出现就显得尤为重要。
Context简化了代码,通过Context,我们可以轻松地在库函数中利用ctx.Done()来判断协程状态。例如,/post/spring...
Vert.x 源码解析(4.x)——Context源码解析
Vert.x 4.x 源码深度解析:Context核心概念详解 Vert.x 通过Context这一核心机制,解决了多线程环境下的资源管理和状态维护难题。Context在异步编程中扮演着协调者角色,确保线程安全的资源访问和有序的异步操作。本文将深入剖析Context的源码结构,包括其接口设计、关键实现以及在Vert.x中的具体应用。Context源代码解析
Context接口定义了基础的事件处理功能,如立即执行和阻塞任务。ContextInternal扩展了Context,包含内部方法和功能,通常开发者无需直接接触,如获取当前线程的Context。在vertx的beginDispatch和endDispatch方法中,Context的切换策略取决于线程类型,Vertx线程会使用上下文切换,而非Vertx线程则依赖ThreadLocal。 ContextBase是ContextInternal的实现类,负责执行耗时任务,内部包含TaskQueue来管理任务顺序。WorkerContext和EventLoopContext分别对应工作线程和EventLoop线程的执行策略,它们通过execute()、runOnContext()和emit()方法处理任务,同时监控性能。 Context的创建和获取贯穿于Vert.x的生命周期,它在DeploymentManager的doDeploy方法中被调用,如NetServer和NetClient等组件的底层实现也依赖于Context来处理网络通信。额外说明
Context与线程并非直接绑定,而是根据场景动态管理。部署时创建新Context,非部署时优先获取Thread和ThreadLocal中的Context。当执行异步任务时,当前线程的Context会被暂时替换,任务完成后才恢复。源码中已加入详细注释,如需获取完整注释版本,可联系作者。 Context的重要性在于其在Vert.x的各个层面如服务器部署、EventBus通信中不可或缺,它负责维护线程同步与异步任务的执行顺序,是异步编程中不可或缺的基石。理解Context的实现,有助于更好地利用Vert.x进行高效开发。Applicationä¸ç Context å Activity ä¸çContextåºå«
Contextå¨æ们å¼åä¸ç»å¸¸ç¨å°ï¼ä¸ç®¡æ¯Frameworkæä¾ç»æ们çå大ç»ä»¶ï¼è¿æ¯åºç¨çº§å«çApplicationï¼è¿æ¯è´è´£è¡¨ç°å±çViewç¸å ³ç±»ï¼çè³è¿æ们å¾å¤æ¶åå建çå®ä½ç±»é½ä¼éè¦ææä¸ä¸ªContextçå¼ç¨ãé£ä¹Contextå°åºæ¯ä»ä¹å¢ï¼å»ºè®®çè¿ä¸ªï¼ /p/bde4cb
Contextè±æéä¹æ¯å½åä¸ä¸æï¼æè å½ååºæ¯ä¸ï¼
å®æ¹ææ¡£ï¼Context
public abstractclass Context extends Object
Interface to globalinformation about an application environment. This is an abstract class whoseimplementation is provided by the Android system. It allows access toapplication-specific resources and classes, as well as up-calls forapplication-level operations such as launching activities, broadcasting andreceiving intents, etc.
ç±å®æ¹ææ¡£ï¼æ们å¯ä»¥ç¥éï¼
1.该类æ¯ä¸ä¸ªæ½è±¡ï¼abstract classï¼ç±»ï¼
2.å®æè¿°çæ¯ä¸ä¸ªåºç¨ç¨åºç¯å¢çä¿¡æ¯ï¼å³ä¸ä¸æï¼
3.éè¿å®ï¼Contextï¼æ们å¯ä»¥è·ååºç¨ç¨åºçèµæºåç±»ï¼ä¹å æ¬ä¸äºåºç¨çº§å«çæä½ï¼ä¾å¦ï¼å¯å¨ Activity,广æåæå¡çï¼;
åé¢æä»¬è®²è¿ Context æ¯ä¸ä¸ªæ½è±¡ç±»ï¼éè¿ Contextæ们å¯ä»¥è·ååºç¨ç¨åºçèµæºåç±»ï¼è°ç¨å®ä»¬çæ¹æ³ï¼é£ä¹å ·ä½å®ä¹çæ¹æ³æåªäºå¢ï¼æ们æ¥çä¸ä¸ Context çæºç ï¼
æºç éçæ¹æ³å¤ªå¤äºï¼æ»å ± è¡ãæ们ä»ä»¥ä¸é¨åæºç çå°äºçæç对象---ApplicationãActivityãServiceãBroadcastãè¿äºå¯¹è±¡å Context çå ³ç³»å°åºæ¯ä»ä¹å¢ï¼æ们çä¸ä¸å®æ¹ææ¡£å¯ç¥ï¼
1.Acitiivity 继æ¿èªContextThemeWrapper--->å继æ¿ContextWrapper--->Contextã
2.Appliction ãService继æ¿èªContextWrapper--->å继æ¿Contextã
3.ApplicationãService å Activity æç»é½æ¯ç»§æ¿èªContextï¼æ以å®ä»¬æ¯åä¸ä¸ªä¸ä¸æã
éè¿ä»¥ä¸ç继æ¿å ³ç³»ï¼æ们就å¯ä»¥ç¥éï¼Contextçå ·ä½ä½ç¨ä¼å æ¬ï¼
- å¯å¨ä¸ä¸ªæ°çActivity
- å¯å¨ååæ¢Service
- åé广ææ¶æ¯(Intent)
- 注å广ææ¶æ¯(Intent)æ¥æ¶è
- å¯ä»¥è®¿é®APKä¸åç§èµæºï¼å¦ResourcesåAssetManager
- å建View
- 访é®Packageçç¸å ³ä¿¡æ¯
- APKçåç§æé管ç
ç±ä¸é¢åæç继æ¿å ³ç³»ï¼æ们å¯ä»¥ç¥éï¼Contextå建çæ¶æºæä¸ä¸ªï¼
â å建Application 对象æ¶ï¼ èä¸æ´ä¸ªAppå ±ä¸ä¸ªApplication对象ï¼
â¡å建Service对象æ¶ï¼
â¢å建Activity对象æ¶ï¼
æ以åºç¨ç¨åºAppå ±æçContextæ°ç®å ¬å¼ä¸ºï¼
Serviceä¸ªæ° + Activityä¸ªæ° + 1ï¼Application对åºçContextå®ä¾ï¼
å¦ä¸ï¼Androidä¸contextå¯ä»¥ä½å¾å¤æä½ï¼ä½æ¯æ主è¦çåè½æ¯å è½½å访é®èµæºãå¨androidä¸å¸¸ç¨çcontextæ两ç§ï¼ä¸ç§æ¯application contextï¼ä¸ç§æ¯activity contextï¼é常æ们å¨åç§ç±»åæ¹æ³é´ä¼ éçæ¯activity contextã
两è çåºå«ï¼
thisæ¯Activity çå®ä¾ï¼æ©å±äºContextï¼å ¶çå½å¨ææ¯Activity å建å°éæ¯ãgetApplicationContext()è¿ååºç¨çä¸ä¸æï¼çå½å¨ææ¯æ´ä¸ªåºç¨ï¼åºç¨æ§æ¯å®æ被æ§æ¯ãActivity.thisçcontext è¿åå½åactivityçä¸ä¸æï¼å±äºactivity ï¼activityæ§æ¯æ¶è¢«æ§æ¯ã
使ç¨Contextæ¶æéè¦æ³¨æçä¸ä¸ªç¹å°±æ¯ï¼ä½¿ç¨äºä¸æ£ç¡®çcontextï¼æ¯å¦æä¸ä¸ªå ¨å±çæ°æ®æä½ç±»ç¨å°äºcontextï¼è¿ä¸ªæ¶åå°±è¦getApplicationContext èä¸æ¯ç¨ACtivityï¼å¦æå¨è¿ä¸ªå ¨å±æä½ä¸å¼ç¨çæ¯Activityçcontextï¼é£ä¹å°±ä¼ä¸ç´å¼ç¨Activityçèµæºï¼å¯¼è´GCæ æ³åæ¶è¿é¨åå åï¼ä»èæç»å¯¼è´äºå åæ³æ¼ã
å åæ³æ¼æ¯å¼åä¸å¸¸è§çé误ä¹ä¸ï¼è½ä¸è½åç°åå³äºå¼åè çç»éªï¼å½ç¶äºæ们ä¹ä¼ä¾èµç°æçå åæ³æ¼åºï¼ä½æ¯å¦ææ们å¨å¼åçæºå¤´åå°å åæ³æ¼çæ¦çï¼é£ä¹åæçå·¥ä½ä¼å°å¾å¤ã
以ä¸æ¯é¿å contextç¸å ³çå åæ³é²ï¼ç»åºçå ç¹å»ºè®®ï¼
以ä¸ç表å举çæ¯ä¸ç§Context对象ç对åºä½¿ç¨åºæ¯ï¼
ä»è¡¨ä¸å¯ä»¥çå°ï¼åUIç¸å ³çé½ä½¿ç¨ActivityçContext对象ã
å°ç»ï¼å¦ä¸åæï¼Contextå¨å¯¹åºå¼åéçæ¥æºå°±æ¯ä¸ä¸ªââActivityãServiceåAppliactionï¼é£ä¹æ们该å¦ä½éæ©ä½¿ç¨åªä¸ä¸ªContext对象å¢ï¼ä¸ä¸ªæ¯è¾ç®åçæ¹æ³æ¯ï¼å½ä½ æ æ³ç¡®å®ä½¿ç¨æ个Context对象æ¯å¦ä¼é æé¿å¼ç¨å¯¼è´å åæ³æ¼æ¶ï¼é£ä¹å°±ä½¿ç¨AppliactionçContext对象ï¼å 为Appliactionåå¨äºæ´ä¸ªåºç¨ççå½å¨æå ã
å¨å®é å¼åä¸ï¼æ们å¾å¾ä¼ä¸ºé¡¹ç®å®ä¹ä¸ä¸ªApplictaionï¼ç¶åå¨AndroidMainfest.xmlæ件ä¸è¿è¡æ³¨åï¼
èä¸å¨èªå®ä¹Applicationå¾å¾ä¼å®ä¹å¥½ä¸ä¸ªéææ¹æ³ï¼ç¨ä»¥å ¨å±è·åapplicationå®ä¾ï¼
ActivityåApplicationé½æ¯Contextçåç±»ï¼ä½æ¯ä»ä»¬ç»´æ¤ççå½å¨æä¸ä¸æ ·ãåè ç»´æ¤ä¸ä¸ªAcitivityççå½å¨æï¼æä»¥å ¶å¯¹åºçContextä¹åªè½è®¿é®è¯¥activityå çåç§èµæºãåè åæ¯ç»´æ¤ä¸ä¸ªApplicationççå½å¨æã
1.å¦ä½å¤æcontextæ¯å±äºåªä¸ªactivityï¼
2.å ¨å±ä¸åå¦ä½è·å对åºçcontextï¼
éæå è½½ä¸ä¸ªFragmentï¼å¨onCreateView()æ¹æ³ä¸éè¿getActivityè·åä¸ä¸æå®ä¾ï¼
3.å大ç»ä»¶å¯ä»¥åæ®éJavaç±»ä¸æ ·ï¼éç¨newçæ¹å¼å®ä¾ååï¼
Androidç¨åºä¸åJavaç¨åºä¸æ ·ï¼é便å建ä¸ä¸ªç±»ï¼å个main()æ¹æ³å°±è½è¿è¡ï¼Androidåºç¨æ¨¡åæ¯åºäºç»ä»¶çåºç¨è®¾è®¡æ¨¡å¼ï¼ç»ä»¶çè¿è¡è¦æä¸ä¸ªå®æ´çAndroidå·¥ç¨ç¯å¢ï¼å¨è¿ä¸ªç¯å¢ä¸ï¼ActivityãServiceçç³»ç»ç»ä»¶æè½å¤æ£å¸¸å·¥ä½ï¼èè¿äºç»ä»¶å¹¶ä¸è½éç¨æ®éçJava对象å建æ¹å¼ï¼newä¸ä¸å°±è½å建å®ä¾äºï¼èæ¯è¦æå®ä»¬åèªçä¸ä¸æç¯å¢ï¼ä¹å°±æ¯æ们è¿é讨论çContextãå¯ä»¥è¿æ ·è®²ï¼Contextæ¯ç»´æAndroidç¨åºä¸åç»ä»¶è½å¤æ£å¸¸å·¥ä½çä¸ä¸ªæ ¸å¿åè½ç±»ã