Skip to content

Commit

Permalink
Implement CTS requestAdapter test
Browse files Browse the repository at this point in the history
  • Loading branch information
greggman committed Dec 15, 2022
1 parent e583fa4 commit 5a5838d
Showing 1 changed file with 123 additions and 0 deletions.
123 changes: 123 additions & 0 deletions src/webgpu/api/operation/adapter/requestAdapter.spec.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,123 @@
export const description = `
Tests for GPU.requestAdapter.
Test all possible options to requestAdapter.
default, low-power, and high performance should all always return adapters.
forceFallbackAdapter may or may not return an adapter.
GPU.requestAdapter can technically return null for any reason
but we need test functionality so the test requires an adapter except
when forceFallbackAdapter is true.
The test runs simple compute shader is run that fills a buffer with consecutive
values and then checks the result to test the adapter for basic functionality.
`;

import { Fixture } from '../../../../common/framework/fixture.js';
import { makeTestGroup } from '../../../../common/framework/test_group.js';
import { getGPU } from '../../../../common/util/navigator_gpu.js';
import { assert, objectEquals, iterRange } from '../../../../common/util/util.js';

export const g = makeTestGroup(Fixture);

const powerPreferenceModes: Array<GPUPowerPreference | undefined> = [
undefined,
'low-power',
'high-performance',
];
const forceFallbackOptions: Array<boolean | undefined> = [undefined, false, true];

async function testAdapter(adapter: GPUAdapter | null) {
assert(adapter !== null, 'Failed to get adapter.');
const device = await adapter.requestDevice();

assert(device !== null, 'Failed to get device.');

const kOffset = 1230000;
const pipeline = await device.createComputePipeline({
compute: {
module: device.createShaderModule({
code: `
struct Buffer { data: array<u32>; };
@group(0) @binding(0) var<storage, read_write> buffer: Buffer;
@stage(compute) @workgroup_size(1u) fn main(
@builtin(global_invocation_id) id: vec3<u32>) {
buffer.data[id.x] = id.x + ${kOffset}u;
}
`,
}),
entryPoint: 'main',
},
});

const kNumElements = 64;
const kBufferSize = kNumElements * 4;
const buffer = device.createBuffer({
size: kBufferSize,
usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC,
});

const resultBuffer = device.createBuffer({
size: kBufferSize,
usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST,
});

const bindGroup = device.createBindGroup({
layout: pipeline.getBindGroupLayout(0),
entries: [{ binding: 0, resource: { buffer } }],
});

const encoder = device.createCommandEncoder();

const pass = encoder.beginComputePass();
pass.setPipeline(pipeline);
pass.setBindGroup(0, bindGroup);
pass.dispatch(kNumElements);
pass.end();

encoder.copyBufferToBuffer(buffer, 0, resultBuffer, 0, kBufferSize);

device.queue.submit([encoder.finish()]);

const expected = new Uint32Array([...iterRange(kNumElements, x => x + kOffset)]);

await resultBuffer.mapAsync(GPUMapMode.READ);
const actual = new Uint32Array(resultBuffer.getMappedRange());

assert(objectEquals(actual, expected), 'compute pipeline ran');

resultBuffer.destroy();
buffer.destroy();
device.destroy();
}

g.test('requestAdapter')
.desc(`request adapter with all possible options and check for basic functionality`)
.params(u =>
u
.combine('powerPreference', powerPreferenceModes)
.combine('forceFallbackAdapter', forceFallbackOptions)
)
.fn(async t => {
const { powerPreference, forceFallbackAdapter } = t.params;
const adapter = await getGPU().requestAdapter({
...(powerPreference !== undefined && { powerPreference }),
...(forceFallbackAdapter !== undefined && { forceFallbackAdapter }),
});

// failing to create an adapter when forceFallbackAdapter is true is ok.
if (forceFallbackAdapter && !adapter) {
t.skip('No adapter available');
return;
}

await testAdapter(adapter);
});

g.test('requestAdapter_no_parameters')
.desc(`request adapter with no parameters`)
.fn(async () => {
const adapter = await getGPU().requestAdapter();
await testAdapter(adapter);
});

0 comments on commit 5a5838d

Please sign in to comment.