brintos

brintos / llvm-project-archived public Read only

0
0
Text · 3.2 KiB · 3f663d9 Raw
93 lines · cpp
1//===----------------------------------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9// UNSUPPORTED: c++03, c++11, c++1410 11// <experimental/simd>12//13// [simd.class]14// template<class U, class Flags> simd(const U* mem, Flags);15 16#include "../test_utils.h"17 18namespace ex = std::experimental::parallelism_v2;19 20template <class T, class SimdAbi, std::size_t array_size>21struct ElementAlignedLoadCtorHelper {22  template <class U>23  void operator()() const {24    U buffer[array_size];25    for (size_t i = 0; i < array_size; ++i)26      buffer[i] = static_cast<U>(i);27    ex::simd<T, SimdAbi> origin_simd(buffer, ex::element_aligned_tag());28    assert_simd_values_equal(origin_simd, buffer);29  }30};31 32template <class T, class SimdAbi, std::size_t array_size>33struct VectorAlignedLoadCtorHelper {34  template <class U>35  void operator()() const {36    alignas(ex::memory_alignment_v<ex::simd<T, SimdAbi>, U>) U buffer[array_size];37    for (size_t i = 0; i < array_size; ++i)38      buffer[i] = static_cast<U>(i);39    ex::simd<T, SimdAbi> origin_simd(buffer, ex::vector_aligned_tag());40    assert_simd_values_equal(origin_simd, buffer);41  }42};43 44template <class T, class SimdAbi, std::size_t array_size>45struct OveralignedLoadCtorHelper {46  template <class U>47  void operator()() const {48    alignas(bit_ceil(sizeof(U) + 1)) U buffer[array_size];49    for (size_t i = 0; i < array_size; ++i)50      buffer[i] = static_cast<U>(i);51    ex::simd<T, SimdAbi> origin_simd(buffer, ex::overaligned_tag<bit_ceil(sizeof(U) + 1)>());52    assert_simd_values_equal(origin_simd, buffer);53  }54};55 56template <class T, std::size_t>57struct CheckSimdLoadCtor {58  template <class SimdAbi>59  void operator()() {60    constexpr std::size_t array_size = ex::simd_size_v<T, SimdAbi>;61 62    types::for_each(simd_test_types(), ElementAlignedLoadCtorHelper<T, SimdAbi, array_size>());63    types::for_each(simd_test_types(), VectorAlignedLoadCtorHelper<T, SimdAbi, array_size>());64    types::for_each(simd_test_types(), OveralignedLoadCtorHelper<T, SimdAbi, array_size>());65  }66};67 68template <class T, std::size_t>69struct CheckLoadCtorTraits {70  template <class SimdAbi>71  void operator()() {72    // This function shall not participate in overload resolution unless73    // is_simd_flag_type_v<Flags> is true, and74    // U is a vectorizable type.75    static_assert(std::is_constructible_v<ex::simd<T, SimdAbi>, const int*, ex::element_aligned_tag>);76 77    // is_simd_flag_type_v<Flags> is false78    static_assert(!std::is_constructible_v<ex::simd<T, SimdAbi>, const int*, T>);79    static_assert(!std::is_constructible_v<ex::simd<T, SimdAbi>, const int*, SimdAbi>);80 81    // U is not a vectorizable type.82    static_assert(!std::is_constructible_v<ex::simd<T, SimdAbi>, const SimdAbi*, ex::element_aligned_tag>);83    static_assert(84        !std::is_constructible_v<ex::simd<T, SimdAbi>, const ex::element_aligned_tag*, ex::element_aligned_tag>);85  }86};87 88int main(int, char**) {89  test_all_simd_abi<CheckSimdLoadCtor>();90  test_all_simd_abi<CheckLoadCtorTraits>();91  return 0;92}93