#!amber
# Copyright (c) 2025 The Khronos Group Inc.
# Copyright (c) 2025 Valve Corporation.
#
# 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
#
#      http://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.

DEVICE_FEATURE Float16Int8Features.shaderFloat16
DEVICE_FEATURE shaderInt64
DEVICE_FEATURE Storage16BitFeatures.storageBuffer16BitAccess
DEVICE_FEATURE Storage16BitFeatures.uniformAndStorageBuffer16BitAccess

SHADER compute compute_shader SPIRV-ASM
; SPIR-V
; Version: 1.0
; Generator: Khronos Glslang Reference Front End; 11
; Bound: 64
; Schema: 0
               OpCapability Shader
               OpCapability Float16
               OpCapability Int64
               OpCapability StorageBuffer16BitAccess
               OpExtension "SPV_KHR_16bit_storage"
          %1 = OpExtInstImport "GLSL.std.450"
               OpMemoryModel Logical GLSL450
               OpEntryPoint GLCompute %main "main" %gl_LocalInvocationIndex
               OpExecutionMode %main LocalSize 64 1 1
               OpSource GLSL 460
               OpSourceExtension "GL_EXT_shader_explicit_arithmetic_types"
               OpName %main "main"
               OpName %idx "idx"
               OpName %gl_LocalInvocationIndex "gl_LocalInvocationIndex"
               OpName %PushConstantBlock "PushConstantBlock"
               OpMemberName %PushConstantBlock 0 "count"
               OpName %_ ""
               OpName %s "s"
               OpName %SignificandBlock "SignificandBlock"
               OpMemberName %SignificandBlock 0 "significands"
               OpName %__0 ""
               OpName %e "e"
               OpName %ExponentsBlock "ExponentsBlock"
               OpMemberName %ExponentsBlock 0 "exponents"
               OpName %__1 ""
               OpName %r "r"
               OpName %ResultsBlock "ResultsBlock"
               OpMemberName %ResultsBlock 0 "results"
               OpName %__2 ""
               OpDecorate %gl_LocalInvocationIndex BuiltIn LocalInvocationIndex
               OpDecorate %PushConstantBlock Block
               OpMemberDecorate %PushConstantBlock 0 Offset 0
               OpDecorate %_runtimearr_v2half ArrayStride 4
               OpDecorate %SignificandBlock BufferBlock
               OpMemberDecorate %SignificandBlock 0 NonWritable
               OpMemberDecorate %SignificandBlock 0 Offset 0
               OpDecorate %__0 NonWritable
               OpDecorate %__0 Binding 0
               OpDecorate %__0 DescriptorSet 0
               OpDecorate %_runtimearr_v2long ArrayStride 16
               OpDecorate %ExponentsBlock BufferBlock
               OpMemberDecorate %ExponentsBlock 0 NonWritable
               OpMemberDecorate %ExponentsBlock 0 Offset 0
               OpDecorate %__1 NonWritable
               OpDecorate %__1 Binding 1
               OpDecorate %__1 DescriptorSet 0
               OpDecorate %_runtimearr_v2half_0 ArrayStride 4
               OpDecorate %ResultsBlock BufferBlock
               OpMemberDecorate %ResultsBlock 0 Offset 0
               OpDecorate %__2 Binding 2
               OpDecorate %__2 DescriptorSet 0
               OpDecorate %gl_WorkGroupSize BuiltIn WorkgroupSize
       %void = OpTypeVoid
          %3 = OpTypeFunction %void
       %uint = OpTypeInt 32 0
%_ptr_Function_uint = OpTypePointer Function %uint
%_ptr_Input_uint = OpTypePointer Input %uint
%gl_LocalInvocationIndex = OpVariable %_ptr_Input_uint Input
%PushConstantBlock = OpTypeStruct %uint
%_ptr_PushConstant_PushConstantBlock = OpTypePointer PushConstant %PushConstantBlock
          %_ = OpVariable %_ptr_PushConstant_PushConstantBlock PushConstant
        %int = OpTypeInt 32 1
      %int_0 = OpConstant %int 0
%_ptr_PushConstant_uint = OpTypePointer PushConstant %uint
       %bool = OpTypeBool
       %half = OpTypeFloat 16
     %v2half = OpTypeVector %half 2
%_ptr_Function_v2half = OpTypePointer Function %v2half
%_runtimearr_v2half = OpTypeRuntimeArray %v2half
%SignificandBlock = OpTypeStruct %_runtimearr_v2half
%_ptr_Uniform_SignificandBlock = OpTypePointer Uniform %SignificandBlock
        %__0 = OpVariable %_ptr_Uniform_SignificandBlock Uniform
%_ptr_Uniform_v2half = OpTypePointer Uniform %v2half
      %long = OpTypeInt 64 1
    %v2long = OpTypeVector %long 2
%_ptr_Function_v2long = OpTypePointer Function %v2long
%_runtimearr_v2long = OpTypeRuntimeArray %v2long
%ExponentsBlock = OpTypeStruct %_runtimearr_v2long
%_ptr_Uniform_ExponentsBlock = OpTypePointer Uniform %ExponentsBlock
        %__1 = OpVariable %_ptr_Uniform_ExponentsBlock Uniform
%_ptr_Uniform_v2long = OpTypePointer Uniform %v2long
%_runtimearr_v2half_0 = OpTypeRuntimeArray %v2half
%ResultsBlock = OpTypeStruct %_runtimearr_v2half_0
%_ptr_Uniform_ResultsBlock = OpTypePointer Uniform %ResultsBlock
        %__2 = OpVariable %_ptr_Uniform_ResultsBlock Uniform
     %v3uint = OpTypeVector %uint 3
    %uint_64 = OpConstant %uint 64
     %uint_1 = OpConstant %uint 1
%gl_WorkGroupSize = OpConstantComposite %v3uint %uint_64 %uint_1 %uint_1
       %main = OpFunction %void None %3
          %5 = OpLabel
        %idx = OpVariable %_ptr_Function_uint Function
          %s = OpVariable %_ptr_Function_v2half Function
          %e = OpVariable %_ptr_Function_v2long Function
          %r = OpVariable %_ptr_Function_v2half Function
         %11 = OpLoad %uint %gl_LocalInvocationIndex
               OpStore %idx %11
         %12 = OpLoad %uint %idx
         %19 = OpAccessChain %_ptr_PushConstant_uint %_ %int_0
         %20 = OpLoad %uint %19
         %22 = OpULessThan %bool %12 %20
               OpSelectionMerge %24 None
               OpBranchConditional %22 %23 %24
         %23 = OpLabel
         %33 = OpLoad %uint %idx
         %35 = OpAccessChain %_ptr_Uniform_v2half %__0 %int_0 %33
         %36 = OpLoad %v2half %35
               OpStore %s %36
         %45 = OpLoad %uint %idx
         %47 = OpAccessChain %_ptr_Uniform_v2long %__1 %int_0 %45
         %48 = OpLoad %v2long %47
               OpStore %e %48
         %50 = OpLoad %v2half %s
         %51 = OpLoad %v2long %e
         %52 = OpExtInst %v2half %1 Ldexp %50 %51
               OpStore %r %52
         %57 = OpLoad %uint %idx
         %58 = OpLoad %v2half %r
         %59 = OpAccessChain %_ptr_Uniform_v2half %__2 %int_0 %57
               OpStore %59 %58
               OpBranch %24
         %24 = OpLabel
               OpReturn
               OpFunctionEnd

END

BUFFER significands DATA_TYPE float16 DATA
-1.0
-1.0
-1.0
-1.0
-1.0
-1.0
-1.0
-1.0
0.5
0.5
0.5
0.5
0.5
0.5
0.5
0.5
1.25
1.25
1.25
1.25
1.25
1.25
1.25
1.25
2.0
2.0
2.0
2.0
2.0
2.0
2.0
2.0
END

BUFFER exponents DATA_TYPE int64 DATA
1
0
-1
-2
-9223372036854775802
-9223372036854775803
-9223372036854775804
-9223372036854775805
1
0
-1
-2
-9223372036854775802
-9223372036854775803
-9223372036854775804
-9223372036854775805
1
0
-1
-2
-9223372036854775802
-9223372036854775803
-9223372036854775804
-9223372036854775805
1
0
-1
-2
-9223372036854775802
-9223372036854775803
-9223372036854775804
-9223372036854775805
END

BUFFER results DATA_TYPE float16 DATA
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
0.0
END

BUFFER pc DATA_TYPE uint32 DATA
16
END

PIPELINE compute pipeline
  ATTACH compute_shader

  BIND BUFFER significands AS storage DESCRIPTOR_SET 0 BINDING 0
  BIND BUFFER exponents AS storage DESCRIPTOR_SET 0 BINDING 1
  BIND BUFFER results AS storage DESCRIPTOR_SET 0 BINDING 2
  BIND BUFFER pc AS push_constant
END

RUN pipeline 1 1 1

EXPECT results IDX 0 TOLERANCE .0001 EQ -2.0
EXPECT results IDX 2 TOLERANCE .0001 EQ -1.0
EXPECT results IDX 4 TOLERANCE .0001 EQ -0.5
EXPECT results IDX 6 TOLERANCE .0001 EQ -0.25
EXPECT results IDX 8 TOLERANCE .0001 EQ -0.0
EXPECT results IDX 10 TOLERANCE .0001 EQ -0.0
EXPECT results IDX 12 TOLERANCE .0001 EQ -0.0
EXPECT results IDX 14 TOLERANCE .0001 EQ -0.0
EXPECT results IDX 16 TOLERANCE .0001 EQ 1.0
EXPECT results IDX 18 TOLERANCE .0001 EQ 0.5
EXPECT results IDX 20 TOLERANCE .0001 EQ 0.25
EXPECT results IDX 22 TOLERANCE .0001 EQ 0.125
EXPECT results IDX 24 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 26 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 28 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 30 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 32 TOLERANCE .0001 EQ 2.5
EXPECT results IDX 34 TOLERANCE .0001 EQ 1.25
EXPECT results IDX 36 TOLERANCE .0001 EQ 0.625
EXPECT results IDX 38 TOLERANCE .0001 EQ 0.3125
EXPECT results IDX 40 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 42 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 44 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 46 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 48 TOLERANCE .0001 EQ 4.0
EXPECT results IDX 50 TOLERANCE .0001 EQ 2.0
EXPECT results IDX 52 TOLERANCE .0001 EQ 1.0
EXPECT results IDX 54 TOLERANCE .0001 EQ 0.5
EXPECT results IDX 56 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 58 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 60 TOLERANCE .0001 EQ 0.0
EXPECT results IDX 62 TOLERANCE .0001 EQ 0.0