OpenMPKinds.cpp 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983
  1. //===--- OpenMPKinds.cpp - Token Kinds Support ----------------------------===//
  2. //
  3. // The LLVM Compiler Infrastructure
  4. //
  5. // This file is distributed under the University of Illinois Open Source
  6. // License. See LICENSE.TXT for details.
  7. //
  8. //===----------------------------------------------------------------------===//
  9. /// \file
  10. /// This file implements the OpenMP enum and support functions.
  11. ///
  12. //===----------------------------------------------------------------------===//
  13. #include "clang/Basic/OpenMPKinds.h"
  14. #include "clang/Basic/IdentifierTable.h"
  15. #include "llvm/ADT/StringRef.h"
  16. #include "llvm/ADT/StringSwitch.h"
  17. #include "llvm/Support/ErrorHandling.h"
  18. #include <cassert>
  19. using namespace clang;
  20. OpenMPDirectiveKind clang::getOpenMPDirectiveKind(StringRef Str) {
  21. return llvm::StringSwitch<OpenMPDirectiveKind>(Str)
  22. #define OPENMP_DIRECTIVE(Name) .Case(#Name, OMPD_##Name)
  23. #define OPENMP_DIRECTIVE_EXT(Name, Str) .Case(Str, OMPD_##Name)
  24. #include "clang/Basic/OpenMPKinds.def"
  25. .Default(OMPD_unknown);
  26. }
  27. const char *clang::getOpenMPDirectiveName(OpenMPDirectiveKind Kind) {
  28. assert(Kind <= OMPD_unknown);
  29. switch (Kind) {
  30. case OMPD_unknown:
  31. return "unknown";
  32. #define OPENMP_DIRECTIVE(Name) \
  33. case OMPD_##Name: \
  34. return #Name;
  35. #define OPENMP_DIRECTIVE_EXT(Name, Str) \
  36. case OMPD_##Name: \
  37. return Str;
  38. #include "clang/Basic/OpenMPKinds.def"
  39. break;
  40. }
  41. llvm_unreachable("Invalid OpenMP directive kind");
  42. }
  43. OpenMPClauseKind clang::getOpenMPClauseKind(StringRef Str) {
  44. // 'flush' clause cannot be specified explicitly, because this is an implicit
  45. // clause for 'flush' directive. If the 'flush' clause is explicitly specified
  46. // the Parser should generate a warning about extra tokens at the end of the
  47. // directive.
  48. if (Str == "flush")
  49. return OMPC_unknown;
  50. return llvm::StringSwitch<OpenMPClauseKind>(Str)
  51. #define OPENMP_CLAUSE(Name, Class) .Case(#Name, OMPC_##Name)
  52. #include "clang/Basic/OpenMPKinds.def"
  53. .Case("uniform", OMPC_uniform)
  54. .Default(OMPC_unknown);
  55. }
  56. const char *clang::getOpenMPClauseName(OpenMPClauseKind Kind) {
  57. assert(Kind <= OMPC_unknown);
  58. switch (Kind) {
  59. case OMPC_unknown:
  60. return "unknown";
  61. #define OPENMP_CLAUSE(Name, Class) \
  62. case OMPC_##Name: \
  63. return #Name;
  64. #include "clang/Basic/OpenMPKinds.def"
  65. case OMPC_uniform:
  66. return "uniform";
  67. case OMPC_threadprivate:
  68. return "threadprivate or thread local";
  69. }
  70. llvm_unreachable("Invalid OpenMP clause kind");
  71. }
  72. unsigned clang::getOpenMPSimpleClauseType(OpenMPClauseKind Kind,
  73. StringRef Str) {
  74. switch (Kind) {
  75. case OMPC_default:
  76. return llvm::StringSwitch<OpenMPDefaultClauseKind>(Str)
  77. #define OPENMP_DEFAULT_KIND(Name) .Case(#Name, OMPC_DEFAULT_##Name)
  78. #include "clang/Basic/OpenMPKinds.def"
  79. .Default(OMPC_DEFAULT_unknown);
  80. case OMPC_proc_bind:
  81. return llvm::StringSwitch<OpenMPProcBindClauseKind>(Str)
  82. #define OPENMP_PROC_BIND_KIND(Name) .Case(#Name, OMPC_PROC_BIND_##Name)
  83. #include "clang/Basic/OpenMPKinds.def"
  84. .Default(OMPC_PROC_BIND_unknown);
  85. case OMPC_schedule:
  86. return llvm::StringSwitch<unsigned>(Str)
  87. #define OPENMP_SCHEDULE_KIND(Name) \
  88. .Case(#Name, static_cast<unsigned>(OMPC_SCHEDULE_##Name))
  89. #define OPENMP_SCHEDULE_MODIFIER(Name) \
  90. .Case(#Name, static_cast<unsigned>(OMPC_SCHEDULE_MODIFIER_##Name))
  91. #include "clang/Basic/OpenMPKinds.def"
  92. .Default(OMPC_SCHEDULE_unknown);
  93. case OMPC_depend:
  94. return llvm::StringSwitch<OpenMPDependClauseKind>(Str)
  95. #define OPENMP_DEPEND_KIND(Name) .Case(#Name, OMPC_DEPEND_##Name)
  96. #include "clang/Basic/OpenMPKinds.def"
  97. .Default(OMPC_DEPEND_unknown);
  98. case OMPC_linear:
  99. return llvm::StringSwitch<OpenMPLinearClauseKind>(Str)
  100. #define OPENMP_LINEAR_KIND(Name) .Case(#Name, OMPC_LINEAR_##Name)
  101. #include "clang/Basic/OpenMPKinds.def"
  102. .Default(OMPC_LINEAR_unknown);
  103. case OMPC_map:
  104. return llvm::StringSwitch<OpenMPMapClauseKind>(Str)
  105. #define OPENMP_MAP_KIND(Name) .Case(#Name, OMPC_MAP_##Name)
  106. #include "clang/Basic/OpenMPKinds.def"
  107. .Default(OMPC_MAP_unknown);
  108. case OMPC_dist_schedule:
  109. return llvm::StringSwitch<OpenMPDistScheduleClauseKind>(Str)
  110. #define OPENMP_DIST_SCHEDULE_KIND(Name) .Case(#Name, OMPC_DIST_SCHEDULE_##Name)
  111. #include "clang/Basic/OpenMPKinds.def"
  112. .Default(OMPC_DIST_SCHEDULE_unknown);
  113. case OMPC_defaultmap:
  114. return llvm::StringSwitch<unsigned>(Str)
  115. #define OPENMP_DEFAULTMAP_KIND(Name) \
  116. .Case(#Name, static_cast<unsigned>(OMPC_DEFAULTMAP_##Name))
  117. #define OPENMP_DEFAULTMAP_MODIFIER(Name) \
  118. .Case(#Name, static_cast<unsigned>(OMPC_DEFAULTMAP_MODIFIER_##Name))
  119. #include "clang/Basic/OpenMPKinds.def"
  120. .Default(OMPC_DEFAULTMAP_unknown);
  121. case OMPC_unknown:
  122. case OMPC_threadprivate:
  123. case OMPC_if:
  124. case OMPC_final:
  125. case OMPC_num_threads:
  126. case OMPC_safelen:
  127. case OMPC_simdlen:
  128. case OMPC_collapse:
  129. case OMPC_private:
  130. case OMPC_firstprivate:
  131. case OMPC_lastprivate:
  132. case OMPC_shared:
  133. case OMPC_reduction:
  134. case OMPC_task_reduction:
  135. case OMPC_in_reduction:
  136. case OMPC_aligned:
  137. case OMPC_copyin:
  138. case OMPC_copyprivate:
  139. case OMPC_ordered:
  140. case OMPC_nowait:
  141. case OMPC_untied:
  142. case OMPC_mergeable:
  143. case OMPC_flush:
  144. case OMPC_read:
  145. case OMPC_write:
  146. case OMPC_update:
  147. case OMPC_capture:
  148. case OMPC_seq_cst:
  149. case OMPC_device:
  150. case OMPC_threads:
  151. case OMPC_simd:
  152. case OMPC_num_teams:
  153. case OMPC_thread_limit:
  154. case OMPC_priority:
  155. case OMPC_grainsize:
  156. case OMPC_nogroup:
  157. case OMPC_num_tasks:
  158. case OMPC_hint:
  159. case OMPC_uniform:
  160. case OMPC_to:
  161. case OMPC_from:
  162. case OMPC_use_device_ptr:
  163. case OMPC_is_device_ptr:
  164. case OMPC_unified_address:
  165. case OMPC_unified_shared_memory:
  166. break;
  167. }
  168. llvm_unreachable("Invalid OpenMP simple clause kind");
  169. }
  170. const char *clang::getOpenMPSimpleClauseTypeName(OpenMPClauseKind Kind,
  171. unsigned Type) {
  172. switch (Kind) {
  173. case OMPC_default:
  174. switch (Type) {
  175. case OMPC_DEFAULT_unknown:
  176. return "unknown";
  177. #define OPENMP_DEFAULT_KIND(Name) \
  178. case OMPC_DEFAULT_##Name: \
  179. return #Name;
  180. #include "clang/Basic/OpenMPKinds.def"
  181. }
  182. llvm_unreachable("Invalid OpenMP 'default' clause type");
  183. case OMPC_proc_bind:
  184. switch (Type) {
  185. case OMPC_PROC_BIND_unknown:
  186. return "unknown";
  187. #define OPENMP_PROC_BIND_KIND(Name) \
  188. case OMPC_PROC_BIND_##Name: \
  189. return #Name;
  190. #include "clang/Basic/OpenMPKinds.def"
  191. }
  192. llvm_unreachable("Invalid OpenMP 'proc_bind' clause type");
  193. case OMPC_schedule:
  194. switch (Type) {
  195. case OMPC_SCHEDULE_unknown:
  196. case OMPC_SCHEDULE_MODIFIER_last:
  197. return "unknown";
  198. #define OPENMP_SCHEDULE_KIND(Name) \
  199. case OMPC_SCHEDULE_##Name: \
  200. return #Name;
  201. #define OPENMP_SCHEDULE_MODIFIER(Name) \
  202. case OMPC_SCHEDULE_MODIFIER_##Name: \
  203. return #Name;
  204. #include "clang/Basic/OpenMPKinds.def"
  205. }
  206. llvm_unreachable("Invalid OpenMP 'schedule' clause type");
  207. case OMPC_depend:
  208. switch (Type) {
  209. case OMPC_DEPEND_unknown:
  210. return "unknown";
  211. #define OPENMP_DEPEND_KIND(Name) \
  212. case OMPC_DEPEND_##Name: \
  213. return #Name;
  214. #include "clang/Basic/OpenMPKinds.def"
  215. }
  216. llvm_unreachable("Invalid OpenMP 'depend' clause type");
  217. case OMPC_linear:
  218. switch (Type) {
  219. case OMPC_LINEAR_unknown:
  220. return "unknown";
  221. #define OPENMP_LINEAR_KIND(Name) \
  222. case OMPC_LINEAR_##Name: \
  223. return #Name;
  224. #include "clang/Basic/OpenMPKinds.def"
  225. }
  226. llvm_unreachable("Invalid OpenMP 'linear' clause type");
  227. case OMPC_map:
  228. switch (Type) {
  229. case OMPC_MAP_unknown:
  230. return "unknown";
  231. #define OPENMP_MAP_KIND(Name) \
  232. case OMPC_MAP_##Name: \
  233. return #Name;
  234. #include "clang/Basic/OpenMPKinds.def"
  235. default:
  236. break;
  237. }
  238. llvm_unreachable("Invalid OpenMP 'map' clause type");
  239. case OMPC_dist_schedule:
  240. switch (Type) {
  241. case OMPC_DIST_SCHEDULE_unknown:
  242. return "unknown";
  243. #define OPENMP_DIST_SCHEDULE_KIND(Name) \
  244. case OMPC_DIST_SCHEDULE_##Name: \
  245. return #Name;
  246. #include "clang/Basic/OpenMPKinds.def"
  247. }
  248. llvm_unreachable("Invalid OpenMP 'dist_schedule' clause type");
  249. case OMPC_defaultmap:
  250. switch (Type) {
  251. case OMPC_DEFAULTMAP_unknown:
  252. case OMPC_DEFAULTMAP_MODIFIER_last:
  253. return "unknown";
  254. #define OPENMP_DEFAULTMAP_KIND(Name) \
  255. case OMPC_DEFAULTMAP_##Name: \
  256. return #Name;
  257. #define OPENMP_DEFAULTMAP_MODIFIER(Name) \
  258. case OMPC_DEFAULTMAP_MODIFIER_##Name: \
  259. return #Name;
  260. #include "clang/Basic/OpenMPKinds.def"
  261. }
  262. llvm_unreachable("Invalid OpenMP 'schedule' clause type");
  263. case OMPC_unknown:
  264. case OMPC_threadprivate:
  265. case OMPC_if:
  266. case OMPC_final:
  267. case OMPC_num_threads:
  268. case OMPC_safelen:
  269. case OMPC_simdlen:
  270. case OMPC_collapse:
  271. case OMPC_private:
  272. case OMPC_firstprivate:
  273. case OMPC_lastprivate:
  274. case OMPC_shared:
  275. case OMPC_reduction:
  276. case OMPC_task_reduction:
  277. case OMPC_in_reduction:
  278. case OMPC_aligned:
  279. case OMPC_copyin:
  280. case OMPC_copyprivate:
  281. case OMPC_ordered:
  282. case OMPC_nowait:
  283. case OMPC_untied:
  284. case OMPC_mergeable:
  285. case OMPC_flush:
  286. case OMPC_read:
  287. case OMPC_write:
  288. case OMPC_update:
  289. case OMPC_capture:
  290. case OMPC_seq_cst:
  291. case OMPC_device:
  292. case OMPC_threads:
  293. case OMPC_simd:
  294. case OMPC_num_teams:
  295. case OMPC_thread_limit:
  296. case OMPC_priority:
  297. case OMPC_grainsize:
  298. case OMPC_nogroup:
  299. case OMPC_num_tasks:
  300. case OMPC_hint:
  301. case OMPC_uniform:
  302. case OMPC_to:
  303. case OMPC_from:
  304. case OMPC_use_device_ptr:
  305. case OMPC_is_device_ptr:
  306. case OMPC_unified_address:
  307. case OMPC_unified_shared_memory:
  308. break;
  309. }
  310. llvm_unreachable("Invalid OpenMP simple clause kind");
  311. }
  312. bool clang::isAllowedClauseForDirective(OpenMPDirectiveKind DKind,
  313. OpenMPClauseKind CKind) {
  314. assert(DKind <= OMPD_unknown);
  315. assert(CKind <= OMPC_unknown);
  316. switch (DKind) {
  317. case OMPD_parallel:
  318. switch (CKind) {
  319. #define OPENMP_PARALLEL_CLAUSE(Name) \
  320. case OMPC_##Name: \
  321. return true;
  322. #include "clang/Basic/OpenMPKinds.def"
  323. default:
  324. break;
  325. }
  326. break;
  327. case OMPD_simd:
  328. switch (CKind) {
  329. #define OPENMP_SIMD_CLAUSE(Name) \
  330. case OMPC_##Name: \
  331. return true;
  332. #include "clang/Basic/OpenMPKinds.def"
  333. default:
  334. break;
  335. }
  336. break;
  337. case OMPD_for:
  338. switch (CKind) {
  339. #define OPENMP_FOR_CLAUSE(Name) \
  340. case OMPC_##Name: \
  341. return true;
  342. #include "clang/Basic/OpenMPKinds.def"
  343. default:
  344. break;
  345. }
  346. break;
  347. case OMPD_for_simd:
  348. switch (CKind) {
  349. #define OPENMP_FOR_SIMD_CLAUSE(Name) \
  350. case OMPC_##Name: \
  351. return true;
  352. #include "clang/Basic/OpenMPKinds.def"
  353. default:
  354. break;
  355. }
  356. break;
  357. case OMPD_sections:
  358. switch (CKind) {
  359. #define OPENMP_SECTIONS_CLAUSE(Name) \
  360. case OMPC_##Name: \
  361. return true;
  362. #include "clang/Basic/OpenMPKinds.def"
  363. default:
  364. break;
  365. }
  366. break;
  367. case OMPD_single:
  368. switch (CKind) {
  369. #define OPENMP_SINGLE_CLAUSE(Name) \
  370. case OMPC_##Name: \
  371. return true;
  372. #include "clang/Basic/OpenMPKinds.def"
  373. default:
  374. break;
  375. }
  376. break;
  377. case OMPD_parallel_for:
  378. switch (CKind) {
  379. #define OPENMP_PARALLEL_FOR_CLAUSE(Name) \
  380. case OMPC_##Name: \
  381. return true;
  382. #include "clang/Basic/OpenMPKinds.def"
  383. default:
  384. break;
  385. }
  386. break;
  387. case OMPD_parallel_for_simd:
  388. switch (CKind) {
  389. #define OPENMP_PARALLEL_FOR_SIMD_CLAUSE(Name) \
  390. case OMPC_##Name: \
  391. return true;
  392. #include "clang/Basic/OpenMPKinds.def"
  393. default:
  394. break;
  395. }
  396. break;
  397. case OMPD_parallel_sections:
  398. switch (CKind) {
  399. #define OPENMP_PARALLEL_SECTIONS_CLAUSE(Name) \
  400. case OMPC_##Name: \
  401. return true;
  402. #include "clang/Basic/OpenMPKinds.def"
  403. default:
  404. break;
  405. }
  406. break;
  407. case OMPD_task:
  408. switch (CKind) {
  409. #define OPENMP_TASK_CLAUSE(Name) \
  410. case OMPC_##Name: \
  411. return true;
  412. #include "clang/Basic/OpenMPKinds.def"
  413. default:
  414. break;
  415. }
  416. break;
  417. case OMPD_flush:
  418. return CKind == OMPC_flush;
  419. break;
  420. case OMPD_atomic:
  421. switch (CKind) {
  422. #define OPENMP_ATOMIC_CLAUSE(Name) \
  423. case OMPC_##Name: \
  424. return true;
  425. #include "clang/Basic/OpenMPKinds.def"
  426. default:
  427. break;
  428. }
  429. break;
  430. case OMPD_target:
  431. switch (CKind) {
  432. #define OPENMP_TARGET_CLAUSE(Name) \
  433. case OMPC_##Name: \
  434. return true;
  435. #include "clang/Basic/OpenMPKinds.def"
  436. default:
  437. break;
  438. }
  439. break;
  440. case OMPD_requires:
  441. switch (CKind) {
  442. #define OPENMP_REQUIRES_CLAUSE(Name) \
  443. case OMPC_##Name: \
  444. return true;
  445. #include "clang/Basic/OpenMPKinds.def"
  446. default:
  447. break;
  448. }
  449. break;
  450. case OMPD_target_data:
  451. switch (CKind) {
  452. #define OPENMP_TARGET_DATA_CLAUSE(Name) \
  453. case OMPC_##Name: \
  454. return true;
  455. #include "clang/Basic/OpenMPKinds.def"
  456. default:
  457. break;
  458. }
  459. break;
  460. case OMPD_target_enter_data:
  461. switch (CKind) {
  462. #define OPENMP_TARGET_ENTER_DATA_CLAUSE(Name) \
  463. case OMPC_##Name: \
  464. return true;
  465. #include "clang/Basic/OpenMPKinds.def"
  466. default:
  467. break;
  468. }
  469. break;
  470. case OMPD_target_exit_data:
  471. switch (CKind) {
  472. #define OPENMP_TARGET_EXIT_DATA_CLAUSE(Name) \
  473. case OMPC_##Name: \
  474. return true;
  475. #include "clang/Basic/OpenMPKinds.def"
  476. default:
  477. break;
  478. }
  479. break;
  480. case OMPD_target_parallel:
  481. switch (CKind) {
  482. #define OPENMP_TARGET_PARALLEL_CLAUSE(Name) \
  483. case OMPC_##Name: \
  484. return true;
  485. #include "clang/Basic/OpenMPKinds.def"
  486. default:
  487. break;
  488. }
  489. break;
  490. case OMPD_target_parallel_for:
  491. switch (CKind) {
  492. #define OPENMP_TARGET_PARALLEL_FOR_CLAUSE(Name) \
  493. case OMPC_##Name: \
  494. return true;
  495. #include "clang/Basic/OpenMPKinds.def"
  496. default:
  497. break;
  498. }
  499. break;
  500. case OMPD_target_update:
  501. switch (CKind) {
  502. #define OPENMP_TARGET_UPDATE_CLAUSE(Name) \
  503. case OMPC_##Name: \
  504. return true;
  505. #include "clang/Basic/OpenMPKinds.def"
  506. default:
  507. break;
  508. }
  509. break;
  510. case OMPD_teams:
  511. switch (CKind) {
  512. #define OPENMP_TEAMS_CLAUSE(Name) \
  513. case OMPC_##Name: \
  514. return true;
  515. #include "clang/Basic/OpenMPKinds.def"
  516. default:
  517. break;
  518. }
  519. break;
  520. case OMPD_declare_simd:
  521. break;
  522. case OMPD_cancel:
  523. switch (CKind) {
  524. #define OPENMP_CANCEL_CLAUSE(Name) \
  525. case OMPC_##Name: \
  526. return true;
  527. #include "clang/Basic/OpenMPKinds.def"
  528. default:
  529. break;
  530. }
  531. break;
  532. case OMPD_ordered:
  533. switch (CKind) {
  534. #define OPENMP_ORDERED_CLAUSE(Name) \
  535. case OMPC_##Name: \
  536. return true;
  537. #include "clang/Basic/OpenMPKinds.def"
  538. default:
  539. break;
  540. }
  541. break;
  542. case OMPD_taskloop:
  543. switch (CKind) {
  544. #define OPENMP_TASKLOOP_CLAUSE(Name) \
  545. case OMPC_##Name: \
  546. return true;
  547. #include "clang/Basic/OpenMPKinds.def"
  548. default:
  549. break;
  550. }
  551. break;
  552. case OMPD_taskloop_simd:
  553. switch (CKind) {
  554. #define OPENMP_TASKLOOP_SIMD_CLAUSE(Name) \
  555. case OMPC_##Name: \
  556. return true;
  557. #include "clang/Basic/OpenMPKinds.def"
  558. default:
  559. break;
  560. }
  561. break;
  562. case OMPD_critical:
  563. switch (CKind) {
  564. #define OPENMP_CRITICAL_CLAUSE(Name) \
  565. case OMPC_##Name: \
  566. return true;
  567. #include "clang/Basic/OpenMPKinds.def"
  568. default:
  569. break;
  570. }
  571. break;
  572. case OMPD_distribute:
  573. switch (CKind) {
  574. #define OPENMP_DISTRIBUTE_CLAUSE(Name) \
  575. case OMPC_##Name: \
  576. return true;
  577. #include "clang/Basic/OpenMPKinds.def"
  578. default:
  579. break;
  580. }
  581. break;
  582. case OMPD_distribute_parallel_for:
  583. switch (CKind) {
  584. #define OPENMP_DISTRIBUTE_PARALLEL_FOR_CLAUSE(Name) \
  585. case OMPC_##Name: \
  586. return true;
  587. #include "clang/Basic/OpenMPKinds.def"
  588. default:
  589. break;
  590. }
  591. break;
  592. case OMPD_distribute_parallel_for_simd:
  593. switch (CKind) {
  594. #define OPENMP_DISTRIBUTE_PARALLEL_FOR_SIMD_CLAUSE(Name) \
  595. case OMPC_##Name: \
  596. return true;
  597. #include "clang/Basic/OpenMPKinds.def"
  598. default:
  599. break;
  600. }
  601. break;
  602. case OMPD_distribute_simd:
  603. switch (CKind) {
  604. #define OPENMP_DISTRIBUTE_SIMD_CLAUSE(Name) \
  605. case OMPC_##Name: \
  606. return true;
  607. #include "clang/Basic/OpenMPKinds.def"
  608. default:
  609. break;
  610. }
  611. break;
  612. case OMPD_target_parallel_for_simd:
  613. switch (CKind) {
  614. #define OPENMP_TARGET_PARALLEL_FOR_SIMD_CLAUSE(Name) \
  615. case OMPC_##Name: \
  616. return true;
  617. #include "clang/Basic/OpenMPKinds.def"
  618. default:
  619. break;
  620. }
  621. break;
  622. case OMPD_target_simd:
  623. switch (CKind) {
  624. #define OPENMP_TARGET_SIMD_CLAUSE(Name) \
  625. case OMPC_##Name: \
  626. return true;
  627. #include "clang/Basic/OpenMPKinds.def"
  628. default:
  629. break;
  630. }
  631. break;
  632. case OMPD_teams_distribute:
  633. switch (CKind) {
  634. #define OPENMP_TEAMS_DISTRIBUTE_CLAUSE(Name) \
  635. case OMPC_##Name: \
  636. return true;
  637. #include "clang/Basic/OpenMPKinds.def"
  638. default:
  639. break;
  640. }
  641. break;
  642. case OMPD_teams_distribute_simd:
  643. switch (CKind) {
  644. #define OPENMP_TEAMS_DISTRIBUTE_SIMD_CLAUSE(Name) \
  645. case OMPC_##Name: \
  646. return true;
  647. #include "clang/Basic/OpenMPKinds.def"
  648. default:
  649. break;
  650. }
  651. break;
  652. case OMPD_teams_distribute_parallel_for_simd:
  653. switch (CKind) {
  654. #define OPENMP_TEAMS_DISTRIBUTE_PARALLEL_FOR_SIMD_CLAUSE(Name) \
  655. case OMPC_##Name: \
  656. return true;
  657. #include "clang/Basic/OpenMPKinds.def"
  658. default:
  659. break;
  660. }
  661. break;
  662. case OMPD_teams_distribute_parallel_for:
  663. switch (CKind) {
  664. #define OPENMP_TEAMS_DISTRIBUTE_PARALLEL_FOR_CLAUSE(Name) \
  665. case OMPC_##Name: \
  666. return true;
  667. #include "clang/Basic/OpenMPKinds.def"
  668. default:
  669. break;
  670. }
  671. break;
  672. case OMPD_target_teams:
  673. switch (CKind) {
  674. #define OPENMP_TARGET_TEAMS_CLAUSE(Name) \
  675. case OMPC_##Name: \
  676. return true;
  677. #include "clang/Basic/OpenMPKinds.def"
  678. default:
  679. break;
  680. }
  681. break;
  682. case OMPD_target_teams_distribute:
  683. switch (CKind) {
  684. #define OPENMP_TARGET_TEAMS_DISTRIBUTE_CLAUSE(Name) \
  685. case OMPC_##Name: \
  686. return true;
  687. #include "clang/Basic/OpenMPKinds.def"
  688. default:
  689. break;
  690. }
  691. break;
  692. case OMPD_target_teams_distribute_parallel_for:
  693. switch (CKind) {
  694. #define OPENMP_TARGET_TEAMS_DISTRIBUTE_PARALLEL_FOR_CLAUSE(Name) \
  695. case OMPC_##Name: \
  696. return true;
  697. #include "clang/Basic/OpenMPKinds.def"
  698. default:
  699. break;
  700. }
  701. break;
  702. case OMPD_target_teams_distribute_parallel_for_simd:
  703. switch (CKind) {
  704. #define OPENMP_TARGET_TEAMS_DISTRIBUTE_PARALLEL_FOR_SIMD_CLAUSE(Name) \
  705. case OMPC_##Name: \
  706. return true;
  707. #include "clang/Basic/OpenMPKinds.def"
  708. default:
  709. break;
  710. }
  711. break;
  712. case OMPD_target_teams_distribute_simd:
  713. switch (CKind) {
  714. #define OPENMP_TARGET_TEAMS_DISTRIBUTE_SIMD_CLAUSE(Name) \
  715. case OMPC_##Name: \
  716. return true;
  717. #include "clang/Basic/OpenMPKinds.def"
  718. default:
  719. break;
  720. }
  721. break;
  722. case OMPD_taskgroup:
  723. switch (CKind) {
  724. #define OPENMP_TASKGROUP_CLAUSE(Name) \
  725. case OMPC_##Name: \
  726. return true;
  727. #include "clang/Basic/OpenMPKinds.def"
  728. default:
  729. break;
  730. }
  731. break;
  732. case OMPD_declare_target:
  733. case OMPD_end_declare_target:
  734. case OMPD_unknown:
  735. case OMPD_threadprivate:
  736. case OMPD_section:
  737. case OMPD_master:
  738. case OMPD_taskyield:
  739. case OMPD_barrier:
  740. case OMPD_taskwait:
  741. case OMPD_cancellation_point:
  742. case OMPD_declare_reduction:
  743. break;
  744. }
  745. return false;
  746. }
  747. bool clang::isOpenMPLoopDirective(OpenMPDirectiveKind DKind) {
  748. return DKind == OMPD_simd || DKind == OMPD_for || DKind == OMPD_for_simd ||
  749. DKind == OMPD_parallel_for || DKind == OMPD_parallel_for_simd ||
  750. DKind == OMPD_taskloop || DKind == OMPD_taskloop_simd ||
  751. DKind == OMPD_distribute || DKind == OMPD_target_parallel_for ||
  752. DKind == OMPD_distribute_parallel_for ||
  753. DKind == OMPD_distribute_parallel_for_simd ||
  754. DKind == OMPD_distribute_simd ||
  755. DKind == OMPD_target_parallel_for_simd || DKind == OMPD_target_simd ||
  756. DKind == OMPD_teams_distribute ||
  757. DKind == OMPD_teams_distribute_simd ||
  758. DKind == OMPD_teams_distribute_parallel_for_simd ||
  759. DKind == OMPD_teams_distribute_parallel_for ||
  760. DKind == OMPD_target_teams_distribute ||
  761. DKind == OMPD_target_teams_distribute_parallel_for ||
  762. DKind == OMPD_target_teams_distribute_parallel_for_simd ||
  763. DKind == OMPD_target_teams_distribute_simd;
  764. }
  765. bool clang::isOpenMPWorksharingDirective(OpenMPDirectiveKind DKind) {
  766. return DKind == OMPD_for || DKind == OMPD_for_simd ||
  767. DKind == OMPD_sections || DKind == OMPD_section ||
  768. DKind == OMPD_single || DKind == OMPD_parallel_for ||
  769. DKind == OMPD_parallel_for_simd || DKind == OMPD_parallel_sections ||
  770. DKind == OMPD_target_parallel_for ||
  771. DKind == OMPD_distribute_parallel_for ||
  772. DKind == OMPD_distribute_parallel_for_simd ||
  773. DKind == OMPD_target_parallel_for_simd ||
  774. DKind == OMPD_teams_distribute_parallel_for_simd ||
  775. DKind == OMPD_teams_distribute_parallel_for ||
  776. DKind == OMPD_target_teams_distribute_parallel_for ||
  777. DKind == OMPD_target_teams_distribute_parallel_for_simd;
  778. }
  779. bool clang::isOpenMPTaskLoopDirective(OpenMPDirectiveKind DKind) {
  780. return DKind == OMPD_taskloop || DKind == OMPD_taskloop_simd;
  781. }
  782. bool clang::isOpenMPParallelDirective(OpenMPDirectiveKind DKind) {
  783. return DKind == OMPD_parallel || DKind == OMPD_parallel_for ||
  784. DKind == OMPD_parallel_for_simd || DKind == OMPD_parallel_sections ||
  785. DKind == OMPD_target_parallel || DKind == OMPD_target_parallel_for ||
  786. DKind == OMPD_distribute_parallel_for ||
  787. DKind == OMPD_distribute_parallel_for_simd ||
  788. DKind == OMPD_target_parallel_for_simd ||
  789. DKind == OMPD_teams_distribute_parallel_for ||
  790. DKind == OMPD_teams_distribute_parallel_for_simd ||
  791. DKind == OMPD_target_teams_distribute_parallel_for ||
  792. DKind == OMPD_target_teams_distribute_parallel_for_simd;
  793. }
  794. bool clang::isOpenMPTargetExecutionDirective(OpenMPDirectiveKind DKind) {
  795. return DKind == OMPD_target || DKind == OMPD_target_parallel ||
  796. DKind == OMPD_target_parallel_for ||
  797. DKind == OMPD_target_parallel_for_simd || DKind == OMPD_target_simd ||
  798. DKind == OMPD_target_teams || DKind == OMPD_target_teams_distribute ||
  799. DKind == OMPD_target_teams_distribute_parallel_for ||
  800. DKind == OMPD_target_teams_distribute_parallel_for_simd ||
  801. DKind == OMPD_target_teams_distribute_simd;
  802. }
  803. bool clang::isOpenMPTargetDataManagementDirective(OpenMPDirectiveKind DKind) {
  804. return DKind == OMPD_target_data || DKind == OMPD_target_enter_data ||
  805. DKind == OMPD_target_exit_data || DKind == OMPD_target_update;
  806. }
  807. bool clang::isOpenMPNestingTeamsDirective(OpenMPDirectiveKind DKind) {
  808. return DKind == OMPD_teams || DKind == OMPD_teams_distribute ||
  809. DKind == OMPD_teams_distribute_simd ||
  810. DKind == OMPD_teams_distribute_parallel_for_simd ||
  811. DKind == OMPD_teams_distribute_parallel_for;
  812. }
  813. bool clang::isOpenMPTeamsDirective(OpenMPDirectiveKind DKind) {
  814. return isOpenMPNestingTeamsDirective(DKind) ||
  815. DKind == OMPD_target_teams || DKind == OMPD_target_teams_distribute ||
  816. DKind == OMPD_target_teams_distribute_parallel_for ||
  817. DKind == OMPD_target_teams_distribute_parallel_for_simd ||
  818. DKind == OMPD_target_teams_distribute_simd;
  819. }
  820. bool clang::isOpenMPSimdDirective(OpenMPDirectiveKind DKind) {
  821. return DKind == OMPD_simd || DKind == OMPD_for_simd ||
  822. DKind == OMPD_parallel_for_simd || DKind == OMPD_taskloop_simd ||
  823. DKind == OMPD_distribute_parallel_for_simd ||
  824. DKind == OMPD_distribute_simd || DKind == OMPD_target_simd ||
  825. DKind == OMPD_teams_distribute_simd ||
  826. DKind == OMPD_teams_distribute_parallel_for_simd ||
  827. DKind == OMPD_target_teams_distribute_parallel_for_simd ||
  828. DKind == OMPD_target_teams_distribute_simd ||
  829. DKind == OMPD_target_parallel_for_simd;
  830. }
  831. bool clang::isOpenMPNestingDistributeDirective(OpenMPDirectiveKind Kind) {
  832. return Kind == OMPD_distribute || Kind == OMPD_distribute_parallel_for ||
  833. Kind == OMPD_distribute_parallel_for_simd ||
  834. Kind == OMPD_distribute_simd;
  835. // TODO add next directives.
  836. }
  837. bool clang::isOpenMPDistributeDirective(OpenMPDirectiveKind Kind) {
  838. return isOpenMPNestingDistributeDirective(Kind) ||
  839. Kind == OMPD_teams_distribute || Kind == OMPD_teams_distribute_simd ||
  840. Kind == OMPD_teams_distribute_parallel_for_simd ||
  841. Kind == OMPD_teams_distribute_parallel_for ||
  842. Kind == OMPD_target_teams_distribute ||
  843. Kind == OMPD_target_teams_distribute_parallel_for ||
  844. Kind == OMPD_target_teams_distribute_parallel_for_simd ||
  845. Kind == OMPD_target_teams_distribute_simd;
  846. }
  847. bool clang::isOpenMPPrivate(OpenMPClauseKind Kind) {
  848. return Kind == OMPC_private || Kind == OMPC_firstprivate ||
  849. Kind == OMPC_lastprivate || Kind == OMPC_linear ||
  850. Kind == OMPC_reduction || Kind == OMPC_task_reduction ||
  851. Kind == OMPC_in_reduction; // TODO add next clauses like 'reduction'.
  852. }
  853. bool clang::isOpenMPThreadPrivate(OpenMPClauseKind Kind) {
  854. return Kind == OMPC_threadprivate || Kind == OMPC_copyin;
  855. }
  856. bool clang::isOpenMPTaskingDirective(OpenMPDirectiveKind Kind) {
  857. return Kind == OMPD_task || isOpenMPTaskLoopDirective(Kind);
  858. }
  859. bool clang::isOpenMPLoopBoundSharingDirective(OpenMPDirectiveKind Kind) {
  860. return Kind == OMPD_distribute_parallel_for ||
  861. Kind == OMPD_distribute_parallel_for_simd ||
  862. Kind == OMPD_teams_distribute_parallel_for_simd ||
  863. Kind == OMPD_teams_distribute_parallel_for ||
  864. Kind == OMPD_target_teams_distribute_parallel_for ||
  865. Kind == OMPD_target_teams_distribute_parallel_for_simd;
  866. }
  867. void clang::getOpenMPCaptureRegions(
  868. SmallVectorImpl<OpenMPDirectiveKind> &CaptureRegions,
  869. OpenMPDirectiveKind DKind) {
  870. assert(DKind <= OMPD_unknown);
  871. switch (DKind) {
  872. case OMPD_parallel:
  873. case OMPD_parallel_for:
  874. case OMPD_parallel_for_simd:
  875. case OMPD_parallel_sections:
  876. case OMPD_distribute_parallel_for:
  877. case OMPD_distribute_parallel_for_simd:
  878. CaptureRegions.push_back(OMPD_parallel);
  879. break;
  880. case OMPD_target_teams:
  881. case OMPD_target_teams_distribute:
  882. case OMPD_target_teams_distribute_simd:
  883. CaptureRegions.push_back(OMPD_task);
  884. CaptureRegions.push_back(OMPD_target);
  885. CaptureRegions.push_back(OMPD_teams);
  886. break;
  887. case OMPD_teams:
  888. case OMPD_teams_distribute:
  889. case OMPD_teams_distribute_simd:
  890. CaptureRegions.push_back(OMPD_teams);
  891. break;
  892. case OMPD_target:
  893. case OMPD_target_simd:
  894. CaptureRegions.push_back(OMPD_task);
  895. CaptureRegions.push_back(OMPD_target);
  896. break;
  897. case OMPD_teams_distribute_parallel_for:
  898. case OMPD_teams_distribute_parallel_for_simd:
  899. CaptureRegions.push_back(OMPD_teams);
  900. CaptureRegions.push_back(OMPD_parallel);
  901. break;
  902. case OMPD_target_parallel:
  903. case OMPD_target_parallel_for:
  904. case OMPD_target_parallel_for_simd:
  905. CaptureRegions.push_back(OMPD_task);
  906. CaptureRegions.push_back(OMPD_target);
  907. CaptureRegions.push_back(OMPD_parallel);
  908. break;
  909. case OMPD_task:
  910. case OMPD_target_enter_data:
  911. case OMPD_target_exit_data:
  912. case OMPD_target_update:
  913. CaptureRegions.push_back(OMPD_task);
  914. break;
  915. case OMPD_taskloop:
  916. case OMPD_taskloop_simd:
  917. CaptureRegions.push_back(OMPD_taskloop);
  918. break;
  919. case OMPD_target_teams_distribute_parallel_for:
  920. case OMPD_target_teams_distribute_parallel_for_simd:
  921. CaptureRegions.push_back(OMPD_task);
  922. CaptureRegions.push_back(OMPD_target);
  923. CaptureRegions.push_back(OMPD_teams);
  924. CaptureRegions.push_back(OMPD_parallel);
  925. break;
  926. case OMPD_simd:
  927. case OMPD_for:
  928. case OMPD_for_simd:
  929. case OMPD_sections:
  930. case OMPD_section:
  931. case OMPD_single:
  932. case OMPD_master:
  933. case OMPD_critical:
  934. case OMPD_taskgroup:
  935. case OMPD_distribute:
  936. case OMPD_ordered:
  937. case OMPD_atomic:
  938. case OMPD_target_data:
  939. case OMPD_distribute_simd:
  940. CaptureRegions.push_back(OMPD_unknown);
  941. break;
  942. case OMPD_threadprivate:
  943. case OMPD_taskyield:
  944. case OMPD_barrier:
  945. case OMPD_taskwait:
  946. case OMPD_cancellation_point:
  947. case OMPD_cancel:
  948. case OMPD_flush:
  949. case OMPD_declare_reduction:
  950. case OMPD_declare_simd:
  951. case OMPD_declare_target:
  952. case OMPD_end_declare_target:
  953. case OMPD_requires:
  954. llvm_unreachable("OpenMP Directive is not allowed");
  955. case OMPD_unknown:
  956. llvm_unreachable("Unknown OpenMP directive");
  957. }
  958. }