Skip to content

Commit e147ecc

Browse files
LuoYuankeLuoYuanke
committed
[X86][AMX] Clear AMX lit test case.
Add nounwind attribute to avoid generating cfi instructions. Also make global buffer 64 bytes align in lit test case. Differential Revision: https://reviews.llvm.org/D94910
1 parent 61f6915 commit e147ecc

File tree

6 files changed

+13
-29
lines changed

6 files changed

+13
-29
lines changed

‎llvm/test/CodeGen/X86/AMX/amx-across-func.ll

Lines changed: 3 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -3,25 +3,16 @@
33

44
%struct.__tile_str = type <{ i16, i16, [60 x i8], <256 x i32> }>
55

6-
@buf = dso_local global [3072 x i8] zeroinitializer, align 16
6+
@buf = dso_local global [3072 x i8] zeroinitializer, align 64
77

8-
define dso_local void @test_api(i16 signext %0, i16 signext %1) local_unnamed_addr {
8+
define dso_local void @test_api(i16 signext %0, i16 signext %1) nounwind {
99
; CHECK-LABEL: test_api:
1010
; CHECK: # %bb.0:
1111
; CHECK-NEXT: pushq %rbp
12-
; CHECK-NEXT: .cfi_def_cfa_offset 16
1312
; CHECK-NEXT: pushq %r15
14-
; CHECK-NEXT: .cfi_def_cfa_offset 24
1513
; CHECK-NEXT: pushq %r14
16-
; CHECK-NEXT: .cfi_def_cfa_offset 32
1714
; CHECK-NEXT: pushq %rbx
18-
; CHECK-NEXT: .cfi_def_cfa_offset 40
1915
; CHECK-NEXT: subq $4056, %rsp # imm = 0xFD8
20-
; CHECK-NEXT: .cfi_def_cfa_offset 4096
21-
; CHECK-NEXT: .cfi_offset %rbx, -40
22-
; CHECK-NEXT: .cfi_offset %r14, -32
23-
; CHECK-NEXT: .cfi_offset %r15, -24
24-
; CHECK-NEXT: .cfi_offset %rbp, -16
2516
; CHECK-NEXT: movl %esi, %ebx
2617
; CHECK-NEXT: movl %edi, %ebp
2718
; CHECK-NEXT: vpxord %zmm0, %zmm0, %zmm0
@@ -58,15 +49,10 @@ define dso_local void @test_api(i16 signext %0, i16 signext %1) local_unnamed_ad
5849
; CHECK-NEXT: tdpbssd %tmm2, %tmm1, %tmm0
5950
; CHECK-NEXT: tilestored %tmm0, (%rax,%r14)
6051
; CHECK-NEXT: addq $4056, %rsp # imm = 0xFD8
61-
; CHECK-NEXT: .cfi_def_cfa_offset 40
6252
; CHECK-NEXT: popq %rbx
63-
; CHECK-NEXT: .cfi_def_cfa_offset 32
6453
; CHECK-NEXT: popq %r14
65-
; CHECK-NEXT: .cfi_def_cfa_offset 24
6654
; CHECK-NEXT: popq %r15
67-
; CHECK-NEXT: .cfi_def_cfa_offset 16
6855
; CHECK-NEXT: popq %rbp
69-
; CHECK-NEXT: .cfi_def_cfa_offset 8
7056
; CHECK-NEXT: tilerelease
7157
; CHECK-NEXT: retq
7258
%3 = tail call x86_amx @llvm.x86.tileloadd64.internal(i16 %0, i16 8, i8* getelementptr inbounds ([3072 x i8], [3072 x i8]* @buf, i64 0, i64 0), i64 32)
@@ -78,7 +64,7 @@ define dso_local void @test_api(i16 signext %0, i16 signext %1) local_unnamed_ad
7864
ret void
7965
}
8066

81-
declare dso_local void @foo(...) local_unnamed_addr
67+
declare dso_local void @foo(...)
8268

8369
declare x86_amx @llvm.x86.tileloadd64.internal(i16, i16, i8*, i64)
8470
declare x86_amx @llvm.x86.tdpbssd.internal(i16, i16, i16, x86_amx, x86_amx, x86_amx)

‎llvm/test/CodeGen/X86/AMX/amx-config.ll

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
22
; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+amx-int8 -mattr=+avx512f -verify-machineinstrs | FileCheck %s
33

4-
@buf = dso_local global [1024 x i8] zeroinitializer, align 16
5-
@buf2 = dso_local global [1024 x i8] zeroinitializer, align 16
4+
@buf = dso_local global [1024 x i8] zeroinitializer, align 64
5+
@buf2 = dso_local global [1024 x i8] zeroinitializer, align 64
66

77
; Function Attrs: nounwind uwtable
8-
define dso_local void @test_api(i32 %0, i16 signext %1, i16 signext %2) local_unnamed_addr {
8+
define dso_local void @test_api(i32 %0, i16 signext %1, i16 signext %2) {
99
; CHECK-LABEL: test_api:
1010
; CHECK: # %bb.0:
1111
; CHECK-NEXT: testl %edi, %edi

‎llvm/test/CodeGen/X86/AMX/amx-intrinsic-chain.ll

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
22
; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+amx-int8 -mattr=+avx512f -verify-machineinstrs | FileCheck %s
33

4-
define dso_local void @test_chain(i8* %A_mem, i8* %B_mem, i8* %C_mem) local_unnamed_addr {
4+
define dso_local void @test_chain(i8* %A_mem, i8* %B_mem, i8* %C_mem) {
55
; CHECK-LABEL: test_chain:
66
; CHECK: # %bb.0: # %entry
77
; CHECK-NEXT: vpxord %zmm0, %zmm0, %zmm0

‎llvm/test/CodeGen/X86/AMX/amx-spill-merge.ll

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
22
; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+amx-int8 -mattr=+avx512f -verify-machineinstrs | FileCheck %s
33

4-
@buf = dso_local global [3072 x i8] zeroinitializer, align 16
4+
@buf = dso_local global [3072 x i8] zeroinitializer, align 64
55

66
define dso_local void @test_api(i16 signext %0, i16 signext %1) nounwind {
77
; CHECK-LABEL: test_api:

‎llvm/test/CodeGen/X86/AMX/amx-spill.ll

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,13 @@
11
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
22
; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+amx-int8 -mattr=+avx512f -verify-machineinstrs | FileCheck %s
33

4-
@buf = dso_local global [1024 x i8] zeroinitializer, align 16
5-
@buf2 = dso_local global [1024 x i8] zeroinitializer, align 16
4+
@buf = dso_local global [1024 x i8] zeroinitializer, align 64
5+
@buf2 = dso_local global [1024 x i8] zeroinitializer, align 64
66

7-
define dso_local void @test_api(i32 %0, i16 signext %1, i16 signext %2) local_unnamed_addr {
7+
define dso_local void @test_api(i32 %0, i16 signext %1, i16 signext %2) nounwind {
88
; CHECK-LABEL: test_api:
99
; CHECK: # %bb.0:
1010
; CHECK-NEXT: subq $2936, %rsp # imm = 0xB78
11-
; CHECK-NEXT: .cfi_def_cfa_offset 2944
1211
; CHECK-NEXT: vpxord %zmm0, %zmm0, %zmm0
1312
; CHECK-NEXT: vmovdqu64 %zmm0, {{[0-9]+}}(%rsp)
1413
; CHECK-NEXT: movb $1, {{[0-9]+}}(%rsp)
@@ -63,7 +62,6 @@ define dso_local void @test_api(i32 %0, i16 signext %1, i16 signext %2) local_un
6362
; CHECK-NEXT: movl $32, %ecx
6463
; CHECK-NEXT: tilestored %tmm0, (%rax,%rcx)
6564
; CHECK-NEXT: addq $2936, %rsp # imm = 0xB78
66-
; CHECK-NEXT: .cfi_def_cfa_offset 8
6765
; CHECK-NEXT: tilerelease
6866
; CHECK-NEXT: vzeroupper
6967
; CHECK-NEXT: retq

‎llvm/test/CodeGen/X86/AMX/amx-type.ll

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33

44
%struct.__tile_str = type { i16, i16, <256 x i32> }
55

6-
@buf = dso_local global [1024 x i8] zeroinitializer, align 16
7-
@buf2 = dso_local global [1024 x i8] zeroinitializer, align 16
6+
@buf = dso_local global [1024 x i8] zeroinitializer, align 64
7+
@buf2 = dso_local global [1024 x i8] zeroinitializer, align 64
88

99
; test bitcast x86_amx to <256 x i32>
1010
define dso_local void @test_user_empty(i16 %m, i16 %n, i8 *%buf, i64 %s) {

0 commit comments

Comments
 (0)