Skip to main content

SCALE

scry_gpu::shaders::backward

Constant SCALE 

Source 
pub const SCALE: &str = "\
struct Dims { N: u32, alpha: f32 }
var<push_constant> dims: Dims;

@group(0) @binding(0) var<storage, read> input: array<f32>;
@group(0) @binding(1) var<storage, read_write> out: array<f32>;

@compute @workgroup_size(256)
fn main(@builtin(global_invocation_id) gid: vec3<u32>) {
    let i = gid.x;
    if i >= dims.N { return; }
    out[i] = input[i] * dims.alpha;
}";
Expand description

Element-wise scale: out[i] = in[i] * alpha.

Push constants: struct Dims { N: u32, alpha: f32 } (8 bytes) Workgroup size: 256 — dispatch N invocations Bindings:

  • @binding(0) input: array<f32> (read)
  • @binding(1) out: array<f32> (read_write)