6ca4be641c
Drop the IsNVIDIAOffloadArch/IsAMDOffloadArch/IsIntel*OffloadArch free functions in favor of the OffloadArch member predicate functions. Co-authored-by: Claude (Claude-Opus-4.8) <noreply@anthropic.com>
69 lines
2.2 KiB
C++
69 lines
2.2 KiB
C++
//===- unittests/Basic/OffloadArchTest.cpp - Test OffloadArch -------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Basic/OffloadArch.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace clang;
|
|
|
|
static OffloadArch parse(llvm::StringRef S) { return StringToOffloadArch(S); }
|
|
|
|
TEST(OffloadArchTest, TargetArchClassification) {
|
|
OffloadArch NV = parse("sm_120a");
|
|
EXPECT_TRUE(parse("sm_20").isNVPTX());
|
|
EXPECT_TRUE(NV.isNVPTX());
|
|
EXPECT_FALSE(NV.isAMDGPU());
|
|
|
|
EXPECT_TRUE(parse("gfx600").isAMDGPU());
|
|
EXPECT_TRUE(parse("gfx1201").isAMDGPU());
|
|
EXPECT_TRUE(parse("gfx12-generic").isAMDGPU());
|
|
EXPECT_FALSE(parse("gfx600").isNVPTX());
|
|
|
|
OffloadArch SPIRV = parse("amdgcnspirv");
|
|
EXPECT_FALSE(SPIRV.isAMDGPU());
|
|
EXPECT_TRUE(SPIRV.isSPIRV());
|
|
|
|
OffloadArch IntelCPU = parse("graniterapids");
|
|
EXPECT_FALSE(IntelCPU.isAMDGPU());
|
|
EXPECT_FALSE(IntelCPU.isSPIRV());
|
|
EXPECT_TRUE(IntelCPU.isIntel());
|
|
EXPECT_TRUE(IntelCPU.isIntelCPU());
|
|
EXPECT_FALSE(IntelCPU.isIntelGPU());
|
|
|
|
OffloadArch IntelGPU = parse("bmg_g21");
|
|
EXPECT_TRUE(IntelGPU.isIntel());
|
|
EXPECT_FALSE(IntelGPU.isIntelCPU());
|
|
EXPECT_TRUE(IntelGPU.isIntelGPU());
|
|
|
|
OffloadArch Generic = parse("generic");
|
|
EXPECT_FALSE(Generic.isNVPTX());
|
|
EXPECT_FALSE(Generic.isAMDGPU());
|
|
EXPECT_FALSE(Generic.isSPIRV());
|
|
EXPECT_FALSE(Generic.isIntel());
|
|
}
|
|
|
|
TEST(OffloadArchTest, Unknown) {
|
|
EXPECT_TRUE(parse("not-a-real-arch").isUnknown());
|
|
EXPECT_TRUE(parse("").isUnused());
|
|
}
|
|
|
|
// Names must round-trip through parse -> string.
|
|
TEST(OffloadArchTest, RoundTrip) {
|
|
for (const char *Name :
|
|
{"sm_52", "sm_90a", "gfx906", "gfx1201", "gfx12-generic", "amdgcnspirv",
|
|
"graniterapids", "bmg_g21", "generic"}) {
|
|
OffloadArch A = parse(Name);
|
|
EXPECT_FALSE(A.isUnknown()) << Name;
|
|
EXPECT_STREQ(OffloadArchToString(A), Name);
|
|
}
|
|
}
|
|
|
|
TEST(OffloadArchTest, Defaults) {
|
|
EXPECT_STREQ(OffloadArchToString(OffloadArch::CudaDefault()), "sm_52");
|
|
EXPECT_STREQ(OffloadArchToString(OffloadArch::HIPDefault()), "gfx906");
|
|
}
|