mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-06-06 06:33:30 +00:00
009d129103
Transform any SPIR-V that has an array with a custom stride:
@stride(S) array<T, N>
into:
struct strided_arr {
@size(S) er : T;
};
array<strided_arr, N>
Also remove any @stride decorations that match the default array stride.
Bug: tint:1394
Bug: tint:1381
Change-Id: I8be8f3a76c5335fdb2bc5183388366091dbc7642
Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/78781
Reviewed-by: David Neto <dneto@google.com>
Kokoro: Kokoro <noreply+kokoro@google.com>
Commit-Queue: Ben Clayton <bclayton@google.com>
80 lines
1.6 KiB
WebGPU Shading Language
80 lines
1.6 KiB
WebGPU Shading Language
type RTArr = array<f32>;
|
|
|
|
type RTArr_1 = array<f32>;
|
|
|
|
struct ssbOut {
|
|
result : RTArr_1;
|
|
}
|
|
|
|
struct ssbA {
|
|
A : RTArr_1;
|
|
}
|
|
|
|
struct Uniforms {
|
|
NAN : f32;
|
|
aShape : i32;
|
|
outShape : i32;
|
|
outShapeStrides : i32;
|
|
size : i32;
|
|
}
|
|
|
|
@group(0) @binding(0) var<storage, read_write> x_16 : ssbOut;
|
|
|
|
@group(0) @binding(1) var<storage, read> x_20 : ssbA;
|
|
|
|
var<private> gl_GlobalInvocationID : vec3<u32>;
|
|
|
|
@group(0) @binding(2) var<uniform> x_24 : Uniforms;
|
|
|
|
fn getAAtOutCoords_() -> f32 {
|
|
let x_42 : u32 = gl_GlobalInvocationID.x;
|
|
let x_44 : f32 = x_20.A[x_42];
|
|
return x_44;
|
|
}
|
|
|
|
fn unaryOperation_f1_(a : ptr<function, f32>) -> f32 {
|
|
let x_47 : f32 = *(a);
|
|
if ((x_47 < 0.0)) {
|
|
return 0x1p+128;
|
|
}
|
|
let x_55 : f32 = *(a);
|
|
return log(x_55);
|
|
}
|
|
|
|
fn setOutput_i1_f1_(flatIndex : ptr<function, i32>, value : ptr<function, f32>) {
|
|
let x_27 : i32 = *(flatIndex);
|
|
let x_28 : f32 = *(value);
|
|
x_16.result[x_27] = x_28;
|
|
return;
|
|
}
|
|
|
|
fn main_1() {
|
|
var index : i32;
|
|
var a_1 : f32;
|
|
var param : f32;
|
|
var param_1 : i32;
|
|
var param_2 : f32;
|
|
let x_61 : u32 = gl_GlobalInvocationID.x;
|
|
index = bitcast<i32>(x_61);
|
|
let x_63 : i32 = index;
|
|
let x_70 : i32 = x_24.size;
|
|
if ((x_63 < x_70)) {
|
|
let x_75 : f32 = getAAtOutCoords_();
|
|
a_1 = x_75;
|
|
let x_77 : f32 = a_1;
|
|
param = x_77;
|
|
let x_78 : f32 = unaryOperation_f1_(&(param));
|
|
let x_80 : i32 = index;
|
|
param_1 = x_80;
|
|
param_2 = x_78;
|
|
setOutput_i1_f1_(&(param_1), &(param_2));
|
|
}
|
|
return;
|
|
}
|
|
|
|
@stage(compute) @workgroup_size(128, 1, 1)
|
|
fn main(@builtin(global_invocation_id) gl_GlobalInvocationID_param : vec3<u32>) {
|
|
gl_GlobalInvocationID = gl_GlobalInvocationID_param;
|
|
main_1();
|
|
}
|