clang/test/Parser/lambda-attr.cu

   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
   4 __attribute__((device)) void device_fn() {}
   5 __attribute__((host, device)) void hd_fn() {}
   6
   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
  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.