PR cleanup from 1827. (tensorflow#1831)

* PR cleanup from 1827.

* Don't export GL context.

Author: nkreeger

tfjs-backend-nodegl/README.md

@@ -2,4 +2,4 @@
 
 ** This project is under heavy development **
 
-This new backend will provide a light-weight headless WebGL runtime for TensorFlow.js running under Node.js. This new backend is powered by the [node-gles](https://github.com/google/node-gles) module which uses [ANGLE](https://github.com/google/angle) to provide an integration layer to system GL runtime. This package aims to provide a think acceleration engine for IoT, desktop, and Node.js applications where CUDA (size/OS compatibility) is not an option./
+This new backend will provide a light-weight headless WebGL runtime for TensorFlow.js running under Node.js. This new backend is powered by the [node-gles](https://github.com/google/node-gles) module which uses [ANGLE](https://github.com/google/angle) to provide an integration layer to system GL runtime. This package aims to provide a think acceleration engine for IoT, desktop, and Node.js applications where CUDA (size/OS compatibility) is not an option.

tfjs-backend-nodegl/package.json

@@ -12,7 +12,7 @@
   "author": "",
   "license": "Apache-2.0",
   "dependencies": {
-    "@tensorflow/tfjs-core": "^1.2.2",
+    "@tensorflow/tfjs-core": "1.2.2",
     "node-gles": "^0.0.13"
   },
   "devDependencies": {

tfjs-backend-nodegl/src/index.ts

@@ -34,7 +34,8 @@ tf.ENV.set(
 tf.webgl.setWebGLContext(2, nodeGl);
 
 tf.registerBackend('headless-nodegl', () => {
-  return new tf.webgl.MathBackendWebGL(new tf.webgl.GPGPUContext(gl));
+  // TODO(kreeger): Consider moving all GL creation here. However, weak-ref to
+  // GL context tends to cause an issue when running unit tests:
+  // https://github.com/tensorflow/tfjs/issues/1732
+  return new tf.webgl.MathBackendWebGL(new tf.webgl.GPGPUContext(nodeGl));
 }, 3 /* priority */);
-
-export const gl = nodeGl;

tfjs-backend-nodegl/src/run_tests.ts

@@ -17,8 +17,9 @@
 
 // We import index.ts so that the Node backend gets registered.
 import * as jasmine_util from '@tensorflow/tfjs-core/dist/jasmine_util';
+import * as tf from '@tensorflow/tfjs-core';
 
-import * as nodegl from './index';
+import './index';
 
 Error.stackTraceLimit = Infinity;
 
@@ -70,7 +71,8 @@ env.specFilter = spec => {
 };
 
 console.log(`Running tests with the following GL info`);
-console.log(`  GL_VERSION: ${nodegl.gl.getParameter(nodegl.gl.VERSION)}`);
-console.log(`  GL_RENDERER: ${nodegl.gl.getParameter(nodegl.gl.RENDERER)}`);
+const gl = (tf.backend() as tf.webgl.MathBackendWebGL).getGPGPUContext().gl;
+console.log(`  GL_VERSION: ${gl.getParameter(gl.VERSION)}`);
+console.log(`  GL_RENDERER: ${gl.getParameter(gl.RENDERER)}`);
 
 runner.execute();

tfjs-backend-nodegl/yarn.lock

@@ -2,7 +2,7 @@
 # yarn lockfile v1
 
 
-"@tensorflow/tfjs-core@^1.2.2":
+"@tensorflow/tfjs-core@1.2.2":
   version "1.2.2"
   resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-core/-/tfjs-core-1.2.2.tgz#2efa89e323612a26aeccee9b3ae9f5ac5a635bbe"
   integrity sha512-2hCHMKjh3UNpLEjbAEaurrTGJyj/KpLtMSAraWgHA1vGY0kmk50BBSbgCDmXWUVm7lyh/SkCq4/GrGDZktEs3g==