9519d52512
This patch handles the special case where an extract value yields an aggregate result, which then is used as an argument to a store. The SPIRV BE uses special intrinsics (`spv_extractv` and `spv_store`) to represent these through IRTranslator, however this creates a problem: `spv_store` is called as a function, and IRTranslator cannot handle arguments that take more than a vreg. For other functions, the aggregate argument replacement pass would have solved things, but it does not apply here. Hence, we apply the same mutate-into-Int32 solution here when dealing with stores, and restore the extract value's type (which we have available as a ValueAttr) during instruction selection.
95 lines
3.4 KiB
LLVM
95 lines
3.4 KiB
LLVM
; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv32-unknown-unknown %s -o - | FileCheck %s
|
|
; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv64-unknown-unknown %s -o - | FileCheck %s
|
|
; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv32-unknown-unknown %s -o - -filetype=obj | spirv-val %}
|
|
; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv64-unknown-unknown %s -o - -filetype=obj | spirv-val %}
|
|
|
|
; CHECK-DAG: OpName [[FOOBAR:%.+]] "foobar"
|
|
; CHECK-DAG: OpName [[PRODUCER:%.+]] "producer"
|
|
; CHECK-DAG: OpName [[CONSUMER:%.+]] "consumer"
|
|
; CHECK-DAG: OpName [[SNEAKY:%.+]] "sneaky"
|
|
; CHECK-DAG: OpName [[ARR_RET:%.+]] "arr_ret"
|
|
|
|
; CHECK-NOT: DAG-FENCE
|
|
|
|
%ty1 = type {i16, i32}
|
|
%ty2 = type {%ty1, i64}
|
|
%ty3 = type {ptr addrspace(4), ptr addrspace(4), [8 x i8]}
|
|
|
|
; CHECK-DAG: [[I16:%.+]] = OpTypeInt 16
|
|
; CHECK-DAG: [[I32:%.+]] = OpTypeInt 32
|
|
; CHECK-DAG: [[I64:%.+]] = OpTypeInt 64
|
|
; CHECK-DAG: [[TY1:%.+]] = OpTypeStruct [[I16]] [[I32]]
|
|
; CHECK-DAG: [[TY2:%.+]] = OpTypeStruct [[TY1]] [[I64]]
|
|
; CHECK-DAG: [[UNDEF_I16:%.+]] = OpUndef [[I16]]
|
|
; CHECK-DAG: [[UNDEF_I64:%.+]] = OpUndef [[I64]]
|
|
; CHECK-DAG: [[UNDEF_TY2:%.+]] = OpUndef [[TY2]]
|
|
; CHECK-DAG: [[CST_42:%.+]] = OpConstant [[I32]] 42
|
|
; CHECK-DAG: [[I8:%.+]] = OpTypeInt 8 0
|
|
; CHECK-DAG: [[PTR:%.+]] = OpTypePointer Generic [[I8]]
|
|
; CHECK-DAG: [[CST_8:%.+]] = OpConstant [[I32]] 8
|
|
; CHECK-DAG: [[ARR:%.+]] = OpTypeArray [[I8]] [[CST_8]]
|
|
; CHECK-DAG: [[TY3:%.+]] = OpTypeStruct [[PTR]] [[PTR]] [[ARR]]
|
|
; CHECK-DAG: [[PTR_ARR:%.+]] = OpTypePointer Generic [[ARR]]
|
|
|
|
; CHECK-NOT: DAG-FENCE
|
|
|
|
define i32 @foobar() {
|
|
%agg = call %ty2 @producer(i16 undef, i32 42, i64 undef)
|
|
%ret = call i32 @consumer(%ty2 %agg)
|
|
ret i32 %ret
|
|
}
|
|
|
|
; CHECK: [[FOOBAR]] = OpFunction
|
|
; CHECK: [[AGG:%.+]] = OpFunctionCall [[TY2]] [[PRODUCER]] [[UNDEF_I16]] [[CST_42]] [[UNDEF_I64]]
|
|
; CHECK: [[RET:%.+]] = OpFunctionCall [[I32]] [[CONSUMER]] [[AGG]]
|
|
; CHECK: OpReturnValue [[RET]]
|
|
; CHECK: OpFunctionEnd
|
|
|
|
|
|
define %ty2 @producer(i16 %a, i32 %b, i64 %c) {
|
|
%agg1 = insertvalue %ty2 undef, i16 %a, 0, 0
|
|
%agg2 = insertvalue %ty2 %agg1, i32 %b, 0, 1
|
|
%agg3 = insertvalue %ty2 %agg2, i64 %c, 1
|
|
ret %ty2 %agg3
|
|
}
|
|
|
|
; CHECK: [[PRODUCER]] = OpFunction
|
|
; CHECK: [[A:%.+]] = OpFunctionParameter [[I16]]
|
|
; CHECK: [[B:%.+]] = OpFunctionParameter [[I32]]
|
|
; CHECK: [[C:%.+]] = OpFunctionParameter [[I64]]
|
|
; CHECK: [[AGG1:%.+]] = OpCompositeInsert [[TY2]] [[A]] [[UNDEF_TY2]] 0 0
|
|
; CHECK: [[AGG2:%.+]] = OpCompositeInsert [[TY2]] [[B]] [[AGG1]] 0 1
|
|
; CHECK: [[AGG3:%.+]] = OpCompositeInsert [[TY2]] [[C]] [[AGG2]] 1
|
|
; CHECK: OpReturnValue [[AGG3]]
|
|
; CHECK: OpFunctionEnd
|
|
|
|
|
|
define i32 @consumer(%ty2 %agg) {
|
|
%ret = extractvalue %ty2 %agg, 0, 1
|
|
ret i32 %ret
|
|
}
|
|
|
|
; CHECK: [[CONSUMER]] = OpFunction
|
|
; CHECK: [[AGG:%.+]] = OpFunctionParameter [[TY2]]
|
|
; CHECK: [[RET:%.+]] = OpCompositeExtract [[I32]] [[AGG]] 0 1
|
|
; CHECK: OpReturnValue [[RET]]
|
|
; CHECK: OpFunctionEnd
|
|
|
|
declare %ty3 @arr_ret()
|
|
|
|
define void @sneaky(ptr addrspace(4) %p, [8 x i8] %a) {
|
|
%1 = call spir_func %ty3 @arr_ret()
|
|
%2 = getelementptr inbounds nuw %ty3, ptr addrspace(4) %p, i32 0, i32 2
|
|
%3 = extractvalue %ty3 %1, 2
|
|
store [8 x i8] %3, ptr addrspace(4) %2, align 8
|
|
ret void
|
|
}
|
|
|
|
; CHECK: [[SNEAKY]] = OpFunction
|
|
; CHECK: [[PTR_AGG:%.+]] = OpFunctionParameter
|
|
; CHECK: [[A:%.+]] = OpFunctionParameter [[ARR]]
|
|
; CHECK: [[AGG_RET:%.+]] = OpFunctionCall [[TY3]]
|
|
; CHECK: [[GEP:%.+]] = OpInBoundsPtrAccessChain [[PTR_ARR]] [[PTR_AGG]]
|
|
; CHECK: [[EXTRACTV:%.+]] = OpCompositeExtract [[ARR]] [[AGG_RET]] 2
|
|
; CHECK: OpStore [[GEP]] [[EXTRACTV]]
|