本文主要基于MXNet1.6.0版本进行分析. MXNet的KVStore模块下有几个比较重要的类.KVStore是一个抽象类,提供了一些通用的API,例如Init.Push和Pull等.因为KVSotre支持int和string两种类型的key,所以这些API以不同类型的key作为参数,提供了两种重载. KVStoreLocal继承自KVStore,负责进程内通信.它主要维护了以下变量:负责不同设备间通信的comm_,机器上的页锁定内存(不进行页交换,一直在物理内存中),本机的key-val…