printf("\033[2J\033[1;1H");

printf("\nHello float16!\n");

printf("--------------\n\n");

Context ctx = createContext(

{}, {},

/*device descriptor, enabling f16 in WGSL*/

{

.requiredFeatureCount = 1,

.requiredFeatures = std::array{WGPUFeatureName_ShaderF16}.data(),

});

static constexpr size_t N = 10000;

std::array<half, N> inputArr, outputArr;

for (int i = 0; i < N; ++i) {

inputArr[i] = half(static_cast<float>(i) / 10.0f); // dummy input data

}

Tensor input = createTensor(ctx, Shape{N}, kf16, inputArr.data());

Tensor output = createTensor(ctx, Shape{N}, kf16);

std::promise<void> promise;

std::future<void> future = promise.get_future();

Kernel op = createKernel(ctx, {kGelu, 256, kf16}, Bindings{input, output},

{cdiv(N, 256), 1, 1});

dispatchKernel(ctx, op, promise);

wait(ctx, future);

toCPU(ctx, output, outputArr.data(), sizeof(outputArr));

for (int i = 0; i < 12; ++i) {

// Cast to float32 for printing to the screen

printf(" gelu(%.2f) = %.2f\n", static_cast<float>(inputArr[i]),

static_cast<float>(outputArr[i]));

}

printf(" ...\n\n");

printf("Computed %zu float16 values of GELU(x: float16)\n\n", N);

return 0;