xref: /llvm-project/clang/test/Parser/lambda-attr.cu (revision c05f1639f7f4a8e81ad83bba99bae95553c6064e)
1 // RUN: %clang_cc1 -std=c++11 -fsyntax-only -verify %s
2 // RUN: %clang_cc1 -std=c++11 -fsyntax-only -fcuda-is-device -verify %s
3 
device_fn()4 __attribute__((device)) void device_fn() {}
hd_fn()5 __attribute__((host, device)) void hd_fn() {}
6 
device_attr()7 __attribute__((device)) void device_attr() {
8   ([]() __attribute__((device)) { device_fn(); })();
9   // expected-warning@-1 {{nvcc does not allow '__device__' to appear after the parameter list in lambdas}}
10   ([] __attribute__((device)) () { device_fn(); })();
11   ([] __attribute__((device)) { device_fn(); })();
12 
13   ([&]() __attribute__((device)){ device_fn(); })();
14   // expected-warning@-1 {{nvcc does not allow '__device__' to appear after the parameter list in lambdas}}
15   ([&] __attribute__((device)) () { device_fn(); })();
16   ([&] __attribute__((device)) { device_fn(); })();
17 
18   ([&](int) __attribute__((device)){ device_fn(); })(0);
19   // expected-warning@-1 {{nvcc does not allow '__device__' to appear after the parameter list in lambdas}}
20   ([&] __attribute__((device)) (int) { device_fn(); })(0);
21 
22   // test that noinline can appear anywhere.
23   ([&] __attribute__((device)) __noinline__ () { device_fn(); })();
24   ([&] __noinline__ __attribute__((device)) () { device_fn(); })();
25 }
26 
host_device_attrs()27 __attribute__((host)) __attribute__((device)) void host_device_attrs() {
28   ([]() __attribute__((host)) __attribute__((device)){ hd_fn(); })();
29   // expected-warning@-1 {{nvcc does not allow '__host__' to appear after the parameter list in lambdas}}
30   // expected-warning@-2 {{nvcc does not allow '__device__' to appear after the parameter list in lambdas}}
31   ([] __attribute__((host)) __attribute__((device)) () { hd_fn(); })();
32   ([] __attribute__((host)) __attribute__((device)) { hd_fn(); })();
33 
34   ([&]() __attribute__((host)) __attribute__((device)){ hd_fn(); })();
35   // expected-warning@-1 {{nvcc does not allow '__host__' to appear after the parameter list in lambdas}}
36   // expected-warning@-2 {{nvcc does not allow '__device__' to appear after the parameter list in lambdas}}
37   ([&] __attribute__((host)) __attribute__((device)) () { hd_fn(); })();
38   ([&] __attribute__((host)) __attribute__((device)) { hd_fn(); })();
39 
40   ([&](int) __attribute__((host)) __attribute__((device)){ hd_fn(); })(0);
41   // expected-warning@-1 {{nvcc does not allow '__host__' to appear after the parameter list in lambdas}}
42   // expected-warning@-2 {{nvcc does not allow '__device__' to appear after the parameter list in lambdas}}
43   ([&] __attribute__((host)) __attribute__((device)) (int) { hd_fn(); })(0);
44 
45   // test that noinline can also appear anywhere.
46   ([] __attribute__((host)) __attribute__((device)) () { hd_fn(); })();
47   ([] __attribute__((host)) __noinline__ __attribute__((device)) () { hd_fn(); })();
48   ([] __attribute__((host)) __attribute__((device)) __noinline__ () { hd_fn(); })();
49 }
50 
51 // TODO: Add tests for __attribute__((global)) once we support global lambdas.
52