1 // RUN: %clang_cc1 %s -ffreestanding -triple=x86_64-unknown-unknown -target-feature +amx-int8 \
2 // RUN: -target-feature +amx-bf16 -emit-llvm -o - -Werror -pedantic | FileCheck %s --check-prefixes=CHECK
3
4 #include <immintrin.h>
5
test_amx(void * data)6 void test_amx(void *data) {
7 //CHECK-LABEL: @test_amx
8 //CHECK: call void @llvm.x86.ldtilecfg(i8* %{{.*}})
9 //CHECK: call void @llvm.x86.sttilecfg(i8* %{{.*}})
10 //CHECK: call void @llvm.x86.tilerelease()
11 //CHECK: call void @llvm.x86.tilezero(i8 3)
12 //CHECK: call void @llvm.x86.tileloadd64(i8 4, i8* %{{.*}}, i64 8)
13 //CHECK: call void @llvm.x86.tileloaddt164(i8 0, i8* %{{.*}}, i64 1)
14 //CHECK: call void @llvm.x86.tilestored64(i8 0, i8* %{{.*}}, i64 1)
15 //CHECK: call void @llvm.x86.tdpbssd(i8 1, i8 2, i8 3)
16 //CHECK: call void @llvm.x86.tdpbsud(i8 1, i8 2, i8 3)
17 //CHECK: call void @llvm.x86.tdpbusd(i8 1, i8 2, i8 3)
18 //CHECK: call void @llvm.x86.tdpbuud(i8 1, i8 2, i8 3)
19 //CHECK: call void @llvm.x86.tdpbf16ps(i8 1, i8 2, i8 3)
20 _tile_loadconfig(data);
21 _tile_storeconfig(data);
22 _tile_release();
23 _tile_zero(3);
24 _tile_loadd(4, data, 8);
25 _tile_stream_loadd(0, data, 1);
26 _tile_stored(0, data, 1);
27 _tile_dpbssd(1, 2, 3);
28 _tile_dpbsud(1, 2, 3);
29 _tile_dpbusd(1, 2, 3);
30 _tile_dpbuud(1, 2, 3);
31 _tile_dpbf16ps(1, 2, 3);
32 }
33