Split the test case

Author: Yang Gu

tfjs-backend-webgpu/src/backend_webgpu_test.ts

@@ -216,65 +216,67 @@ describeWebGPU('backend webgpu', () => {
     expect(() => c.dataSync()).not.toThrow();
   });
 
-  it('should reuse textures when fromPixels have same input size', async () => {
+  it('should behave well if WEBGPU_USER_IMPORT is true or false', async () => {
     const useImport = tf.env().getBool('WEBGPU_USE_IMPORT');
     const backend = tf.backend() as WebGPUBackend;
     const textureManager = backend.getTextureManager();
+    textureManager.dispose();
+
+    const video = document.createElement('video');
+    const source = document.createElement('source');
+    source.src =
+        // tslint:disable-next-line:max-line-length
+        'data:video/mp4;base64,AAAAIGZ0eXBpc29tAAACAGlzb21pc28yYXZjMW1wNDEAAAAIZnJlZQAAAu1tZGF0AAACrQYF//+p3EXpvebZSLeWLNgg2SPu73gyNjQgLSBjb3JlIDE1NSByMjkwMSA3ZDBmZjIyIC0gSC4yNjQvTVBFRy00IEFWQyBjb2RlYyAtIENvcHlsZWZ0IDIwMDMtMjAxOCAtIGh0dHA6Ly93d3cudmlkZW9sYW4ub3JnL3gyNjQuaHRtbCAtIG9wdGlvbnM6IGNhYmFjPTEgcmVmPTMgZGVibG9jaz0xOjA6MCBhbmFseXNlPTB4MzoweDExMyBtZT1oZXggc3VibWU9NyBwc3k9MSBwc3lfcmQ9MS4wMDowLjAwIG1peGVkX3JlZj0xIG1lX3JhbmdlPTE2IGNocm9tYV9tZT0xIHRyZWxsaXM9MSA4eDhkY3Q9MSBjcW09MCBkZWFkem9uZT0yMSwxMSBmYXN0X3Bza2lwPTEgY2hyb21hX3FwX29mZnNldD0tMiB0aHJlYWRzPTMgbG9va2FoZWFkX3RocmVhZHM9MSBzbGljZWRfdGhyZWFkcz0wIG5yPTAgZGVjaW1hdGU9MSBpbnRlcmxhY2VkPTAgYmx1cmF5X2NvbXBhdD0wIGNvbnN0cmFpbmVkX2ludHJhPTAgYmZyYW1lcz0zIGJfcHlyYW1pZD0yIGJfYWRhcHQ9MSBiX2JpYXM9MCBkaXJlY3Q9MSB3ZWlnaHRiPTEgb3Blbl9nb3A9MCB3ZWlnaHRwPTIga2V5aW50PTI1MCBrZXlpbnRfbWluPTEgc2NlbmVjdXQ9NDAgaW50cmFfcmVmcmVzaD0wIHJjX2xvb2thaGVhZD00MCByYz1jcmYgbWJ0cmVlPTEgY3JmPTI4LjAgcWNvbXA9MC42MCBxcG1pbj0wIHFwbWF4PTY5IHFwc3RlcD00IGlwX3JhdGlvPTEuNDAgYXE9MToxLjAwAIAAAAAwZYiEAD//8m+P5OXfBeLGOfKE3xkODvFZuBflHv/+VwJIta6cbpIo4ABLoKBaYTkTAAAC7m1vb3YAAABsbXZoZAAAAAAAAAAAAAAAAAAAA+gAAAPoAAEAAAEAAAAAAAAAAAAAAAABAAAAAAAAAAAAAAAAAAAAAQAAAAAAAAAAAAAAAAAAQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAIAAAIYdHJhawAAAFx0a2hkAAAAAwAAAAAAAAAAAAAAAQAAAAAAAAPoAAAAAAAAAAAAAAAAAAAAAAABAAAAAAAAAAAAAAAAAAAAAQAAAAAAAAAAAAAAAAAAQAAAAACgAAAAWgAAAAAAJGVkdHMAAAAcZWxzdAAAAAAAAAABAAAD6AAAAAAAAQAAAAABkG1kaWEAAAAgbWRoZAAAAAAAAAAAAAAAAAAAQAAAAEAAVcQAAAAAAC1oZGxyAAAAAAAAAAB2aWRlAAAAAAAAAAAAAAAAVmlkZW9IYW5kbGVyAAAAATttaW5mAAAAFHZtaGQAAAABAAAAAAAAAAAAAAAkZGluZgAAABxkcmVmAAAAAAAAAAEAAAAMdXJsIAAAAAEAAAD7c3RibAAAAJdzdHNkAAAAAAAAAAEAAACHYXZjMQAAAAAAAAABAAAAAAAAAAAAAAAAAAAAAACgAFoASAAAAEgAAAAAAAAAAQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABj//wAAADFhdmNDAWQACv/hABhnZAAKrNlCjfkhAAADAAEAAAMAAg8SJZYBAAZo6+JLIsAAAAAYc3R0cwAAAAAAAAABAAAAAQAAQAAAAAAcc3RzYwAAAAAAAAABAAAAAQAAAAEAAAABAAAAFHN0c3oAAAAAAAAC5QAAAAEAAAAUc3RjbwAAAAAAAAABAAAAMAAAAGJ1ZHRhAAAAWm1ldGEAAAAAAAAAIWhkbHIAAAAAAAAAAG1kaXJhcHBsAAAAAAAAAAAAAAAALWlsc3QAAAAlqXRvbwAAAB1kYXRhAAAAAQAAAABMYXZmNTguMTIuMTAw';
+    source.type = 'video/mp4';
+    video.appendChild(source);
+    document.body.appendChild(video);
+
+    video.autoplay = true;
+    video.loop = true;
+    video.muted = true;
+    video.preload = 'auto';
+    await video.play();
+
+    // ensure video element to be loaded
+    if ('requestVideoFrameCallback' in video) {
+      // tslint:disable-next-line:no-any
+      await new Promise(go => (video as any).requestVideoFrameCallback(go));
+    }
 
     {
-      const video = document.createElement('video');
-      const source = document.createElement('source');
-      source.src =
-          // tslint:disable-next-line:max-line-length
-          'data:video/mp4;base64,AAAAIGZ0eXBpc29tAAACAGlzb21pc28yYXZjMW1wNDEAAAAIZnJlZQAAAu1tZGF0AAACrQYF//+p3EXpvebZSLeWLNgg2SPu73gyNjQgLSBjb3JlIDE1NSByMjkwMSA3ZDBmZjIyIC0gSC4yNjQvTVBFRy00IEFWQyBjb2RlYyAtIENvcHlsZWZ0IDIwMDMtMjAxOCAtIGh0dHA6Ly93d3cudmlkZW9sYW4ub3JnL3gyNjQuaHRtbCAtIG9wdGlvbnM6IGNhYmFjPTEgcmVmPTMgZGVibG9jaz0xOjA6MCBhbmFseXNlPTB4MzoweDExMyBtZT1oZXggc3VibWU9NyBwc3k9MSBwc3lfcmQ9MS4wMDowLjAwIG1peGVkX3JlZj0xIG1lX3JhbmdlPTE2IGNocm9tYV9tZT0xIHRyZWxsaXM9MSA4eDhkY3Q9MSBjcW09MCBkZWFkem9uZT0yMSwxMSBmYXN0X3Bza2lwPTEgY2hyb21hX3FwX29mZnNldD0tMiB0aHJlYWRzPTMgbG9va2FoZWFkX3RocmVhZHM9MSBzbGljZWRfdGhyZWFkcz0wIG5yPTAgZGVjaW1hdGU9MSBpbnRlcmxhY2VkPTAgYmx1cmF5X2NvbXBhdD0wIGNvbnN0cmFpbmVkX2ludHJhPTAgYmZyYW1lcz0zIGJfcHlyYW1pZD0yIGJfYWRhcHQ9MSBiX2JpYXM9MCBkaXJlY3Q9MSB3ZWlnaHRiPTEgb3Blbl9nb3A9MCB3ZWlnaHRwPTIga2V5aW50PTI1MCBrZXlpbnRfbWluPTEgc2NlbmVjdXQ9NDAgaW50cmFfcmVmcmVzaD0wIHJjX2xvb2thaGVhZD00MCByYz1jcmYgbWJ0cmVlPTEgY3JmPTI4LjAgcWNvbXA9MC42MCBxcG1pbj0wIHFwbWF4PTY5IHFwc3RlcD00IGlwX3JhdGlvPTEuNDAgYXE9MToxLjAwAIAAAAAwZYiEAD//8m+P5OXfBeLGOfKE3xkODvFZuBflHv/+VwJIta6cbpIo4ABLoKBaYTkTAAAC7m1vb3YAAABsbXZoZAAAAAAAAAAAAAAAAAAAA+gAAAPoAAEAAAEAAAAAAAAAAAAAAAABAAAAAAAAAAAAAAAAAAAAAQAAAAAAAAAAAAAAAAAAQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAIAAAIYdHJhawAAAFx0a2hkAAAAAwAAAAAAAAAAAAAAAQAAAAAAAAPoAAAAAAAAAAAAAAAAAAAAAAABAAAAAAAAAAAAAAAAAAAAAQAAAAAAAAAAAAAAAAAAQAAAAACgAAAAWgAAAAAAJGVkdHMAAAAcZWxzdAAAAAAAAAABAAAD6AAAAAAAAQAAAAABkG1kaWEAAAAgbWRoZAAAAAAAAAAAAAAAAAAAQAAAAEAAVcQAAAAAAC1oZGxyAAAAAAAAAAB2aWRlAAAAAAAAAAAAAAAAVmlkZW9IYW5kbGVyAAAAATttaW5mAAAAFHZtaGQAAAABAAAAAAAAAAAAAAAkZGluZgAAABxkcmVmAAAAAAAAAAEAAAAMdXJsIAAAAAEAAAD7c3RibAAAAJdzdHNkAAAAAAAAAAEAAACHYXZjMQAAAAAAAAABAAAAAAAAAAAAAAAAAAAAAACgAFoASAAAAEgAAAAAAAAAAQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABj//wAAADFhdmNDAWQACv/hABhnZAAKrNlCjfkhAAADAAEAAAMAAg8SJZYBAAZo6+JLIsAAAAAYc3R0cwAAAAAAAAABAAAAAQAAQAAAAAAcc3RzYwAAAAAAAAABAAAAAQAAAAEAAAABAAAAFHN0c3oAAAAAAAAC5QAAAAEAAAAUc3RjbwAAAAAAAAABAAAAMAAAAGJ1ZHRhAAAAWm1ldGEAAAAAAAAAIWhkbHIAAAAAAAAAAG1kaXJhcHBsAAAAAAAAAAAAAAAALWlsc3QAAAAlqXRvbwAAAB1kYXRhAAAAAQAAAABMYXZmNTguMTIuMTAw';
-      source.type = 'video/mp4';
-      video.appendChild(source);
-      document.body.appendChild(video);
-
-      video.autoplay = true;
-      video.loop = true;
-      video.muted = true;
-      video.preload = 'auto';
-      await video.play();
-
-      // ensure video element to be loaded
-      if ('requestVideoFrameCallback' in video) {
-        // tslint:disable-next-line:no-any
-        await new Promise(go => (video as any).requestVideoFrameCallback(go));
-      }
-
-      {
-        textureManager.dispose();
-        tf.env().set('WEBGPU_USE_IMPORT', true);
-        const res = await tf.browser.fromPixelsAsync(video);
-        expect(res.shape).toEqual([90, 160, 3]);
-        const data = await res.data();
-        expect(data.length).toEqual(90 * 160 * 3);
-        const freeTexturesAfterFromPixels = textureManager.getNumFreeTextures();
-        expect(freeTexturesAfterFromPixels).toEqual(0);
-        const usedTexturesAfterFromPixels = textureManager.getNumUsedTextures();
-        expect(usedTexturesAfterFromPixels).toEqual(0);
-      }
-
-      {
-        textureManager.dispose();
-        tf.env().set('WEBGPU_USE_IMPORT', false);
-        const res = await tf.browser.fromPixelsAsync(video);
-        expect(res.shape).toEqual([90, 160, 3]);
-        const data = await res.data();
-        expect(data.length).toEqual(90 * 160 * 3);
-        const freeTexturesAfterFromPixels = textureManager.getNumFreeTextures();
-        expect(freeTexturesAfterFromPixels).toEqual(1);
-        const usedTexturesAfterFromPixels = textureManager.getNumUsedTextures();
-        expect(usedTexturesAfterFromPixels).toEqual(0);
-      }
-
-      document.body.removeChild(video);
-      tf.env().set('WEBGPU_USE_IMPORT', useImport);
+      tf.env().set('WEBGPU_USE_IMPORT', true);
+      const res = await tf.browser.fromPixelsAsync(video);
+      expect(res.shape).toEqual([90, 160, 3]);
+      const data = await res.data();
+      expect(data.length).toEqual(90 * 160 * 3);
+      const freeTexturesAfterFromPixels = textureManager.getNumFreeTextures();
+      expect(freeTexturesAfterFromPixels).toEqual(0);
+      const usedTexturesAfterFromPixels = textureManager.getNumUsedTextures();
+      expect(usedTexturesAfterFromPixels).toEqual(0);
+    }
+
+    {
+      tf.env().set('WEBGPU_USE_IMPORT', false);
+      const res = await tf.browser.fromPixelsAsync(video);
+      expect(res.shape).toEqual([90, 160, 3]);
+      const data = await res.data();
+      expect(data.length).toEqual(90 * 160 * 3);
+      const freeTexturesAfterFromPixels = textureManager.getNumFreeTextures();
+      expect(freeTexturesAfterFromPixels).toEqual(1);
+      const usedTexturesAfterFromPixels = textureManager.getNumUsedTextures();
+      expect(usedTexturesAfterFromPixels).toEqual(0);
     }
 
+    document.body.removeChild(video);
+    tf.env().set('WEBGPU_USE_IMPORT', useImport);
+  });
+
+  it('should reuse textures when fromPixels have same input size', async () => {
+    const backend = tf.backend() as WebGPUBackend;
+    const textureManager = backend.getTextureManager();
+    textureManager.dispose();
+
     {
-      textureManager.dispose();
       const img = new Image(10, 10);
       img.src = 'data:image/gif;base64' +
           ',R0lGODlhAQABAIAAAP///wAAACH5BAEAAAAALAAAAAABAAEAAAICRAEAOw==';
@@ -296,7 +298,6 @@ describeWebGPU('backend webgpu', () => {
     }
 
     {
-      textureManager.dispose();
       const img = new Image(10, 10);
       img.src =
           'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAIAAAACUFjqAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAAJcEhZcwAAEnQAABJ0Ad5mH3gAAABfSURBVChTY/gPBu8NLd/KqLxT1oZw4QAqDZSDoPeWDj9WrYUIAgG6NBAhm4FFGoIgxuCUBiKgMfikv1bW4pQGav334wdUGshBk/6SVQAUh0p/mzIDTQ6oFSGNHfz/DwAwi8mNzTi6rwAAAABJRU5ErkJggg==';