Merge pull request #441 from JingyuanZhang/master

fix converter and update split op

Author: JingyuanZhang

packages/paddlejs-backend-webgl/package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@paddlejs/paddlejs-backend-webgl",
-  "version": "1.2.0",
+  "version": "1.2.7",
   "description": "",
   "main": "lib/index",
   "scripts": {

packages/paddlejs-backend-webgl/package.json

@@ -4,12 +4,15 @@
 
 function mainFunc(
     {},
-    { target_length, num, dim }
+    { target_length, num, dim, sections }
 ) {
+
+    const splitChip = sections && sections.length > 1 ? sections[0] : target_length / num;
+
     return `
     // start函数
     void main(void) {
-        int length = int(${target_length} / ${num});
+        int length = int(${splitChip});
         ivec4 oPos = getOutputTensorPos();
         // 输出坐标转换为输入坐标
         oPos[${dim}] = oPos[${dim}] + layer_run_time * length;

packages/paddlejs-backend-webgl/src/ops/shader/split.ts

@@ -60,7 +60,7 @@ def opExistSingleNode(opName):
                 if op['type'] == fuse:
                     prevOp = ops[index - 1]
 
-                    if opExistSingleNode(prevOp['outputs']['Out'][0]):
+                    if opExistSingleNode(prevOp['outputs']['Out'][0]) and len(prevOp['outputs']['Out']) == 1 :
                         prevOp['attrs']['fuse_opt'] = {}
                         if 'fuse_opt' in op['attrs']:
                             prevOp['attrs']['fuse_opt'] = op['attrs']['fuse_opt']