diff --git a/clang/include/clang/Basic/Attr.td b/clang/include/clang/Basic/Attr.td --- a/clang/include/clang/Basic/Attr.td +++ b/clang/include/clang/Basic/Attr.td @@ -1470,6 +1470,11 @@ let ASTNode = 0; } +def ArmMveStrictPolymorphism : TypeAttr { + let Spellings = [Clang<"__clang_arm_mve_strict_polymorphism">]; + let Documentation = [ArmMveStrictPolymorphismDocs]; +} + def NoUniqueAddress : InheritableAttr, TargetSpecificAttr { let Spellings = [CXX11<"", "no_unique_address", 201803>]; let Subjects = SubjectList<[NonBitField], ErrorDiag>; diff --git a/clang/include/clang/Basic/AttrDocs.td b/clang/include/clang/Basic/AttrDocs.td --- a/clang/include/clang/Basic/AttrDocs.td +++ b/clang/include/clang/Basic/AttrDocs.td @@ -4777,3 +4777,22 @@ zx_status_t zx_handle_close(zx_handle_t handle [[clang::release_handle]]); }]; } + +def ArmMveStrictPolymorphismDocs : Documentation { + let Category = DocCatType; + let Content = [{ +This attribute is used in the implementation of the ACLE intrinsics for the Arm +MVE instruction set. It is used to define the vector types used by the MVE +intrinsics. + +Its effect is to modify the behavior of a vector type with respect to function +overloading. If a candidate function for overload resolution has a parameter +type with this attribute, then the selection of that candidate function will be +disallowed if the actual argument can only be converted via a lax vector +conversion. The aim is to prevent spurious ambiguity in ARM MVE polymorphic +intrinsics. + +However, this attribute does not prohibit lax vector conversions in contexts +other than overloading. + }]; +} diff --git a/clang/lib/AST/TypePrinter.cpp b/clang/lib/AST/TypePrinter.cpp --- a/clang/lib/AST/TypePrinter.cpp +++ b/clang/lib/AST/TypePrinter.cpp @@ -1558,6 +1558,9 @@ case attr::AcquireHandle: OS << "acquire_handle"; break; + case attr::ArmMveStrictPolymorphism: + OS << "__clang_arm_mve_strict_polymorphism"; + break; } OS << "))"; } diff --git a/clang/lib/Sema/SemaOverload.cpp b/clang/lib/Sema/SemaOverload.cpp --- a/clang/lib/Sema/SemaOverload.cpp +++ b/clang/lib/Sema/SemaOverload.cpp @@ -1653,9 +1653,13 @@ // 1)vector types are equivalent AltiVec and GCC vector types // 2)lax vector conversions are permitted and the vector types are of the // same size + // 3)the destination type does not have the ARM MVE strict-polymorphism + // attribute, which inhibits lax vector conversion for overload resolution + // only if (ToType->isVectorType() && FromType->isVectorType()) { if (S.Context.areCompatibleVectorTypes(FromType, ToType) || - S.isLaxVectorConversion(FromType, ToType)) { + (S.isLaxVectorConversion(FromType, ToType) && + !ToType->hasAttr(attr::ArmMveStrictPolymorphism))) { ICK = ICK_Vector_Conversion; return true; } diff --git a/clang/lib/Sema/SemaType.cpp b/clang/lib/Sema/SemaType.cpp --- a/clang/lib/Sema/SemaType.cpp +++ b/clang/lib/Sema/SemaType.cpp @@ -7543,6 +7543,14 @@ VectorType::NeonPolyVector); attr.setUsedAsTypeAttr(); break; + case ParsedAttr::AT_ArmMveStrictPolymorphism: { + ASTContext &Ctx = state.getSema().Context; + type = state.getAttributedType( + createSimpleAttr(Ctx, attr), type, + type); + attr.setUsedAsTypeAttr(); + break; + } case ParsedAttr::AT_OpenCLAccess: HandleOpenCLAccessAttr(type, attr, state.getSema()); attr.setUsedAsTypeAttr(); diff --git a/clang/test/Sema/arm-mve-overloading.c b/clang/test/Sema/arm-mve-overloading.c new file mode 100644 --- /dev/null +++ b/clang/test/Sema/arm-mve-overloading.c @@ -0,0 +1,112 @@ +// RUN: %clang_cc1 -triple thumbv8.1m.main-arm-none-eabi -target-feature +mve.fp -flax-vector-conversions=all -Werror -emit-llvm -o - %s | FileCheck %s +// RUN: %clang_cc1 -triple thumbv8.1m.main-arm-none-eabi -target-feature +mve.fp -flax-vector-conversions=all -verify -fsyntax-only -DERROR_CHECK %s + +typedef signed short int16_t; +typedef signed int int32_t; +typedef signed long long int64_t; +typedef unsigned short uint16_t; +typedef unsigned int uint32_t; +typedef unsigned long long uint64_t; + +typedef __attribute__((neon_vector_type(8), __clang_arm_mve_strict_polymorphism)) int16_t int16x8_t; +typedef __attribute__((neon_vector_type(4), __clang_arm_mve_strict_polymorphism)) int32_t int32x4_t; +typedef __attribute__((neon_vector_type(2), __clang_arm_mve_strict_polymorphism)) int64_t int64x2_t; +typedef __attribute__((neon_vector_type(8), __clang_arm_mve_strict_polymorphism)) uint16_t uint16x8_t; +typedef __attribute__((neon_vector_type(4), __clang_arm_mve_strict_polymorphism)) uint32_t uint32x4_t; +typedef __attribute__((neon_vector_type(2), __clang_arm_mve_strict_polymorphism)) uint64_t uint64x2_t; + +__attribute__((overloadable)) +int overload(int16x8_t x, int16_t y); // expected-note {{candidate function}} +__attribute__((overloadable)) +int overload(int32x4_t x, int32_t y); // expected-note {{candidate function}} +__attribute__((overloadable)) +int overload(uint16x8_t x, uint16_t y); // expected-note {{candidate function}} +__attribute__((overloadable)) +int overload(uint32x4_t x, uint32_t y); // expected-note {{candidate function}} + +int16_t s16; +int32_t s32; +uint16_t u16; +uint32_t u32; + +int16x8_t vs16; +int32x4_t vs32; +uint16x8_t vu16; +uint32x4_t vu32; + +// ---------------------------------------------------------------------- +// Simple cases where the types are correctly matched + +// CHECK-LABEL: @test_easy_s16( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_int16 +int test_easy_s16(void) { return overload(vs16, s16); } + +// CHECK-LABEL: @test_easy_u16( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_uint16 +int test_easy_u16(void) { return overload(vu16, u16); } + +// CHECK-LABEL: @test_easy_s32( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_int32 +int test_easy_s32(void) { return overload(vs32, s32); } + +// CHECK-LABEL: @test_easy_u32( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_uint32 +int test_easy_u32(void) { return overload(vu32, u32); } + +// ---------------------------------------------------------------------- +// Do arithmetic on the scalar, and it may get promoted. We still expect the +// same overloads to be selected if that happens. + +// CHECK-LABEL: @test_promote_s16( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_int16 +int test_promote_s16(void) { return overload(vs16, s16 + 1); } + +// CHECK-LABEL: @test_promote_u16( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_uint16 +int test_promote_u16(void) { return overload(vu16, u16 + 1); } + +// CHECK-LABEL: @test_promote_s32( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_int32 +int test_promote_s32(void) { return overload(vs32, s32 + 1); } + +// CHECK-LABEL: @test_promote_u32( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_uint32 +int test_promote_u32(void) { return overload(vu32, u32 + 1); } + +// ---------------------------------------------------------------------- +// Write a simple integer literal without qualification, and expect +// the vector type to make it unambiguous which integer type you meant +// the literal to be. + +// CHECK-LABEL: @test_literal_s16( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_int16 +int test_literal_s16(void) { return overload(vs16, 1); } + +// CHECK-LABEL: @test_literal_u16( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_uint16 +int test_literal_u16(void) { return overload(vu16, 1); } + +// CHECK-LABEL: @test_literal_s32( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_int32 +int test_literal_s32(void) { return overload(vs32, 1); } + +// CHECK-LABEL: @test_literal_u32( +// CHECK: call i32 @_Z8overload{{[a-zA-Z0-9_]+}}_uint32 +int test_literal_u32(void) { return overload(vu32, 1); } + +// ---------------------------------------------------------------------- +// All of those overload resolutions are supposed to be unambiguous even when +// lax vector conversion is enabled. Check here that a lax conversion in a +// different context still works. + +int16x8_t lax_conversion(void) { return vu32; } + +// ---------------------------------------------------------------------- +// Use a vector type that there really _isn't_ any overload for, and +// make sure that we get a fatal compile error. + +#ifdef ERROR_CHECK +int expect_error(uint64x2_t v) { + return overload(v, 2); // expected-error {{no matching function for call to 'overload'}} +} +#endif diff --git a/clang/utils/TableGen/MveEmitter.cpp b/clang/utils/TableGen/MveEmitter.cpp --- a/clang/utils/TableGen/MveEmitter.cpp +++ b/clang/utils/TableGen/MveEmitter.cpp @@ -1454,8 +1454,9 @@ raw_ostream &OS = parts[ST->requiresFloat() ? Float : 0]; const VectorType *VT = getVectorType(ST); - OS << "typedef __attribute__((neon_vector_type(" << VT->lanes() << "))) " - << ST->cName() << " " << VT->cName() << ";\n"; + OS << "typedef __attribute__((neon_vector_type(" << VT->lanes() + << "), __clang_arm_mve_strict_polymorphism)) " << ST->cName() << " " + << VT->cName() << ";\n"; // Every vector type also comes with a pair of multi-vector types for // the VLD2 and VLD4 instructions.