@hoff97/tensor-js/dist/lib/tensor/gpu/gl.js

PyTorch like deep learning inferrence library

import REGL from 'regl';
import { Dict } from '../../util/datastructs/dict';
import { GPUMemoryAllocator } from './memory';
const canvas = document.createElement('canvas');
export let glContext;
export let gl;
export let defaultAllocator;
function setup() {
    //@ts-ignore
    glContext = canvas.getContext('webgl', {
        failIfMajorPerformanceCaveat: false,
    });
    gl = REGL({
        gl: glContext,
        extensions: [
            'OES_texture_float',
            'WEBGL_color_buffer_float',
            'OES_texture_half_float',
        ],
    });
    defaultAllocator = new GPUMemoryAllocator(gl, () => {
        return new Dict((key) => key);
    });
}
setup();
//# sourceMappingURL=gl.js.map