Method createProgram

tfjs-backend-webgl/src/gpgpu_context.ts:331–348

(fragmentShader: WebGLShader)

  }

  public createProgram(fragmentShader: WebGLShader): GPGPUContextProgram {
    this.throwIfDisposed();
    const gl = this.gl;
    if (this.vertexShader == null) {
      this.vertexShader = gpgpu_util.createVertexShader(gl);
    }
    const program: WebGLProgram = webgl_util.createProgram(gl);
    webgl_util.callAndCheck(
        gl, () => gl.attachShader(program, this.vertexShader));
    webgl_util.callAndCheck(gl, () => gl.attachShader(program, fragmentShader));
    webgl_util.linkProgram(gl, program);

    const program2 = Object.assign(program, {vao: this.createVertexArray()});
    if (this.debug) {
      webgl_util.validateProgram(gl, program2);
    }
    return program2;
  }

  public buildVao(program: GPGPUContextProgram) {
    this.setProgram(program);