feat: Add a benchmark suite for vector creation (#1063)

Author: reczkok

apps/typegpu-docs/src/pages/benchmark/suites.ts

@@ -5,6 +5,7 @@ import type { TypeGPUDataModule, TypeGPUModule } from './modules';
 import type { BenchParameterSet } from './parameter-set';
 import { compiledWriteSuite } from './test-suites/compiled-write';
 import { partialWriteSuite } from './test-suites/partial-write';
+import { vectorCreationSuite } from './test-suites/vector-creation';
 
 export type TestIdentifier = `${string}_${string}`;
 
@@ -39,6 +40,7 @@ export function createSuite<T extends { bench: Bench }>(
 export const unfilteredSuites: Record<string, Suite> = {
   'Partial write': partialWriteSuite,
   'Compiled write': compiledWriteSuite,
+  'Vector creation': vectorCreationSuite,
 };
 
 export function getFilteredSuites(selectedTests: TestIdentifier[]) {

apps/typegpu-docs/src/pages/benchmark/test-suites/vector-creation.ts

@@ -0,0 +1,77 @@
+import { Bench } from 'tinybench';
+import type { v2u, v3f, v4i } from 'typegpu/data';
+import { stringifyLocator } from '../parameter-set';
+import { createSuite } from '../suites';
+
+export const vectorCreationSuite = createSuite(
+  ({ params, d }) => {
+    const ctx = {
+      bench: null as unknown as Bench,
+      d,
+      thousandVectors: [] as (v3f | v2u | v4i)[],
+      tenThousandVectors: [] as (v3f | v2u | v4i)[],
+      hundredThousandVectors: [] as (v3f | v2u | v4i)[],
+    };
+
+    ctx.bench = new Bench({
+      name: stringifyLocator('typegpu', params.typegpu),
+      time: 1000,
+      setup: () => {
+        ctx.thousandVectors = Array.from({ length: 1000 });
+        ctx.tenThousandVectors = Array.from({ length: 10000 });
+        ctx.hundredThousandVectors = Array.from({ length: 100000 });
+      },
+    });
+
+    return ctx;
+  },
+  {
+    '1k vectors': (getCtx) => async () => {
+      const { thousandVectors: container, d } = getCtx();
+
+      for (let i = 0; i < container.length; i++) {
+        container[i] = d.vec3f(1, 2, 3);
+      }
+
+      for (let i = 0; i < container.length; i++) {
+        container[i] = d.vec2u(1, 2);
+      }
+
+      for (let i = 0; i < container.length; i++) {
+        container[i] = d.vec4i(1, 2, 3, 4);
+      }
+    },
+
+    '10k vectors': (getCtx) => async () => {
+      const { tenThousandVectors: container, d } = getCtx();
+
+      for (let i = 0; i < container.length; i++) {
+        container[i] = d.vec3f(1, 2, 3);
+      }
+
+      for (let i = 0; i < container.length; i++) {
+        container[i] = d.vec2u(1, 2);
+      }
+
+      for (let i = 0; i < container.length; i++) {
+        container[i] = d.vec4i(1, 2, 3, 4);
+      }
+    },
+
+    '100k vectors': (getCtx) => async () => {
+      const { hundredThousandVectors: container, d } = getCtx();
+
+      for (let i = 0; i < container.length; i++) {
+        container[i] = d.vec3f(1, 2, 3);
+      }
+
+      for (let i = 0; i < container.length; i++) {
+        container[i] = d.vec2u(1, 2);
+      }
+
+      for (let i = 0; i < container.length; i++) {
+        container[i] = d.vec4i(1, 2, 3, 4);
+      }
+    },
+  },
+);

packages/typegpu/tsup.config.ts

@@ -10,7 +10,7 @@ export default defineConfig({
   entry,
   outDir: 'dist',
   format: ['cjs', 'esm'],
-  target: 'es2017',
+  target: 'es2024',
   splitting: true,
   sourcemap: true,
   minify: !inDevMode,