Error: Kernel 'Draw' not registered for backend 'custom-webgl2' at Engine.runKernel at Module.draw about tfjs

[Apj-infinty]

@Linchenn
#3457

[Linchenn]

Draw API currently does not have implementation for WebGL backend.

[Apj-infinty]

Thank you for your reply, I was able to complete the objective using WebGPU Kernel! It is almost 100x faster than tf.browser.toPixels() method. Not sure if WebGL will be faster than WebGPU although.

   //importing the WebGPUBackend from npm package @tensorflow/tfjs-backend-webgpu
   import {WebGPUBackend} from @tensorflow/tfjs-backend-webgpu


   //Register the Backend Kernel 
      if (!navigator.gpu) {
        throw new Error("WebGPU not supported on this browser.");
      }

      const customBackendName = 'custom-webgpu';
      const kernels = tf.getKernelsForBackend('webgpu');

      kernels.forEach(kernelConfig => {
        const newKernelConfig = {...kernelConfig, backendName: customBackendName};
        tf.registerKernel(newKernelConfig);
      });
      
      const adapter = await navigator.gpu.requestAdapter({powerPreference: 'low-power'});
      if (!adapter) {
        throw new Error("No appropriate GPUAdapter found.");
      }

      const device = await adapter.requestDevice();

      tf.registerBackend('custom-webgpu', () => new WebGPUBackend(device));

      await tf.setBackend('custom-webgpu')

      await tf.ready()
      const canvas = document.getElementById("imgCanvas2");
      const context =  canvas.getContext("webgpu");
      const canvasFormat = navigator.gpu.getPreferredCanvasFormat();
      context.configure({
        device: device,
        format: canvasFormat,
      });

    //calling Draw API:         
        tf.browser.draw(<input tensor>,canvas);

@Linchenn Shall I close the issue or keep it?

[Linchenn]

WebGPU is typically faster than WebGL, while WebGL has more device coverage.

[google-ml-butler]

Are you satisfied with the resolution of your issue?
Yes
No