redgpu/ispc: Intel® Implicit SPMD Program Compiler

tests/lit-tests/loop_unroll.ispc
66 строк2 KB
Arina Neshlyaeva
Tune -O1 pipeline for code size
11 янв 2025, 00:29
11 янв 2025, 00:298684eeb
// Test to check '#pragma unroll' functionality for different loop statemnts.
// Pragma has higher priority that -O1.

// RUN: %{ispc} %s --target=host --nowrap -O0 --emit-llvm-text --no-discard-value-names --nostdlib -o - | FileCheck %s -check-prefix=CHECKO0
// RUN: %{ispc} %s --target=host --nowrap -O1 --emit-llvm-text --no-discard-value-names --nostdlib -o - | FileCheck %s -check-prefix=CHECKO2
// RUN: %{ispc} %s --target=host --nowrap -O2 --emit-llvm-text --no-discard-value-names --nostdlib -o - | FileCheck %s -check-prefix=CHECKO2

// CHECKO0: define void @foo_for___
// CHECKO0: br label %{{[a-zA-Z_][a-zA-Z0-9_]*}}, !llvm.loop

// CHECKO2: define void @foo_for___
// CHECKO2: call void @goo_for___uni
// CHECKO2: call void @goo_for___uni
// CHECKO2: call void @goo_for___uni

extern void goo_for(uniform int);
void foo_for() {
#pragma unroll(3)
    for (uniform int iter1 = 0; iter1 < 45; iter1++) {
        goo_for(iter1);
    }   
}

// CHECKO0: define void @foo_do___
// CHECKO0: br i1 %{{[a-zA-Z_][a-zA-Z0-9_]*}}, label %{{[a-zA-Z_][a-zA-Z0-9_]*}}, label %{{[a-zA-Z_][a-zA-Z0-9_]*}}, !llvm.loop

// CHECKO2: define void @foo_do___
// CHECKO2: call void @goo_do___uni
// CHECKO2: call void @goo_do___uni
// CHECKO2: call void @goo_do___uni

extern void goo_do(uniform int);
void foo_do() {
    uniform int iter1 = 0;
#pragma unroll(3)
    do {
        goo_do(iter1);
        iter1++;
    } while(iter1 < 45);
}

// CHECKO0: define void @foo_for_no_count___
// CHECKO0: br label %{{[a-zA-Z_][a-zA-Z0-9_]*}}, !llvm.loop

// CHECKO2: define void @foo_for_no_count___
// CHECKO2: call void @goo_for_no_count___uni
// CHECKO2: call void @goo_for_no_count___uni
// CHECKO2: call void @goo_for_no_count___uni
// CHECKO2: call void @goo_for_no_count___uni
// CHECKO2: call void @goo_for_no_count___uni
// CHECKO2: call void @goo_for_no_count___uni
// CHECKO2: call void @goo_for_no_count___uni
// CHECKO2: call void @goo_for_no_count___uni



extern void goo_for_no_count(uniform int);
void foo_for_no_count() {
#pragma unroll
    for (uniform int iter1 = 0; iter1 < 8; iter1++) {
        goo_for_no_count(iter1);
    }   
}

// CHECKO0: !{!"llvm.loop.unroll.count", i32 3}
// CHECKO0: !{!"llvm.loop.unroll.enable"}