#!amber
# Copyright 2024 The Amber Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

SHADER compute atomic_sum_all GLSL
#version 430

layout(local_size_x = 16, local_size_y = 16, local_size_z = 1) in;

layout(set = 0, binding = 0) buffer BlockUint {
  uint data;
} ssbo_uint;

void main() {
    atomicAdd(ssbo_uint.data, uint(1));
}
END

BUFFER buf_uint DATA_TYPE uint32 DATA
0
END


PIPELINE compute pipeline
  ATTACH atomic_sum_all
  BIND BUFFER buf_uint AS storage DESCRIPTOR_SET 0 BINDING 0
END

RUN TIMED_EXECUTION pipeline 128 128 1

EXPECT buf_uint IDX 0 EQ 4194304