matmul_bench.ts

import { ones } from "./index";

const t1 = ones([100, 100]);
const t2 = ones([100, 100]);

let count = 10;
let total = 0;
const begin = Date.now() / 1000;
let end = begin;

for (let i = 0; i < 20; i++) {
  const start = end;

  for (let i = 0; i < count; i++) {
    const t3 = t1.matMul(t2);
  }

  end = Date.now() / 1000;
  const elapsed = end - start;
  const throughput = count / elapsed;

  console.log(
    `time: ${elapsed}s  count: ${count}  throughput: ${throughput}/s`);

  total += count;
  count = Math.round(total / (end - begin));
}