1// RUN: %clang_cc1 -triple x86_64-apple-darwin -std=c++11 -emit-llvm -o - %s | FileCheck %s 2 3// Verify while loop is recognized after unroll pragma. 4void while_test(int *List, int Length) { 5 // CHECK: define {{.*}} @_Z10while_test 6 int i = 0; 7 8#pragma unroll 9 while (i < Length) { 10 // CHECK: br label {{.*}}, !llvm.loop ![[LOOP_1:.*]] 11 List[i] = i * 2; 12 i++; 13 } 14} 15 16// Verify do loop is recognized after multi-option pragma clang loop directive. 17void do_test(int *List, int Length) { 18 // CHECK: define {{.*}} @_Z7do_test 19 int i = 0; 20 21#pragma nounroll 22 do { 23 // CHECK: br i1 {{.*}}, label {{.*}}, label {{.*}}, !llvm.loop ![[LOOP_2:.*]] 24 List[i] = i * 2; 25 i++; 26 } while (i < Length); 27} 28 29// Verify for loop is recognized after unroll pragma. 30void for_test(int *List, int Length) { 31// CHECK: define {{.*}} @_Z8for_test 32#pragma unroll 8 33 for (int i = 0; i < Length; i++) { 34 // CHECK: br label {{.*}}, !llvm.loop ![[LOOP_3:.*]] 35 List[i] = i * 2; 36 } 37} 38 39// Verify c++11 for range loop is recognized after unroll pragma. 40void for_range_test() { 41 // CHECK: define {{.*}} @_Z14for_range_test 42 double List[100]; 43 44#pragma unroll(4) 45 for (int i : List) { 46 // CHECK: br label {{.*}}, !llvm.loop ![[LOOP_4:.*]] 47 List[i] = i; 48 } 49} 50 51#define UNROLLCOUNT 8 52 53// Verify defines are correctly resolved in unroll pragmas. 54void for_define_test(int *List, int Length, int Value) { 55// CHECK: define {{.*}} @_Z15for_define_test 56#pragma unroll(UNROLLCOUNT) 57 for (int i = 0; i < Length; i++) { 58 // CHECK: br label {{.*}}, !llvm.loop ![[LOOP_5:.*]] 59 List[i] = i * Value; 60 } 61} 62 63// Verify metadata is generated when template is used. 64template <typename A> 65void for_template_test(A *List, int Length, A Value) { 66// CHECK: define {{.*}} @_Z13template_test 67#pragma unroll 8 68 for (int i = 0; i < Length; i++) { 69 // CHECK: br label {{.*}}, !llvm.loop ![[LOOP_6:.*]] 70 List[i] = i * Value; 71 } 72} 73 74// Verify define is resolved correctly when template is used. 75template <typename A> 76void for_template_define_test(A *List, int Length, A Value) { 77// CHECK: define {{.*}} @_Z24for_template_define_test 78 79#pragma unroll(UNROLLCOUNT) 80 for (int i = 0; i < Length; i++) { 81 // CHECK: br label {{.*}}, !llvm.loop ![[LOOP_7:.*]] 82 List[i] = i * Value; 83 } 84} 85 86#undef UNROLLCOUNT 87 88// Use templates defined above. Test verifies metadata is generated correctly. 89void template_test(double *List, int Length) { 90 double Value = 10; 91 92 for_template_test<double>(List, Length, Value); 93 for_template_define_test<double>(List, Length, Value); 94} 95 96// CHECK: ![[LOOP_1]] = distinct !{![[LOOP_1]], ![[UNROLL_ENABLE:.*]]} 97// CHECK: ![[UNROLL_ENABLE]] = !{!"llvm.loop.unroll.enable"} 98// CHECK: ![[LOOP_2]] = distinct !{![[LOOP_2:.*]], ![[UNROLL_DISABLE:.*]]} 99// CHECK: ![[UNROLL_DISABLE]] = !{!"llvm.loop.unroll.disable"} 100// CHECK: ![[LOOP_3]] = distinct !{![[LOOP_3]], ![[UNROLL_8:.*]]} 101// CHECK: ![[UNROLL_8]] = !{!"llvm.loop.unroll.count", i32 8} 102// CHECK: ![[LOOP_4]] = distinct !{![[LOOP_4]], ![[UNROLL_4:.*]]} 103// CHECK: ![[UNROLL_4]] = !{!"llvm.loop.unroll.count", i32 4} 104// CHECK: ![[LOOP_5]] = distinct !{![[LOOP_5]], ![[UNROLL_8:.*]]} 105// CHECK: ![[LOOP_6]] = distinct !{![[LOOP_6]], ![[UNROLL_8:.*]]} 106// CHECK: ![[LOOP_7]] = distinct !{![[LOOP_7]], ![[UNROLL_8:.*]]} 107