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