fix (#1858)

tensorflow · Jul 26, 2019 · 225c9b1 · 225c9b1
1 parent 8361f6a
commit 225c9b1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/backends/webgl/encode_matrix_gpu.ts b/src/backends/webgl/encode_matrix_gpu.ts
@@ -45,7 +45,8 @@ export class EncodeMatrixProgram implements GPGPUProgram {
         int flatIndex = getFlatIndex(coords);
         int offset = imod(flatIndex, 4);
 
-        flatIndex /= 4;
+        flatIndex = idiv(flatIndex, 4, 1.);
+        
         int r = flatIndex / ${width};
         int c = imod(flatIndex, ${width});
         vec2 uv = (vec2(c, r) + halfCR) / vec2(${width}.0, ${height}.0);

diff --git a/src/backends/webgl/encode_matrix_packed_gpu.ts b/src/backends/webgl/encode_matrix_packed_gpu.ts
@@ -67,7 +67,8 @@ export class EncodeMatrixPackedProgram implements GPGPUProgram {
               flatIndex = getFlatIndex(localCoords);
               offset = imod(flatIndex, 4);
     
-              flatIndex /= 4;
+              flatIndex = idiv(flatIndex, 4, 1.);
+
               r = flatIndex / ${width};
               c = imod(flatIndex, ${width});
               uv = (vec2(c, r) + halfCR) / vec2(${width}.0, ${height}.0);