xref: /llvm-project/clang/test/CodeGenHLSL/this-assignment-overload.hlsl (revision c2063de1593610eda0f4de33c3b89324642ed54c) 
1// RUN: %clang_cc1 -triple dxil-pc-shadermodel6.0-compute -emit-llvm -disable-llvm-passes -o - -std=hlsl202x %s | FileCheck %s
2
3struct Pair {
4  int First;
5  int Second;
6  int getFirst() {
7    Pair Another = {5, 10};
8    this = Another;
9      return this.First;
10  }
11  int getSecond() {
12    this = Pair();
13    return Second;
14  }
15  void operator=(Pair P) {
16    First = P.First;
17    Second = 2;
18  }
19};
20[numthreads(1, 1, 1)]
21void main() {
22  Pair Vals = {1, 2};
23  Vals.First = Vals.getFirst();
24  Vals.Second = Vals.getSecond();
25}
26
27// This test makes a probably safe assumption that HLSL 202x includes operator overloading for assignment operators.
28// CHECK:     define linkonce_odr noundef i32 @_ZN4Pair8getFirstEv(ptr noundef nonnull align 4 dereferenceable(8) %this) #0 align 2 {
29// CHECK-NEXT:entry:
30// CHECK-NEXT:%this.addr = alloca ptr, align 4
31// CHECK-NEXT:%Another = alloca %struct.Pair, align 4
32// CHECK-NEXT:%agg.tmp = alloca %struct.Pair, align 4
33// CHECK-NEXT:store ptr %this, ptr %this.addr, align 4
34// CHECK-NEXT:%this1 = load ptr, ptr %this.addr, align 4
35// CHECK-NEXT:%First = getelementptr inbounds nuw %struct.Pair, ptr %Another, i32 0, i32 0
36// CHECK-NEXT:store i32 5, ptr %First, align 4
37// CHECK-NEXT:%Second = getelementptr inbounds nuw %struct.Pair, ptr %Another, i32 0, i32 1
38// CHECK-NEXT:store i32 10, ptr %Second, align 4
39// CHECK-NEXT:call void @llvm.memcpy.p0.p0.i32(ptr align 4 %agg.tmp, ptr align 4 %Another, i32 8, i1 false)
40// CHECK-NEXT:call void @_ZN4PairaSES_(ptr noundef nonnull align 4 dereferenceable(8) %this1, ptr noundef byval(%struct.Pair) align 4 %agg.tmp)
41// CHECK-NEXT:%First2 = getelementptr inbounds nuw %struct.Pair, ptr %this1, i32 0, i32 0
42// CHECK-NEXT:%0 = load i32, ptr %First2, align 4
43// CHECK-NEXT:ret i32 %0
44
45// CHECK:     define linkonce_odr noundef i32 @_ZN4Pair9getSecondEv(ptr noundef nonnull align 4 dereferenceable(8) %this) #0 align 2 {
46// CHECK-NEXT:entry:
47// CHECK-NEXT:%this.addr = alloca ptr, align 4
48// CHECK-NEXT:%agg.tmp = alloca %struct.Pair, align 4
49// CHECK-NEXT:store ptr %this, ptr %this.addr, align 4
50// CHECK-NEXT:%this1 = load ptr, ptr %this.addr, align 4
51// CHECK-NEXT:call void @llvm.memset.p0.i32(ptr align 4 %agg.tmp, i8 0, i32 8, i1 false)
52// CHECK-NEXT:call void @_ZN4PairaSES_(ptr noundef nonnull align 4 dereferenceable(8) %this1, ptr noundef byval(%struct.Pair) align 4 %agg.tmp)
53// CHECK-NEXT:%Second = getelementptr inbounds nuw %struct.Pair, ptr %this1, i32 0, i32 1
54// CHECK-NEXT:%0 = load i32, ptr %Second, align 4
55// CHECK-NEXT:ret i32 %0
56