IQ.Pilot Release Commit @ bec7652

artifacts/package_sources/tinygrad/test/web/test_webgpu.js

@@ -0,0 +1,65 @@
+const puppeteer = require("puppeteer");
+const { spawn } = require("child_process");
+const res = spawn("python", ["-m", "http.server", "8000"], { shell: true });
+
+async function timeout(time) {
+  return new Promise((resolve) => setTimeout(resolve, time));
+}
+
+function cleanup(err) {
+  console.log("cleaning up");
+  res.kill();
+  if (err != null) {
+    console.error(err);
+    process.exit(1);
+  }
+  process.exit(0);
+}
+
+async function waitForText(selector, text) {
+  let n = 0;
+  let ready = false;
+  while (n < 30) {
+    const res = await (await selector.getProperty("textContent")).jsonValue();
+    console.log(`waiting for text ${text} got ${res}`);
+    if (res == text) {
+      ready = true;
+      break;
+    }
+    await timeout(1000);
+    n += 1;
+  }
+  return ready;
+}
+
+async function runTest() {
+  const browser = await puppeteer.launch({
+    headless: false,
+    args: ["--enable-unsafe-webgpu"],
+  });
+  const page = await browser.newPage();
+
+  page
+    .on("console", (message) =>
+      console.log(`message from console ${message.text()}`),
+    )
+    .on("pageerror", ({ message }) =>
+      console.log(`error from page ${message}`),
+    );
+
+  const res = await page.goto("http://localhost:8000/examples/webgpu/efficientnet/index.html");
+  if (res.status() !== 200) throw new Error("Failed to load page");
+
+  const textSelector = await page.waitForSelector("#result");
+  const buttonSelector = await page.waitForSelector("input[type=button]");
+  const ready = await waitForText(textSelector, "ready");
+  if (!ready) throw new Error("Failed to load page");
+
+  await buttonSelector.evaluate((e) => e.click());
+  const done = await waitForText(textSelector, "hen");
+  if (!done) throw new Error("failed to get hen");
+
+  cleanup(null);
+}
+
+runTest().catch((err) => cleanup(err));