46 lines · plain
1// RUN: %clang_cc1 -std=hlsl2021 -finclude-default-header -triple \2// RUN: dxil-pc-shadermodel6.3-compute %s -emit-llvm -disable-llvm-passes -o - | \3// RUN: FileCheck %s --check-prefixes=CHECK,CHECK-DXIL4// RUN: %clang_cc1 -std=hlsl2021 -finclude-default-header -triple \5// RUN: spirv-pc-vulkan-compute %s -emit-llvm -disable-llvm-passes -o - | \6// RUN: FileCheck %s --check-prefixes=CHECK,CHECK-SPIRV7 8// Test basic lowering to runtime function call.9 10// CHECK-LABEL: test_int11int test_int(int expr) {12 // CHECK-SPIRV: %[[RET:.*]] = call spir_func [[TY:.*]] @llvm.spv.wave.reduce.sum.i32([[TY]] %[[#]])13 // CHECK-DXIL: %[[RET:.*]] = call [[TY:.*]] @llvm.dx.wave.reduce.sum.i32([[TY]] %[[#]])14 // CHECK: ret [[TY]] %[[RET]]15 return WaveActiveSum(expr);16}17 18// CHECK-DXIL: declare [[TY]] @llvm.dx.wave.reduce.sum.i32([[TY]]) #[[#attr:]]19// CHECK-SPIRV: declare [[TY]] @llvm.spv.wave.reduce.sum.i32([[TY]]) #[[#attr:]]20 21// CHECK-LABEL: test_uint64_t22uint64_t test_uint64_t(uint64_t expr) {23 // CHECK-SPIRV: %[[RET:.*]] = call spir_func [[TY:.*]] @llvm.spv.wave.reduce.sum.i64([[TY]] %[[#]])24 // CHECK-DXIL: %[[RET:.*]] = call [[TY:.*]] @llvm.dx.wave.reduce.usum.i64([[TY]] %[[#]])25 // CHECK: ret [[TY]] %[[RET]]26 return WaveActiveSum(expr);27}28 29// CHECK-DXIL: declare [[TY]] @llvm.dx.wave.reduce.usum.i64([[TY]]) #[[#attr:]]30// CHECK-SPIRV: declare [[TY]] @llvm.spv.wave.reduce.sum.i64([[TY]]) #[[#attr:]]31 32// Test basic lowering to runtime function call with array and float value.33 34// CHECK-LABEL: test_floatv435float4 test_floatv4(float4 expr) {36 // CHECK-SPIRV: %[[RET1:.*]] = call reassoc nnan ninf nsz arcp afn spir_func [[TY1:.*]] @llvm.spv.wave.reduce.sum.v4f32([[TY1]] %[[#]]37 // CHECK-DXIL: %[[RET1:.*]] = call reassoc nnan ninf nsz arcp afn [[TY1:.*]] @llvm.dx.wave.reduce.sum.v4f32([[TY1]] %[[#]])38 // CHECK: ret [[TY1]] %[[RET1]]39 return WaveActiveSum(expr);40}41 42// CHECK-DXIL: declare [[TY1]] @llvm.dx.wave.reduce.sum.v4f32([[TY1]]) #[[#attr]]43// CHECK-SPIRV: declare [[TY1]] @llvm.spv.wave.reduce.sum.v4f32([[TY1]]) #[[#attr]]44 45// CHECK: attributes #[[#attr]] = {{{.*}} convergent {{.*}}}46