diff options
Diffstat (limited to 'tests/language-feature/lambda/lambda-1.slang')
| -rw-r--r-- | tests/language-feature/lambda/lambda-1.slang | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/tests/language-feature/lambda/lambda-1.slang b/tests/language-feature/lambda/lambda-1.slang new file mode 100644 index 000000000..66f68d334 --- /dev/null +++ b/tests/language-feature/lambda/lambda-1.slang @@ -0,0 +1,47 @@ + +//TEST:COMPARE_COMPUTE(filecheck-buffer=CHECK):-vk -output-using-type +//TEST:COMPARE_COMPUTE(filecheck-buffer=CHECK):-cpu -output-using-type + +//TEST_INPUT: set outputBuffer = out ubuffer(data=[0 0 0 0], stride=4) +RWStructuredBuffer<float> outputBuffer; + +struct Matrix +{ + float data[16]; + + [mutating] + void map(IFunc<float, float> f) + { + for (int i = 0; i < 16; ++i) + { + data[i] = f(data[i]); + } + } +} + +struct Applier +{ + float c; + void apply(inout Matrix m) + { + m.map((float x) => + { + if (x > 2.5) + return x * c; + else + return x + c; + }); + } +} + +[numthreads(1,1,1)] +void computeMain() +{ + Matrix m = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16 }; + Applier applier = { 1.5 }; + applier.apply(m); + outputBuffer[0] = m.data[1]; + // CHECK: 3.5 + outputBuffer[1] = m.data[3]; + // CHECK: 6.0 +}
\ No newline at end of file |