/ vulkan-shaders / relu.comp
relu.comp
1 #version 450 2 3 #include "generic_head.comp" 4 #include "types.comp" 5 6 #extension GL_EXT_control_flow_attributes : enable 7 8 layout(local_size_x = 512, local_size_y = 1, local_size_z = 1) in; 9 10 layout (binding = 0) readonly buffer X {A_TYPE data_a[];}; 11 layout (binding = 1) writeonly buffer D {D_TYPE data_d[];}; 12 13 void main() { 14 const uint i = gl_GlobalInvocationID.x; 15 16 if (i >= p.KX) { 17 return; 18 } 19 20 data_d[i] = max(float(data_a[i]), 0); 21 }