xref: /external/clang/test/CodeGen/avx2-builtins.c

// RUN: %clang_cc1 %s -O3 -triple=x86_64-apple-darwin -target-feature +avx2 -emit-llvm -o - | FileCheck %s

// Don't include mm_malloc.h, it's system specific.
#define __MM_MALLOC_H

#include <immintrin.h>

__m256 test_mm256_mpsadbw_epu8(__m256 x, __m256 y) {
  // CHECK: @llvm.x86.avx2.mpsadbw({{.*}}, {{.*}}, i32 3)
  return _mm256_mpsadbw_epu8(x, y, 3);
}

__m256 test_mm256_abs_epi8(__m256 a) {
  // CHECK: @llvm.x86.avx2.pabs.b
  return _mm256_abs_epi8(a);
}

__m256 test_mm256_abs_epi16(__m256 a) {
  // CHECK: @llvm.x86.avx2.pabs.w
  return _mm256_abs_epi16(a);
}

__m256 test_mm256_abs_epi32(__m256 a) {
  // CHECK: @llvm.x86.avx2.pabs.d
  return _mm256_abs_epi32(a);
}

__m256 test_mm256_packs_epi16(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.packsswb
  return _mm256_packs_epi16(a, b);
}

__m256 test_mm256_packs_epi32(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.packssdw
  return _mm256_packs_epi32(a, b);
}

__m256 test_mm256_packs_epu16(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.packuswb
  return _mm256_packus_epi16(a, b);
}

__m256 test_mm256_packs_epu32(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.packusdw
  return _mm256_packus_epi32(a, b);
}

__m256 test_mm256_add_epi8(__m256 a, __m256 b) {
  // CHECK: add <32 x i8>
  return _mm256_add_epi8(a, b);
}

__m256 test_mm256_add_epi16(__m256 a, __m256 b) {
  // CHECK: add <16 x i16>
  return _mm256_add_epi16(a, b);
}

__m256 test_mm256_add_epi32(__m256 a, __m256 b) {
  // CHECK: add <8 x i32>
  return _mm256_add_epi32(a, b);
}

__m256 test_mm256_add_epi64(__m256 a, __m256 b) {
  // CHECK: add <4 x i64>
  return _mm256_add_epi64(a, b);
}

__m256 test_mm256_adds_epi8(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.padds.b
  return _mm256_adds_epi8(a, b);
}

__m256 test_mm256_adds_epi16(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.padds.w
  return _mm256_adds_epi16(a, b);
}

__m256 test_mm256_adds_epu8(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.paddus.b
  return _mm256_adds_epu8(a, b);
}

__m256 test_mm256_adds_epu16(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.paddus.w
  return _mm256_adds_epu16(a, b);
}

__m256 test_mm256_alignr_epi8(__m256 a, __m256 b) {
  // CHECK: shufflevector <32 x i8> %1, <32 x i8> %0, <32 x i32> <i32 2, i32 3, i32 4, i32 5, i32 6, i32 7, i32 8, i32 9, i32 10, i32 11, i32 12, i32 13, i32 14, i32 15, i32 32, i32 33, i32 18, i32 19, i32 20, i32 21, i32 22, i32 23, i32 24, i32 25, i32 26, i32 27, i32 28, i32 29, i32 30, i32 31, i32 48, i32 49>
  return _mm256_alignr_epi8(a, b, 2);
}

__m256 test2_mm256_alignr_epi8(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.psrl.dq({{.*}}, i32 8)
  return _mm256_alignr_epi8(a, b, 17);
}

__m256 test_mm256_sub_epi8(__m256 a, __m256 b) {
  // CHECK: sub <32 x i8>
  return _mm256_sub_epi8(a, b);
}

__m256 test_mm256_sub_epi16(__m256 a, __m256 b) {
  // CHECK: sub <16 x i16>
  return _mm256_sub_epi16(a, b);
}

__m256 test_mm256_sub_epi32(__m256 a, __m256 b) {
  // CHECK: sub <8 x i32>
  return _mm256_sub_epi32(a, b);
}

__m256 test_mm256_sub_epi64(__m256 a, __m256 b) {
  // CHECK: sub <4 x i64>
  return _mm256_sub_epi64(a, b);
}

__m256 test_mm256_subs_epi8(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.psubs.b
  return _mm256_subs_epi8(a, b);
}

__m256 test_mm256_subs_epi16(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.psubs.w
  return _mm256_subs_epi16(a, b);
}

__m256 test_mm256_subs_epu8(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.psubus.b
  return _mm256_subs_epu8(a, b);
}

__m256 test_mm256_subs_epu16(__m256 a, __m256 b) {
  // CHECK: @llvm.x86.avx2.psubus.w
  return _mm256_subs_epu16(a, b);
}

__m256 test_mm256_and_si256(__m256 a, __m256 b) {
  // CHECK: and <4 x i64>
  return _mm256_and_si256(a, b);
}

__m256 test_mm256_andnot_si256(__m256 a, __m256 b) {
  // CHECK: xor <4 x i64>
  // CHECK: and <4 x i64>
  return _mm256_andnot_si256(a, b);
}

__m256 test_mm256_or_si256(__m256 a, __m256 b) {
  // CHECK: or <4 x i64>
  return _mm256_or_si256(a, b);
}

__m256 test_mm256_xor_si256(__m256 a, __m256 b) {
  // CHECK: xor <4 x i64>
  return _mm256_xor_si256(a, b);
}