clang/test/CodeGen/X86/amx_movrs_tranpose_api.c

   1 // RUN: %clang_cc1 %s -ffreestanding -triple=x86_64-unknown-unknown \
   2 // RUN: -target-feature +amx-movrs  -emit-llvm -o - -Wall -Werror -pedantic \
   3 // RUN: -target-feature +amx-transpose -Wno-gnu-statement-expression| FileCheck %s
   4
   5 #include <immintrin.h>
   6 #include <stddef.h>
   7
   8 char buf[2048];
   9 #define STRIDE 32
  10
  11 void test_tile_2rpntlvwz0rs(const void *A, size_t B) {
  12   // CHECK-LABEL: @test_tile_2rpntlvwz0rs
  13   // CHECK: call void @llvm.x86.t2rpntlvwz0rs(i8 1, ptr %{{.*}}, i64 %{{.*}})
  14   _tile_2rpntlvwz0rs(1, A, B);
  15 }
  16
  17 void test_tile_2rpntlvwz0rst1(const void *A, size_t B) {
  18   // CHECK-LABEL: @test_tile_2rpntlvwz0rst1
  19   // CHECK: call void @llvm.x86.t2rpntlvwz0rst1(i8 1, ptr %{{.*}}, i64 %{{.*}})
  20   _tile_2rpntlvwz0rst1(1, A, B);
  21 }
  22
  23 void test_tile_2rpntlvwz1rs(const void *A, size_t B) {
  24   // CHECK-LABEL: @test_tile_2rpntlvwz1rs
  25   // CHECK: call void @llvm.x86.t2rpntlvwz1rs(i8 1, ptr %{{.*}}, i64 %{{.*}})
  26   _tile_2rpntlvwz1rs(1, A, B);
  27 }
  28
  29 void test_tile_2rpntlvwz1rst1(const void *A, size_t B) {
  30   // CHECK-LABEL: @test_tile_2rpntlvwz1rst1
  31   // CHECK: call void @llvm.x86.t2rpntlvwz1rst1(i8 1, ptr %{{.*}}, i64 %{{.*}})
  32   _tile_2rpntlvwz1rst1(1, A, B);
  33 }
  34
  35 void test__tile_2rpntlvwz0rs(__tile1024i dst0, __tile1024i dst1) {
  36   //CHECK-LABEL: @test__tile_2rpntlvwz0rs
  37   //CHECK: call { x86_amx, x86_amx } @llvm.x86.t2rpntlvwz0rs.internal
  38   //CHECK-NEXT: {{%.*}} = extractvalue { x86_amx, x86_amx } {{%.*}}, 0
  39   //CHECK-NEXT: {{%.*}} = call <256 x i32> @llvm.x86.cast.tile.to.vector.v256i32(x86_amx {{%.*}})
  40   //CHECK-NEXT: store <256 x i32> {{%.*}}, ptr {{%.*}}
  41   //CHECK-NEXT: {{%.*}} = extractvalue { x86_amx, x86_amx } {{%.*}}, 1
  42   //CHECK-NEXT: {{%.*}} = call <256 x i32> @llvm.x86.cast.tile.to.vector.v256i32(x86_amx {{%.*}})
  43   //CHECK-NEXT: store <256 x i32> {{%.*}}, ptr {{%.*}}
  44   __tile_2rpntlvwz0rs(&dst0, &dst1, buf, STRIDE);
  45 }
  46
  47 void test__tile_2rpntlvwz0rst1(__tile1024i dst0, __tile1024i dst1) {
  48   //CHECK-LABEL: @test__tile_2rpntlvwz0rst1
  49   //CHECK: call { x86_amx, x86_amx } @llvm.x86.t2rpntlvwz0rst1.internal
  50   //CHECK-NEXT: {{%.*}} = extractvalue { x86_amx, x86_amx } {{%.*}}, 0
  51   //CHECK-NEXT: {{%.*}} = call <256 x i32> @llvm.x86.cast.tile.to.vector.v256i32(x86_amx {{%.*}})
  52   //CHECK-NEXT: store <256 x i32> {{%.*}}, ptr {{%.*}}
  53   //CHECK-NEXT: {{%.*}} = extractvalue { x86_amx, x86_amx } {{%.*}}, 1
  54   //CHECK-NEXT: {{%.*}} = call <256 x i32> @llvm.x86.cast.tile.to.vector.v256i32(x86_amx {{%.*}})
  55   //CHECK-NEXT: store <256 x i32> {{%.*}}, ptr {{%.*}}
  56   __tile_2rpntlvwz0rst1(&dst0, &dst1, buf, STRIDE);
  57 }
  58
  59 void test__tile_2rpntlvwz1rs(__tile1024i dst0, __tile1024i dst1) {
  60   //CHECK-LABEL: @test__tile_2rpntlvwz1rs
  61   //CHECK: call { x86_amx, x86_amx } @llvm.x86.t2rpntlvwz1rs.internal
  62   //CHECK-NEXT: {{%.*}} = extractvalue { x86_amx, x86_amx } {{%.*}}, 0
  63   //CHECK-NEXT: {{%.*}} = call <256 x i32> @llvm.x86.cast.tile.to.vector.v256i32(x86_amx {{%.*}})
  64   //CHECK-NEXT: store <256 x i32> {{%.*}}, ptr {{%.*}}
  65   //CHECK-NEXT: {{%.*}} = extractvalue { x86_amx, x86_amx } {{%.*}}, 1
  66   //CHECK-NEXT: {{%.*}} = call <256 x i32> @llvm.x86.cast.tile.to.vector.v256i32(x86_amx {{%.*}})
  67   //CHECK-NEXT: store <256 x i32> {{%.*}}, ptr {{%.*}}
  68   __tile_2rpntlvwz1rs(&dst0, &dst1, buf, STRIDE);
  69 }
  70
  71 void test__tile_2rpntlvwz1rst1(__tile1024i dst0, __tile1024i dst1) {
  72   //CHECK-LABEL: @test__tile_2rpntlvwz1rst1
  73   //CHECK: call { x86_amx, x86_amx } @llvm.x86.t2rpntlvwz1rst1.internal
  74   //CHECK-NEXT: {{%.*}} = extractvalue { x86_amx, x86_amx } {{%.*}}, 0
  75   //CHECK-NEXT: {{%.*}} = call <256 x i32> @llvm.x86.cast.tile.to.vector.v256i32(x86_amx {{%.*}})
  76   //CHECK-NEXT: store <256 x i32> {{%.*}}, ptr {{%.*}}
  77   //CHECK-NEXT: {{%.*}} = extractvalue { x86_amx, x86_amx } {{%.*}}, 1
  78   //CHECK-NEXT: {{%.*}} = call <256 x i32> @llvm.x86.cast.tile.to.vector.v256i32(x86_amx {{%.*}})
  79   //CHECK-NEXT: store <256 x i32> {{%.*}}, ptr {{%.*}}
  80   __tile_2rpntlvwz1rst1(&dst0, &dst1, buf, STRIDE);
  81 }