TheRock/rocm-libraries/shared/rocroller/test/unit/GPUArchitectureGeneratorTest.cpp

// Copyright Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT

#include <GPUArchitectureGenerator/GPUArchitectureGenerator.hpp>
#include <GPUArchitectureGenerator/GPUArchitectureGenerator_defs.hpp>

#include "SimpleFixture.hpp"

class GPUArchitectureGeneratorTest : public SimpleFixture
{
protected:
    void SetUp() override
    {
        GPUArchitectureGenerator::FillArchitectures();
    }
};

TEST_F(GPUArchitectureGeneratorTest, BasicYAML)
{
    GPUArchitectureGenerator::GenerateFile("output.yaml", true);

    std::ifstream     generated_source_file("output.yaml");
    std::stringstream ss;
    ss << generated_source_file.rdbuf();
    std::string generated_source = ss.str();
    EXPECT_NE(generated_source, "");

    auto readback = rocRoller::GPUArchitecture::readYaml("output.yaml");
    EXPECT_EQ(readback.size(), rocRoller::SupportedArchitectures.size());
    for(auto& x : readback)
    {
        EXPECT_TRUE(x.second.HasCapability("SupportedISA"));
    }
    std::remove("output.yaml");
}

TEST_F(GPUArchitectureGeneratorTest, BasicMsgpack)
{
    GPUArchitectureGenerator::GenerateFile("output.msgpack");

    std::ifstream     generated_source_file("output.msgpack");
    std::stringstream ss;
    ss << generated_source_file.rdbuf();
    std::string generated_source = ss.str();
    EXPECT_NE(generated_source, "");

    auto readback = rocRoller::GPUArchitecture::readMsgpack("output.msgpack");
    EXPECT_EQ(readback.size(), rocRoller::SupportedArchitectures.size());
    for(auto& x : readback)
    {
        EXPECT_TRUE(x.second.HasCapability("SupportedISA"));
    }
    std::remove("output.msgpack");
}