clang/test/CodeGen/X86/amx.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=CHECK
   3
   4 #include <immintrin.h>
   5
   6 void test_amx(void *data) {
   7   //CHECK-LABEL: @test_amx
   8   //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 }