tree be4a07238a62581b3ee13f3e7e2208b8285e687b
parent eb2440207f14db79cb0cadaca5386ecd0cd35656
author Shilei Tian <tianshilei1992@gmail.com> 1612037681 -0500
committer Copybara-Service <copybara-worker@google.com> 1612524221 -0800

[OpenMP][NVPTX] Refined CMake logic to choose compute capabilites

This patch refines the logic to choose compute capabilites via the
environment variable `LIBOMPTARGET_NVPTX_COMPUTE_CAPABILITIES`. It supports the
following values (all case insensitive):
- "all": Build `deviceRTLs` for all supported compute capabilites;
- "auto": Only build for the compute capability auto detected. Note that this
  requires CUDA. If CUDA is not found, a CMake fatal error will be raised.
- "xx,yy" or "xx;yy": Build for compute capabilities `xx` and `yy`.

If `LIBOMPTARGET_NVPTX_COMPUTE_CAPABILITIES` is not set, it is equivalent to set
it to `all`.

Reviewed By: jdoerfert

Differential Revision: https://reviews.llvm.org/D95687

GitOrigin-RevId: 26d38f6d20ff137d89cb7c891b739662de1ca508