[CUDA] Use Triple::isNVPTX() instead of enumerating nvptx && nvptx64.
Summary: No functional changes.
Subscribers: tra, echristo, jhen, cfe-commits
Differential Revision: http://reviews.llvm.org/D16495
llvm-svn: 258640
diff --git a/clang/lib/Driver/Tools.cpp b/clang/lib/Driver/Tools.cpp
index 24a726b..6a1517f 100644
--- a/clang/lib/Driver/Tools.cpp
+++ b/clang/lib/Driver/Tools.cpp
@@ -10660,8 +10660,7 @@
const char *LinkingOutput) const {
const auto &TC =
static_cast<const toolchains::CudaToolChain &>(getToolChain());
- assert(TC.getArch() == llvm::Triple::nvptx ||
- TC.getArch() == llvm::Triple::nvptx64);
+ assert(TC.getTriple().isNVPTX() && "Wrong platform");
std::vector<std::string> gpu_archs =
Args.getAllArgValues(options::OPT_march_EQ);
@@ -10729,8 +10728,7 @@
const char *LinkingOutput) const {
const auto &TC =
static_cast<const toolchains::CudaToolChain &>(getToolChain());
- assert(TC.getArch() == llvm::Triple::nvptx ||
- TC.getArch() == llvm::Triple::nvptx64);
+ assert(TC.getTriple().isNVPTX() && "Wrong platform");
ArgStringList CmdArgs;
CmdArgs.push_back("--cuda");