[cfe-commits] r156642 - in /cfe/trunk: lib/CodeGen/TargetInfo.cpp test/CodeGen/mips-vector-return.c

[Akira Hatanaka]

Author: ahatanak
Date: Fri May 11 16:01:17 2012
New Revision: 156642

URL: http://llvm.org/viewvc/llvm-project?rev=156642&view=rev
Log:
Fix handling of vector return types.
 
A vector should be returned via the hidden pointer argument except if its size
is equal to or smaller than 16-bytes and the target ABI is N32 or N64.


Added:
    cfe/trunk/test/CodeGen/mips-vector-return.c
Modified:
    cfe/trunk/lib/CodeGen/TargetInfo.cpp

Modified: cfe/trunk/lib/CodeGen/TargetInfo.cpp
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/lib/CodeGen/TargetInfo.cpp?rev=156642&r1=156641&r2=156642&view=diff
==============================================================================
--- cfe/trunk/lib/CodeGen/TargetInfo.cpp (original)
+++ cfe/trunk/lib/CodeGen/TargetInfo.cpp Fri May 11 16:01:17 2012
@@ -3360,7 +3360,7 @@
   if (RetTy->isVoidType() || Size == 0)
     return ABIArgInfo::getIgnore();
 
-  if (isAggregateTypeForABI(RetTy)) {
+  if (isAggregateTypeForABI(RetTy) || RetTy->isVectorType()) {
     if (Size <= 128) {
       if (RetTy->isAnyComplexType())
         return ABIArgInfo::getDirect();

Added: cfe/trunk/test/CodeGen/mips-vector-return.c
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/test/CodeGen/mips-vector-return.c?rev=156642&view=auto
==============================================================================
--- cfe/trunk/test/CodeGen/mips-vector-return.c (added)
+++ cfe/trunk/test/CodeGen/mips-vector-return.c Fri May 11 16:01:17 2012
@@ -0,0 +1,21 @@
+// RUN: %clang -target mipsel-unknown-linux -ccc-clang-archs mipsel -O3 -S -o - -emit-llvm %s | FileCheck %s -check-prefix=O32
+// RUN: %clang -target mips64el-unknown-linux -ccc-clang-archs mips64el -O3 -S -mabi=n64 -o - -emit-llvm %s | FileCheck %s -check-prefix=N64
+
+// vectors larger than 16-bytes are returned via the hidden pointer argument. 
+// N64/N32 returns vectors whose size is equal to or smaller than 16-bytes in
+// integer registers. 
+typedef float  v4sf __attribute__ ((__vector_size__ (16)));
+typedef double v4df __attribute__ ((__vector_size__ (32)));
+
+// O32: define void @test_v4sf(<4 x float>* noalias nocapture sret
+// N64: define { i64, i64 } @test_v4sf
+v4sf test_v4sf(float a) {
+  return (v4sf){0.0f, a, 0.0f, 0.0f};
+}
+
+// O32: define void @test_v4df(<4 x double>* noalias nocapture sret
+// N64: define void @test_v4df(<4 x double>* noalias nocapture sret
+v4df test_v4df(double a) {
+  return (v4df){0.0, a, 0.0, 0.0};
+}
+