| ; RUN: opt %loadPolly %defaultOpts -polly-import-jscop -polly-import-jscop-dir=%S -polly-codegen %vector-opt -S -dce < %s | FileCheck %s |
| |
| ;#define M 1024 |
| ;#define N 1024 |
| ;#define K 1024 |
| ;float A[K][M]; |
| ;float B[N][K]; |
| ;float C[M][N]; |
| ;/* |
| ;void matmul_vec(void) { |
| ; int i, j, k; |
| ; |
| ; |
| ; /* With much unrolling |
| ; for (i=0;i<=M;i++) |
| ; for (j=0;j<=N;j+=4) |
| ; for (k=0;k<=K;k+=8) |
| ; for (kk=k;kk<=k+7;kk++) |
| ; for (jj=j;jj<=j+3;jj++) |
| ; C[i][jj] += A[kk][i] * B[jj][kk]; |
| ; vec_load splat scalar_load |
| ; */ |
| ; /* Without unrolling |
| ; for (i=0;i<=M;i++) |
| ; for (j=0;j<=N;j+=4) |
| ; for (k=0;k<=K;k++) |
| ; for (jj=j;jj<=j+3;jj++) |
| ; C[i][jj] += A[k][i] * B[jj][kk]; |
| ; vec_load splat scalar_load |
| ; / |
| ; |
| ;} |
| ;i*/ |
| ;int main() |
| ;{ |
| ; int i, j, k; |
| ; //matmul_vec(); |
| ; for(i=0; i<M/4; i++) |
| ; for(k=0; k<K; k++) { |
| ; for(j=0; j<N; j++) |
| ; C[i+0][j] += A[k][i+0] * B[j][k]; |
| ; C[i+1][j] += A[k][i+1] * B[j][k]; |
| ; C[i+2][j] += A[k][i+2] * B[j][k]; |
| ; C[i+3][j] += A[k][i+3] * B[j][k]; |
| ; } |
| ; |
| ; return A[42][42]; |
| ;} |
| |
| target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64" |
| target triple = "x86_64-unknown-linux-gnu" |
| |
| @A = common global [1024 x [1024 x float]] zeroinitializer, align 16 |
| @B = common global [1024 x [1024 x float]] zeroinitializer, align 16 |
| @C = common global [1024 x [1024 x float]] zeroinitializer, align 16 |
| |
| define void @matmul_vec() nounwind { |
| ; <label>:0 |
| br label %1 |
| |
| ; <label>:1 ; preds = %16, %0 |
| %indvar3 = phi i64 [ %indvar.next4, %16 ], [ 0, %0 ] |
| %exitcond9 = icmp ne i64 %indvar3, 1024 |
| br i1 %exitcond9, label %2, label %17 |
| |
| ; <label>:2 ; preds = %1 |
| br label %3 |
| |
| ; <label>:3 ; preds = %14, %2 |
| %indvar1 = phi i64 [ %indvar.next2, %14 ], [ 0, %2 ] |
| %scevgep8 = getelementptr [1024 x [1024 x float]]* @C, i64 0, i64 %indvar3, i64 %indvar1 |
| %exitcond6 = icmp ne i64 %indvar1, 1024 |
| br i1 %exitcond6, label %4, label %15 |
| |
| ; <label>:4 ; preds = %3 |
| br label %5 |
| |
| ; <label>:5 ; preds = %12, %4 |
| %indvar = phi i64 [ %indvar.next, %12 ], [ 0, %4 ] |
| %scevgep5 = getelementptr [1024 x [1024 x float]]* @A, i64 0, i64 %indvar, i64 %indvar3 |
| %scevgep = getelementptr [1024 x [1024 x float]]* @B, i64 0, i64 %indvar1, i64 %indvar |
| %exitcond = icmp ne i64 %indvar, 1024 |
| br i1 %exitcond, label %6, label %13 |
| |
| ; <label>:6 ; preds = %5 |
| %7 = load float* %scevgep5, align 4 |
| %8 = load float* %scevgep, align 4 |
| %9 = fmul float %7, %8 |
| %10 = load float* %scevgep8, align 4 |
| %11 = fadd float %10, %9 |
| store float %11, float* %scevgep8, align 4 |
| br label %12 |
| |
| ; <label>:12 ; preds = %6 |
| %indvar.next = add i64 %indvar, 1 |
| br label %5 |
| |
| ; <label>:13 ; preds = %5 |
| br label %14 |
| |
| ; <label>:14 ; preds = %13 |
| %indvar.next2 = add i64 %indvar1, 1 |
| br label %3 |
| |
| ; <label>:15 ; preds = %3 |
| br label %16 |
| |
| ; <label>:16 ; preds = %15 |
| %indvar.next4 = add i64 %indvar3, 1 |
| br label %1 |
| |
| ; <label>:17 ; preds = %1 |
| ret void |
| } |
| |
| define i32 @main() nounwind { |
| call void @matmul_vec() |
| %1 = load float* getelementptr inbounds ([1024 x [1024 x float]]* @A, i64 0, i64 42, i64 42), align 8 |
| %2 = fptosi float %1 to i32 |
| ret i32 %2 |
| } |
| |
| ; CHECK: load <1 x float>* |
| ; CHECK: shufflevector <1 x float> |
| ; CHECK: load float* |
| ; CHECK: insertelement <4 x float> |
| ; CHECK: load float* |
| ; CHECK: insertelement <4 x float> |
| ; CHECK: load float* |
| ; CHECK: insertelement <4 x float> |
| ; CHECK: load float* |
| ; CHECK: insertelement <4 x float> |
| ; CHECK: fmul <4 x float> |
| ; CHECK: bitcast float* |
| ; CHECK: load <4 x float>* |
| ; CHECK: fadd <4 x float> |
| ; CHECK: bitcast float* |
| ; CHECK: store <4 x float> |