fix to run on Intel GPUs

neuroneural · Jun 21, 2024 · 01d5c51 · 01d5c51
1 parent e7f707d
commit 01d5c51
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/brainchop-mainthread.js b/brainchop-mainthread.js
@@ -1219,7 +1219,7 @@ async function enableProductionMode(textureF16Flag = true) {
   tf.env().set('DEBUG', false)
   tf.env().set('WEBGL_FORCE_F16_TEXTURES', textureF16Flag)
   // -- set this flag so that textures are deleted when tensors are disposed.
-  tf.env().set('WEBGL_DELETE_TEXTURE_THRESHOLD', 0)
+  tf.env().set('WEBGL_DELETE_TEXTURE_THRESHOLD', -1)
   // -- tf.env().set('WEBGL_PACK', false)
   // -- Put ready after sets above
   await tf.ready()

diff --git a/brainchop-webworker.js b/brainchop-webworker.js
@@ -1222,7 +1222,7 @@ async function enableProductionMode(textureF16Flag = true) {
   tf.env().set('DEBUG', false)
   tf.env().set('WEBGL_FORCE_F16_TEXTURES', textureF16Flag)
   // -- set this flag so that textures are deleted when tensors are disposed.
-  tf.env().set('WEBGL_DELETE_TEXTURE_THRESHOLD', 0)
+  tf.env().set('WEBGL_DELETE_TEXTURE_THRESHOLD', -1)
   // -- tf.env().set('WEBGL_PACK', false)
   // -- Put ready after sets above
   await tf.ready()

diff --git a/package.json b/package.json
@@ -12,7 +12,7 @@
   },
   "dependencies": {
     "@niivue/niivue": "^0.43.3",
-    "@tensorflow/tfjs": "^4.19.0",
+    "@tensorflow/tfjs": "^4.20.0",
     "gl-matrix": "^3.4.3"
   },
   "devDependencies": {