33 lines · c
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=CHECK3 4#include <immintrin.h>5 6void test_amx(void *data) {7 //CHECK-LABEL: @test_amx8 //CHECK: call void @llvm.x86.ldtilecfg(ptr %{{.*}})9 //CHECK: call void @llvm.x86.sttilecfg(ptr %{{.*}})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, ptr %{{.*}}, i64 8)13 //CHECK: call void @llvm.x86.tileloaddt164(i8 0, ptr %{{.*}}, i64 1)14 //CHECK: call void @llvm.x86.tilestored64(i8 0, ptr %{{.*}}, 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