106 lines · plain
1; RUN: opt -mtriple=dxil-pc-shadermodel6.2-compute -S -dxil-intrinsic-expansion %s | FileCheck %s --check-prefixes=CHECK,CHECK622; RUN: opt -mtriple=dxil-pc-shadermodel6.3-compute -S -dxil-intrinsic-expansion %s | FileCheck %s --check-prefixes=CHECK,CHECK633 4define void @storef64(double %0, i32 %index) {5 ; CHECK: [[B:%.*]] = tail call target("dx.RawBuffer", double, 1, 0)6 ; CHECK-SAME: @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f64_1_0t(7 ; CHECK-SAME: i32 0, i32 0, i32 1, i32 0, ptr null)8 %buffer = tail call target("dx.RawBuffer", double, 1, 0)9 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f64_1_0t(10 i32 0, i32 0, i32 1, i32 0, ptr null)11 12 ; check we don't modify the code in sm6.3 or later13 ; CHECK63: call void @llvm.dx.resource.store.rawbuffer14 ; CHECK63-SAME: target("dx.RawBuffer", double, 1, 0) [[B]], i32 %index, i32 0, double %0)15 16 ; check we split the double and store the lo and hi bits17 ; CHECK62: [[SD:%.*]] = call { i32, i32 } @llvm.dx.splitdouble.i32(double %0)18 ; CHECK62: [[Lo:%.*]] = extractvalue { i32, i32 } [[SD]], 019 ; CHECK62: [[Hi:%.*]] = extractvalue { i32, i32 } [[SD]], 120 ; CHECK62: [[Vec1:%.*]] = insertelement <2 x i32> poison, i32 [[Lo]], i32 021 ; CHECK62: [[Vec2:%.*]] = insertelement <2 x i32> [[Vec1]], i32 [[Hi]], i32 122 ; CHECK62: call void @llvm.dx.resource.store.rawbuffer.tdx.RawBuffer_f64_1_0t.v2i32(23 ; CHECK62-SAME: target("dx.RawBuffer", double, 1, 0) [[B]], i32 %index, i32 0, <2 x i32> [[Vec2]])24 call void @llvm.dx.resource.store.rawbuffer(25 target("dx.RawBuffer", double, 1, 0) %buffer, i32 %index, i32 0,26 double %0)27 ret void28}29 30define void @storev2f64(<2 x double> %0, i32 %index) {31 ; CHECK: [[B:%.*]] = tail call target("dx.RawBuffer", <2 x double>, 1, 0)32 ; CHECK-SAME: @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v2f64_1_0t(33 ; CHECK-SAME: i32 0, i32 0, i32 1, i32 0, ptr null)34 %buffer = tail call target("dx.RawBuffer", <2 x double>, 1, 0)35 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v2f64_1_0t(36 i32 0, i32 0, i32 1, i32 0, ptr null)37 38 ; check we don't modify the code in sm6.3 or later39 ; CHECK63: call void @llvm.dx.resource.store.rawbuffer40 ; CHECK63-SAME: target("dx.RawBuffer", <2 x double>, 1, 0) [[B]], i32 %index, i32 0, <2 x double> %0)41 42 ; CHECK62: [[SD:%.*]] = call { <2 x i32>, <2 x i32> }43 ; CHECK62-SAME: @llvm.dx.splitdouble.v2i32(<2 x double> %0)44 ; CHECK62: [[Lo:%.*]] = extractvalue { <2 x i32>, <2 x i32> } [[SD]], 045 ; CHECK62: [[Hi:%.*]] = extractvalue { <2 x i32>, <2 x i32> } [[SD]], 146 ; CHECK62: [[Vec:%.*]] = shufflevector <2 x i32> [[Lo]], <2 x i32> [[Hi]], <4 x i32> <i32 0, i32 2, i32 1, i32 3>47 ; CHECK62: call void @llvm.dx.resource.store.rawbuffer.tdx.RawBuffer_v2f64_1_0t.v4i32(48 ; CHECK62-SAME: target("dx.RawBuffer", <2 x double>, 1, 0) [[B]], i32 %index, i32 0, <4 x i32> [[Vec]])49 call void @llvm.dx.resource.store.rawbuffer(50 target("dx.RawBuffer", <2 x double>, 1, 0) %buffer, i32 %index, i32 0,51 <2 x double> %0)52 ret void53}54 55define void @storev3f64(<3 x double> %0, i32 %index) {56 ; CHECK: [[Buf:%.*]] = tail call target("dx.RawBuffer", <3 x double>, 1, 0)57 ; CHECK-SAME: @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v3f64_1_0t(58 ; CHECK-SAME: i32 0, i32 0, i32 1, i32 0, ptr null)59 %buffer = tail call target("dx.RawBuffer", <3 x double>, 1, 0)60 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v3f64_1_0t(61 i32 0, i32 0, i32 1, i32 0, ptr null)62 63 ; check we don't modify the code in sm6.3 or later64 ; CHECK63: call void @llvm.dx.resource.store.rawbuffer65 ; CHECK63-SAME: target("dx.RawBuffer", <3 x double>, 1, 0) [[B]], i32 %index, i32 0, <3 x double> %0)66 67 ; CHECK62: [[A:%.*]] = call { <3 x i32>, <3 x i32> } @llvm.dx.splitdouble.v3i32(<3 x double> %0)68 ; CHECK62: [[B:%.*]] = extractvalue { <3 x i32>, <3 x i32> } [[A]], 069 ; CHECK62: [[C:%.*]] = extractvalue { <3 x i32>, <3 x i32> } [[A]], 170 ; CHECK62: [[D:%.*]] = shufflevector <3 x i32> [[B]], <3 x i32> [[C]], <6 x i32> <i32 0, i32 3, i32 1, i32 4, i32 2, i32 5>71 ; CHECK62: [[E:%.*]] = shufflevector <6 x i32> [[D]], <6 x i32> poison, <4 x i32> <i32 0, i32 1, i32 2, i32 3>72 ; CHECK62: call void @llvm.dx.resource.store.rawbuffer.tdx.RawBuffer_v3f64_1_0t.v4i32(target("dx.RawBuffer", <3 x double>, 1, 0) [[Buf]], i32 %index, i32 0, <4 x i32> [[E]])73 ; CHECK62: [[F:%.*]] = shufflevector <6 x i32> [[D]], <6 x i32> poison, <2 x i32> <i32 4, i32 5>74 ; CHECK62: call void @llvm.dx.resource.store.rawbuffer.tdx.RawBuffer_v3f64_1_0t.v2i32(target("dx.RawBuffer", <3 x double>, 1, 0) [[Buf]], i32 %index, i32 16, <2 x i32> [[F]])75 call void @llvm.dx.resource.store.rawbuffer(76 target("dx.RawBuffer", <3 x double>, 1, 0) %buffer, i32 %index, i32 0,77 <3 x double> %0)78 ret void79}80 81define void @storev4f64(<4 x double> %0, i32 %index) {82 ; CHECK: [[Buf:%.*]] = tail call target("dx.RawBuffer", <4 x double>, 1, 0)83 ; CHECK-SAME: @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v4f64_1_0t(84 ; CHECK-SAME: i32 0, i32 0, i32 1, i32 0, ptr null)85 %buffer = tail call target("dx.RawBuffer", <4 x double>, 1, 0)86 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v4f64_1_0t(87 i32 0, i32 0, i32 1, i32 0, ptr null)88 89 ; check we don't modify the code in sm6.3 or later90 ; CHECK63: call void @llvm.dx.resource.store.rawbuffer91 ; CHECK63-SAME: target("dx.RawBuffer", <4 x double>, 1, 0) [[B]], i32 %index, i32 0, <4 x double> %0)92 93 ; CHECK62: [[A:%.*]] = call { <4 x i32>, <4 x i32> } @llvm.dx.splitdouble.v4i32(<4 x double> %0)94 ; CHECK62: [[B:%.*]] = extractvalue { <4 x i32>, <4 x i32> } [[A]], 095 ; CHECK62: [[C:%.*]] = extractvalue { <4 x i32>, <4 x i32> } [[A]], 196 ; CHECK62: [[D:%.*]] = shufflevector <4 x i32> [[B]], <4 x i32> [[C]], <8 x i32> <i32 0, i32 4, i32 1, i32 5, i32 2, i32 6, i32 3, i32 7>97 ; CHECK62: [[E:%.*]] = shufflevector <8 x i32> [[D]], <8 x i32> poison, <4 x i32> <i32 0, i32 1, i32 2, i32 3>98 ; CHECK62: call void @llvm.dx.resource.store.rawbuffer.tdx.RawBuffer_v4f64_1_0t.v4i32(target("dx.RawBuffer", <4 x double>, 1, 0) [[Buf]], i32 %index, i32 0, <4 x i32> [[E]])99 ; CHECK62: [[F:%.*]] = shufflevector <8 x i32> [[D]], <8 x i32> poison, <4 x i32> <i32 4, i32 5, i32 6, i32 7>100 ; CHECK62: call void @llvm.dx.resource.store.rawbuffer.tdx.RawBuffer_v4f64_1_0t.v4i32(target("dx.RawBuffer", <4 x double>, 1, 0) [[Buf]], i32 %index, i32 16, <4 x i32> [[F]])101 call void @llvm.dx.resource.store.rawbuffer(102 target("dx.RawBuffer", <4 x double>, 1, 0) %buffer, i32 %index, i32 0,103 <4 x double> %0)104 ret void105}106