🏡 index : cold-air-networks/mirror-webgpu-cts.git

import { anyOf, skipUndefined } from '../../../../../util/compare.js';
import { f32, pack2x16float, u32, vec2 } from '../../../../../util/conversion.js';
import { cartesianProduct, quantizeToF32, scalarF32Range } from '../../../../../util/math.js';
import { Case } from '../../case.js';
import { makeCaseCache } from '../../case_cache.js';

// pack2x16float has somewhat unusual behaviour, specifically around how it is
// supposed to behave when values go OOB and when they are considered to have
// gone OOB, so has its own bespoke implementation.

/**
 * @returns a Case for `pack2x16float`
 * @param param0 first param for the case
 * @param param1 second param for the case
 * @param filter_undefined should inputs that cause an undefined expectation be
 *                         filtered out, needed for const-eval
 */
function makeCase(param0: number, param1: number, filter_undefined: boolean): Case | undefined {
  param0 = quantizeToF32(param0);
  param1 = quantizeToF32(param1);

  const results = pack2x16float(param0, param1);
  if (filter_undefined && results.some(r => r === undefined)) {
    return undefined;
  }

  return {
    input: [vec2(f32(param0), f32(param1))],
    expected: anyOf(
      ...results.map(r => (r === undefined ? skipUndefined(undefined) : skipUndefined(u32(r))))
    ),
  };
}

/**
 * @returns an array of Cases for `pack2x16float`
 * @param param0s array of inputs to try for the first param
 * @param param1s array of inputs to try for the second param
 * @param filter_undefined should inputs that cause an undefined expectation be
 *                         filtered out, needed for const-eval
 */
function generateCases(param0s: number[], param1s: number[], filter_undefined: boolean): Case[] {
  return cartesianProduct(param0s, param1s)
    .map(e => makeCase(e[0], e[1], filter_undefined))
    .filter((c): c is Case => c !== undefined);
}

export const d = makeCaseCache('pack2x16float', {
  f32_const: () => {
    return generateCases(scalarF32Range(), scalarF32Range(), true);
  },
  f32_non_const: () => {
    return generateCases(scalarF32Range(), scalarF32Range(), false);
  },
});