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. 4 void 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. 17 void 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. 30 void 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. 40 void 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. 54 void 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. 64 template <typename A> 65 void 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. 75 template <typename A> 76 void 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. 89 void 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
