• C/C++接口
    • PSAgent初始化和启动
    • Matrix操作

    当前主流的深度学习库，一般使用C/C++来实现，为了让它们能使用Angel作为参数服务器，Angel需要提供C/C++接口，让深度学习框架，可以从它们内部，启动Angel并调用相应接口，实现深度学习的参数服务器。

    Angel目前基于Java实现，目前已经实现了一个基本的可以启动PSAgent，并可以发送更新以及获取参数。

    • 用C来调用，需要借助JNI
    • 支持多线程环境，JNI的JavaVM对象可以在线程之间共享，但JNIEnv不可以。
    • 抽象出C版本的PSAgent和MatrixClient类或接口
    • PSAgent确保每个进程只启动一个实例

    具体接口如下：

    C/C++接口

    PSAgent初始化和启动

    1. 初始化

      1.      PSAgent psAgent = new PSAgent(conf, masterIp, masterPort, clientIndex);
      2.      conf:Configuration //Angel系统配置
      3.      masterIp:String //Angel app master IP地址
      4.      masterPort:int //Angel app master 端口
      5.      clientIndex:int //client 编号（全局唯一）

    2. 启动

      1.  psAgent.initAndStart();

    Matrix操作

    1. 获取矩阵客户端

      1.      MatrixClient matrixClient = psAgent.getMatrixClient(matrixName, taskIndex);
      2.      matrixName:String //矩阵的名字
      3.      taskIndex:int task //编号（全局唯一）任务执行单元编号，用作同步控制（BSP，SSP）之用

    2. Matrix接口说明

      1.      //将矩阵每一个元素置为随机数
      2.      public void random(int range) throws Exception;
      4.      //将矩阵每一个元素置为0
      5.      public void zero() throws Exception;
      7.      //更新矩阵
      8.      public void increment(double[] delta) throws Exception;
      10.      //按索引更新矩阵
      11.      public void increment(int[] indexes, double[] delta) throws Exception;
      13.      //按索引获取矩阵
      14.      public double[] get(int[] indexes) throws Exception;
      16.      //获取矩阵
      17.      public double[] get() throws Exception;
      19.      //将PSAgent本地缓存的矩阵更新刷新到PS
      20.      public Future<VoidResult> flush() throws Exception;
      22.      //将PSAgent本地缓存的矩阵更新刷新到PS，同时更新clock值
      23.      public Future<VoidResult> clock() throws Exception;  
      25.      //聚合UDF，例如sum，max等
      26.      public AggrResult aggr(AggrFunc func) throws Exception;
      28.      //矩阵更新UDF，例如scalar，increment等
      29.      public Future<VoidResult> update(UpdaterFunc func) throws Exception;