xref: /llvm-project/clang/test/CodeGenHLSL/this-assignment.hlsl (revision c2063de1593610eda0f4de33c3b89324642ed54c)
1// RUN: %clang_cc1 -triple dxil-pc-shadermodel6.0-compute -emit-llvm -disable-llvm-passes -o - -hlsl-entry main %s | FileCheck %s
2
3struct Pair {
4  int First;
5  int Second;
6
7  int getFirst() {
8    Pair Another = {5, 10};
9    this = Another;
10	  return this.First;
11  }
12
13  // In HLSL 202x, this is a move assignment rather than a copy.
14  int getSecond() {
15    this = Pair();
16    return Second;
17  }
18
19  // In HLSL 202x, this is a copy assignment.
20  Pair DoSilly(Pair Obj) {
21    this = Obj;
22    First += 2;
23    return Obj;
24  }
25};
26
27[numthreads(1, 1, 1)]
28void main() {
29  Pair Vals = {1, 2.0};
30  Vals.First = Vals.getFirst();
31  Vals.Second = Vals.getSecond();
32  (void) Vals.DoSilly(Vals);
33}
34
35// This tests reference like implicit this in HLSL
36// CHECK-LABEL:     define {{.*}}getFirst
37// CHECK-NEXT:entry:
38// CHECK-NEXT:%this.addr = alloca ptr, align 4
39// CHECK-NEXT:%Another = alloca %struct.Pair, align 4
40// CHECK-NEXT:store ptr %this, ptr %this.addr, align 4
41// CHECK-NEXT:%this1 = load ptr, ptr %this.addr, align 4
42// CHECK-NEXT:call void @llvm.memcpy.p0.p0.i32(ptr align 4 %Another, ptr align 4 @__const._ZN4Pair8getFirstEv.Another, i32 8, i1 false)
43// CHECK-NEXT:call void @llvm.memcpy.p0.p0.i32(ptr align 4 %this1, ptr align 4 %Another, i32 8, i1 false)
44// CHECK-NEXT:%First = getelementptr inbounds nuw %struct.Pair, ptr %this1, i32 0, i32 0
45
46// CHECK-LABEL:     define {{.*}}getSecond
47// CHECK-NEXT:entry:
48// CHECK-NEXT:%this.addr = alloca ptr, align 4
49// CHECK-NEXT:%ref.tmp = alloca %struct.Pair, align 4
50// CHECK-NEXT:store ptr %this, ptr %this.addr, align 4
51// CHECK-NEXT:%this1 = load ptr, ptr %this.addr, align 4
52// CHECK-NEXT:call void @llvm.memset.p0.i32(ptr align 4 %ref.tmp, i8 0, i32 8, i1 false)
53// CHECK-NEXT:call void @llvm.memcpy.p0.p0.i32(ptr align 4 %this1, ptr align 4 %ref.tmp, i32 8, i1 false)
54// CHECK-NEXT:%Second = getelementptr inbounds nuw %struct.Pair, ptr %this1, i32 0, i32 1
55
56// CHECK-LABEL:     define {{.*}}DoSilly
57// CHECK-NEXT:entry:
58// CHECK-NEXT: [[ThisPtrAddr:%.*]] = alloca ptr
59// CHECK-NEXT: store ptr {{.*}}, ptr [[ThisPtrAddr]]
60// CHECK-NEXT: [[ThisPtr:%.*]] = load ptr, ptr [[ThisPtrAddr]]
61// CHECK-NEXT: call void @llvm.memcpy.p0.p0.i32(ptr align 4 [[ThisPtr]], ptr align 4 [[Obj:%.*]], i32 8, i1 false)
62// CHECK-NEXT: [[FirstAddr:%.*]] = getelementptr inbounds nuw %struct.Pair, ptr [[ThisPtr]], i32 0, i32 0
63// CHECK-NEXT: [[First:%.*]] = load i32, ptr [[FirstAddr]]
64// CHECK-NEXT: [[FirstPlusTwo:%.*]] = add nsw i32 [[First]], 2
65// CHECK-NEXT: store i32 [[FirstPlusTwo]], ptr [[FirstAddr]]
66// CHECK-NEXT: call void @llvm.memcpy.p0.p0.i32(ptr align 4 {{.*}}, ptr align 4 [[Obj]], i32 8, i1 false)
67