763bc9249c
Update the Chipset struct to follow the `IsaVersion` definition from llvm's `TargetParser`. This is a follow up to https://github.com/llvm/llvm-project/pull/106169#discussion_r1733955012. * Add the stepping version. Note: This may break downstream code that compares against the minor version directly. * Use comparisons with full Chipset version where possible. Note that we can't use the code in `TargetParser` directly because the chipset utility is outside of `mlir/Target` that re-exports llvm's target library.
67 lines
2.3 KiB
C++
67 lines
2.3 KiB
C++
//===- AMDGPUUtilsTest.cpp - Unit tests for AMDGPU dialect utils ----------===//
|
|
//
|
|
// 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 "mlir/Dialect/AMDGPU/Utils/Chipset.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace mlir::amdgpu {
|
|
namespace {
|
|
|
|
TEST(ChipsetTest, Parsing) {
|
|
FailureOr<Chipset> chipset = Chipset::parse("gfx90a");
|
|
ASSERT_TRUE(succeeded(chipset));
|
|
EXPECT_EQ(chipset->majorVersion, 9u);
|
|
EXPECT_EQ(chipset->minorVersion, 0u);
|
|
EXPECT_EQ(chipset->steppingVersion, 0xau);
|
|
|
|
chipset = Chipset::parse("gfx940");
|
|
ASSERT_TRUE(succeeded(chipset));
|
|
EXPECT_EQ(chipset->majorVersion, 9u);
|
|
EXPECT_EQ(chipset->minorVersion, 4u);
|
|
EXPECT_EQ(chipset->steppingVersion, 0u);
|
|
|
|
chipset = Chipset::parse("gfx1103");
|
|
ASSERT_TRUE(succeeded(chipset));
|
|
EXPECT_EQ(chipset->majorVersion, 11u);
|
|
EXPECT_EQ(chipset->minorVersion, 0u);
|
|
EXPECT_EQ(chipset->steppingVersion, 3u);
|
|
}
|
|
|
|
TEST(ChipsetTest, ParsingInvalid) {
|
|
EXPECT_TRUE(failed(Chipset::parse("navi33")));
|
|
EXPECT_TRUE(failed(Chipset::parse("rdna2")));
|
|
EXPECT_TRUE(failed(Chipset::parse("sm_80")));
|
|
EXPECT_TRUE(failed(Chipset::parse("GFX940")));
|
|
EXPECT_TRUE(failed(Chipset::parse("Gfx940")));
|
|
EXPECT_TRUE(failed(Chipset::parse("gfx9")));
|
|
EXPECT_TRUE(failed(Chipset::parse("gfx_940")));
|
|
EXPECT_TRUE(failed(Chipset::parse("gfx940_")));
|
|
EXPECT_TRUE(failed(Chipset::parse("gfxmeow")));
|
|
EXPECT_TRUE(failed(Chipset::parse("gfx1fff")));
|
|
}
|
|
|
|
TEST(ChipsetTest, Comparison) {
|
|
EXPECT_EQ(Chipset(9, 4, 0), Chipset(9, 4, 0));
|
|
EXPECT_NE(Chipset(9, 4, 0), Chipset(9, 4, 2));
|
|
EXPECT_NE(Chipset(9, 0, 0), Chipset(10, 0, 0));
|
|
|
|
EXPECT_LT(Chipset(9, 0, 0), Chipset(10, 0, 0));
|
|
EXPECT_LT(Chipset(9, 0, 0), Chipset(9, 4, 2));
|
|
EXPECT_LE(Chipset(9, 4, 1), Chipset(9, 4, 1));
|
|
EXPECT_FALSE(Chipset(9, 4, 2) < Chipset(9, 4, 2));
|
|
EXPECT_FALSE(Chipset(9, 4, 2) < Chipset(9, 4, 0));
|
|
|
|
EXPECT_GT(Chipset(9, 0, 0xa), Chipset(9, 0, 8));
|
|
EXPECT_GE(Chipset(9, 0, 0xa), Chipset(9, 0, 0xa));
|
|
EXPECT_FALSE(Chipset(9, 4, 1) >= Chipset(9, 4, 2));
|
|
EXPECT_FALSE(Chipset(9, 0, 0xa) >= Chipset(9, 4, 0));
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace mlir::amdgpu
|