CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
1574 lines
48 KiB
C++
1574 lines
48 KiB
C++
/*******************************************************************************
|
|
*
|
|
* MIT License
|
|
*
|
|
* Copyright (c) 2017 Advanced Micro Devices, Inc.
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*
|
|
*******************************************************************************/
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <miopen/filesystem.hpp>
|
|
#include <miopen/db.hpp>
|
|
#include <miopen/db_record.hpp>
|
|
#include <miopen/lock_file.hpp>
|
|
#include <miopen/process.hpp>
|
|
#include <miopen/ramdb.hpp>
|
|
#include <miopen/readonlyramdb.hpp>
|
|
#include <miopen/returns.hpp>
|
|
#include <miopen/temp_file.hpp>
|
|
|
|
#include <array>
|
|
#include <cstdio>
|
|
#include <fstream>
|
|
#include <mutex>
|
|
#include <limits>
|
|
#include <optional>
|
|
#include <random>
|
|
#include <shared_mutex>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <system_error>
|
|
#include <thread>
|
|
#include <vector>
|
|
|
|
namespace {
|
|
using namespace miopen;
|
|
|
|
#if MIOPEN_EMBED_DB
|
|
struct TestRordbEmbedFsOverrideLock
|
|
{
|
|
TestRordbEmbedFsOverrideLock() : cached(debug::rordb_embed_fs_override())
|
|
{
|
|
debug::rordb_embed_fs_override() = true;
|
|
}
|
|
|
|
~TestRordbEmbedFsOverrideLock() { debug::rordb_embed_fs_override() = cached; }
|
|
|
|
private:
|
|
bool cached;
|
|
};
|
|
#endif
|
|
|
|
static fs::path& exe_path()
|
|
{
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static fs::path exe_path;
|
|
return exe_path;
|
|
}
|
|
|
|
static std::optional<fs::path>& thread_logs_root()
|
|
{
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static std::mutex mutex;
|
|
std::lock_guard<std::mutex> lock(mutex);
|
|
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static std::optional<fs::path> path;
|
|
return path;
|
|
}
|
|
|
|
static bool& full_set()
|
|
{
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static bool full_set = false;
|
|
return full_set;
|
|
}
|
|
|
|
struct ArgsHelper
|
|
{
|
|
static constexpr const char* logs_path_arg = "thread-logs-root";
|
|
static constexpr const char* write_arg = "mp-test-child-write";
|
|
static constexpr const char* id_arg = "mp-test-child";
|
|
static constexpr const char* path_arg = "mp-test-child-path";
|
|
static constexpr const char* db_class_arg = "mp-test-child-db-class";
|
|
|
|
// Exact gtest filter for spawning child worker processes.
|
|
// Must match the CPU_PerfDb_ChildWorker_NONE.ProcessWork test case
|
|
static constexpr const char* child_worker_filter = "CPU_PerfDb_ChildWorker_NONE.ProcessWork";
|
|
|
|
struct db_class
|
|
{
|
|
static constexpr const char* db = "db";
|
|
static constexpr const char* ramdb = "ramdb";
|
|
|
|
template <class TDb>
|
|
static constexpr std::enable_if_t<std::is_same<TDb, PlainTextDb>::value, const char*> Get()
|
|
{
|
|
return db;
|
|
}
|
|
|
|
template <class TDb>
|
|
static constexpr std::enable_if_t<std::is_same<TDb, RamDb>::value, const char*> Get()
|
|
{
|
|
return ramdb;
|
|
}
|
|
};
|
|
};
|
|
|
|
class Random
|
|
{
|
|
public:
|
|
Random(unsigned int seed = 0) : rng(seed) {}
|
|
|
|
std::mt19937::result_type Next() { return dist(rng); }
|
|
|
|
private:
|
|
std::mt19937 rng;
|
|
std::uniform_int_distribution<std::mt19937::result_type> dist;
|
|
};
|
|
|
|
struct TestData
|
|
{
|
|
int x;
|
|
int y;
|
|
|
|
struct NoInit
|
|
{
|
|
};
|
|
|
|
TestData(NoInit) : x(0), y(0) {}
|
|
TestData(Random& rnd) : x(rnd.Next()), y(rnd.Next()) {}
|
|
TestData() : x(Rnd().Next()), y(Rnd().Next()) {}
|
|
TestData(int x_, int y_) : x(x_), y(y_) {}
|
|
|
|
template <unsigned int seed>
|
|
static TestData Seeded()
|
|
{
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static Random rnd(seed);
|
|
return {static_cast<int>(rnd.Next()), static_cast<int>(rnd.Next())};
|
|
}
|
|
|
|
template <class TSelf, class Visitor>
|
|
static void VisitAll(TSelf&& self, Visitor visitor)
|
|
{
|
|
visitor(self.x, "x");
|
|
visitor(self.y, "y");
|
|
}
|
|
|
|
void Serialize(std::ostream& s) const
|
|
{
|
|
static const auto sep = 'x';
|
|
s << x << sep << y;
|
|
}
|
|
|
|
bool Deserialize(const std::string& s)
|
|
{
|
|
static const auto sep = 'x';
|
|
TestData t(NoInit{});
|
|
std::istringstream ss(s);
|
|
|
|
const auto success = DeserializeField(ss, &t.x, sep) && DeserializeField(ss, &t.y, sep);
|
|
|
|
if(!success)
|
|
return false;
|
|
|
|
*this = t;
|
|
return true;
|
|
}
|
|
|
|
bool operator==(const TestData& other) const { return x == other.x && y == other.y; }
|
|
|
|
private:
|
|
static bool DeserializeField(std::istream& from, int* ret, char separator)
|
|
{
|
|
std::string part;
|
|
|
|
if(!std::getline(from, part, separator))
|
|
return false;
|
|
|
|
const auto start = part.c_str();
|
|
char* end;
|
|
const auto value = std::strtol(start, &end, 10);
|
|
|
|
if(start == end)
|
|
return false;
|
|
|
|
*ret = value;
|
|
return true;
|
|
}
|
|
|
|
static Random& Rnd()
|
|
{
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static Random rnd;
|
|
return rnd;
|
|
}
|
|
};
|
|
|
|
std::ostream& operator<<(std::ostream& s, const TestData& td)
|
|
{
|
|
s << "x: " << td.x << ", y: " << td.y;
|
|
return s;
|
|
}
|
|
|
|
class DbTest
|
|
{
|
|
public:
|
|
DbTest(TempFile& temp_file_) : temp_file(temp_file_) { ResetDb(); }
|
|
|
|
virtual ~DbTest()
|
|
{
|
|
std::error_code ec;
|
|
fs::remove(LockFilePath(temp_file.Path()), ec);
|
|
}
|
|
|
|
protected:
|
|
TempFile& temp_file;
|
|
|
|
static const std::array<std::pair<const std::string, TestData>, 2>& common_data()
|
|
{
|
|
static const std::array<std::pair<const std::string, TestData>, 2> data{{
|
|
{id1(), value1()},
|
|
{id0(), value0()},
|
|
}};
|
|
|
|
return data;
|
|
}
|
|
|
|
static void ResetDbFile(TempFile& tmp_file) { tmp_file = TempFile{tmp_file.GetPathInfix()}; }
|
|
|
|
void ResetDb() { ResetDbFile(temp_file); }
|
|
|
|
static const TestData& key()
|
|
{
|
|
static const TestData data(1, 2);
|
|
return data;
|
|
}
|
|
|
|
static const TestData& value0()
|
|
{
|
|
static const TestData data(3, 4);
|
|
return data;
|
|
}
|
|
|
|
static const TestData& value1()
|
|
{
|
|
static const TestData data(5, 6);
|
|
return data;
|
|
}
|
|
|
|
static const TestData& value2()
|
|
{
|
|
static const TestData data(7, 8);
|
|
return data;
|
|
}
|
|
|
|
static const std::string& id0()
|
|
{
|
|
static const std::string id0_ = "0";
|
|
return id0_;
|
|
}
|
|
static const std::string& id1()
|
|
{
|
|
static const std::string id1_ = "1";
|
|
return id1_;
|
|
}
|
|
static const std::string& id2()
|
|
{
|
|
static const std::string id2_ = "2";
|
|
return id2_;
|
|
}
|
|
static const std::string& missing_id()
|
|
{
|
|
static const std::string missing_id_ = "3";
|
|
return missing_id_;
|
|
}
|
|
|
|
template <class TKey, class TValue, size_t count>
|
|
static void RawWrite(const fs::path& db_path,
|
|
const TKey& key,
|
|
const std::array<std::pair<const std::string, TValue>, count> values)
|
|
{
|
|
std::ostringstream ss_vals;
|
|
ss_vals << key.x << 'x' << key.y << '=';
|
|
|
|
auto first = true;
|
|
|
|
for(const auto& id_value : values)
|
|
{
|
|
if(!first)
|
|
ss_vals << ";";
|
|
|
|
first = false;
|
|
ss_vals << id_value.first << ':' << id_value.second.x << 'x' << id_value.second.y;
|
|
}
|
|
|
|
std::ofstream(db_path, std::ios::out | std::ios::ate) << ss_vals.str() << std::endl;
|
|
}
|
|
|
|
template <class TDb, class TKey, class TValue, size_t count>
|
|
static void ValidateSingleEntry(
|
|
TKey key, const std::array<std::pair<const std::string, TValue>, count> values, TDb& db)
|
|
{
|
|
auto record = db.FindRecord(key);
|
|
|
|
ASSERT_TRUE(record);
|
|
|
|
for(const auto& id_value : values)
|
|
{
|
|
TValue read;
|
|
ASSERT_TRUE(record->GetValues(id_value.first, read));
|
|
ASSERT_EQ(id_value.second, read);
|
|
}
|
|
}
|
|
|
|
static fs::path LockFilePath(const fs::path& db_path) { return db_path + ".test.lock"; }
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbFindTest : public DbTest
|
|
{
|
|
public:
|
|
DbFindTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for reading premade file by FindRecord...");
|
|
|
|
RawWrite(temp_file, key(), common_data());
|
|
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
ValidateSingleEntry(key(), common_data(), db);
|
|
|
|
const TestData invalid_key(100, 200);
|
|
auto record1 = db.FindRecord(invalid_key);
|
|
ASSERT_FALSE(record1);
|
|
}
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbStoreTest : public DbTest
|
|
{
|
|
public:
|
|
DbStoreTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for reading stored data...");
|
|
|
|
DbRecord record(DbKinds::PerfDb, key());
|
|
ASSERT_TRUE(record.SetValues(id0(), value0()));
|
|
ASSERT_TRUE(record.SetValues(id1(), value1()));
|
|
|
|
{
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
|
|
ASSERT_TRUE(db.StoreRecord(record));
|
|
}
|
|
|
|
std::string read;
|
|
ASSERT_TRUE(std::getline(std::ifstream(temp_file.Path()), read).good());
|
|
|
|
TDb db{DbKinds::PerfDb, temp_file};
|
|
ValidateSingleEntry(key(), common_data(), db);
|
|
}
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbUpdateTest : public DbTest
|
|
{
|
|
public:
|
|
DbUpdateTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for updating existing records...");
|
|
|
|
DbRecord record0(DbKinds::PerfDb, key());
|
|
ASSERT_TRUE(record0.SetValues(id0(), value0()));
|
|
|
|
{
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
|
|
ASSERT_TRUE(db.StoreRecord(record0));
|
|
}
|
|
|
|
DbRecord record1(DbKinds::PerfDb, key());
|
|
ASSERT_TRUE(record1.SetValues(id1(), value1()));
|
|
|
|
{
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
|
|
ASSERT_TRUE(db.UpdateRecord(record1));
|
|
}
|
|
|
|
TestData read0, read1;
|
|
ASSERT_TRUE(record1.GetValues(id0(), read0));
|
|
ASSERT_TRUE(record1.GetValues(id1(), read1));
|
|
ASSERT_EQ(value0(), read0);
|
|
ASSERT_EQ(value1(), read1);
|
|
|
|
TDb db{DbKinds::PerfDb, temp_file};
|
|
ValidateSingleEntry(key(), common_data(), db);
|
|
}
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbRemoveTest : public DbTest
|
|
{
|
|
public:
|
|
DbRemoveTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for removing records...");
|
|
|
|
DbRecord record(DbKinds::PerfDb, key());
|
|
ASSERT_TRUE(record.SetValues(id0(), value0()));
|
|
ASSERT_TRUE(record.SetValues(id1(), value1()));
|
|
|
|
{
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
|
|
ASSERT_TRUE(db.StoreRecord(record));
|
|
}
|
|
|
|
{
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
|
|
ASSERT_TRUE(db.FindRecord(key()));
|
|
ASSERT_TRUE(db.RemoveRecord(key()));
|
|
ASSERT_FALSE(db.FindRecord(key()));
|
|
}
|
|
}
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbReadTest : public DbTest
|
|
{
|
|
public:
|
|
DbReadTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for reading premade file by Load...");
|
|
|
|
RawWrite(temp_file, key(), common_data());
|
|
TDb db{DbKinds::PerfDb, temp_file};
|
|
ValidateSingleEntry(key(), common_data(), db);
|
|
}
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbWriteTest : public DbTest
|
|
{
|
|
public:
|
|
DbWriteTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for storing unexistent records by update...");
|
|
|
|
{
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
|
|
ASSERT_TRUE(db.Update(key(), id0(), value0()));
|
|
ASSERT_TRUE(db.Update(key(), id1(), value1()));
|
|
}
|
|
|
|
std::string read;
|
|
ASSERT_TRUE(std::getline(std::ifstream(temp_file.Path()), read).good());
|
|
|
|
TDb db{DbKinds::PerfDb, temp_file};
|
|
ValidateSingleEntry(key(), common_data(), db);
|
|
}
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbOperationsTest : public DbTest
|
|
{
|
|
public:
|
|
DbOperationsTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing different " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " operations...");
|
|
|
|
const TestData to_be_rewritten(7, 8);
|
|
|
|
{
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
|
|
ASSERT_TRUE(db.Update(key(), id0(), to_be_rewritten));
|
|
ASSERT_TRUE(db.Update(key(), id1(), to_be_rewritten));
|
|
ASSERT_TRUE(db.Update(key(), id1(), value1()));
|
|
|
|
// Rewritting existing value with same one. expecting no DB manipulations performed
|
|
ASSERT_TRUE(db.Update(key(), id1(), value1()));
|
|
}
|
|
|
|
{
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
ASSERT_TRUE(db.Update(key(), id0(), value0()));
|
|
}
|
|
|
|
{
|
|
TestData read0, read1, read_missing;
|
|
const auto read_missing_cmp(read_missing);
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
|
|
ASSERT_FALSE(db.Load(key(), missing_id(), read_missing));
|
|
|
|
// value should not be changed.
|
|
ASSERT_EQ(read_missing, read_missing_cmp);
|
|
|
|
ASSERT_TRUE(db.Load(key(), id0(), read0));
|
|
ASSERT_TRUE(db.Load(key(), id1(), read1));
|
|
|
|
ASSERT_EQ(read0, value0());
|
|
ASSERT_EQ(read1, value1());
|
|
|
|
ASSERT_TRUE(db.Remove(key(), id0()));
|
|
|
|
read0 = read_missing_cmp;
|
|
|
|
ASSERT_FALSE(db.Load(key(), id0(), read0));
|
|
ASSERT_TRUE(db.Load(key(), id1(), read1));
|
|
|
|
ASSERT_EQ(read0, read_missing_cmp);
|
|
ASSERT_EQ(read1, value1());
|
|
|
|
ASSERT_FALSE(db.Remove(key(), id0()));
|
|
}
|
|
|
|
{
|
|
TestData read0, read1;
|
|
const auto read_missing_cmp(read0);
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
|
|
ASSERT_FALSE(db.Load(key(), id0(), read0));
|
|
ASSERT_TRUE(db.Load(key(), id1(), read1));
|
|
|
|
ASSERT_EQ(read0, read_missing_cmp);
|
|
ASSERT_EQ(read1, value1());
|
|
}
|
|
}
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbParallelTest : public DbTest
|
|
{
|
|
public:
|
|
DbParallelTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(
|
|
LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for using two objects targeting one file existing in one scope...");
|
|
|
|
{
|
|
TDb db(DbKinds::PerfDb, temp_file);
|
|
ASSERT_TRUE(db.Update(key(), id0(), value0()));
|
|
}
|
|
|
|
{
|
|
TDb db0(DbKinds::PerfDb, temp_file);
|
|
TDb db1(DbKinds::PerfDb, temp_file);
|
|
|
|
auto r0 = db0.FindRecord(key());
|
|
auto r1 = db1.FindRecord(key());
|
|
|
|
ASSERT_TRUE(r0);
|
|
ASSERT_TRUE(r1);
|
|
|
|
ASSERT_TRUE(r0->SetValues(id1(), value1()));
|
|
ASSERT_TRUE(r1->SetValues(id2(), value2()));
|
|
|
|
ASSERT_TRUE(db0.UpdateRecord(*r0));
|
|
ASSERT_TRUE(db1.UpdateRecord(*r1));
|
|
}
|
|
|
|
const std::array<std::pair<const std::string, TestData>, 3> data{{
|
|
{id0(), value0()},
|
|
{id1(), value1()},
|
|
{id2(), value2()},
|
|
}};
|
|
|
|
TDb db{DbKinds::PerfDb, temp_file};
|
|
ValidateSingleEntry(key(), data, db);
|
|
}
|
|
};
|
|
|
|
class DBMultiThreadedTestWork
|
|
{
|
|
public:
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static unsigned int threads_count;
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static unsigned int common_part_size;
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static unsigned int unique_part_size;
|
|
static constexpr unsigned int ids_per_key = 16;
|
|
static constexpr unsigned int common_part_seed = 435345;
|
|
|
|
static const std::vector<TestData>& common_part()
|
|
{
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static std::mutex mutex;
|
|
std::lock_guard<std::mutex> lock(mutex);
|
|
|
|
static const auto& ref = common_part_init();
|
|
return ref;
|
|
}
|
|
|
|
static void Initialize() { (void)common_part(); }
|
|
|
|
template <class TDbConstructor>
|
|
static void
|
|
WorkItem(unsigned int id, const TDbConstructor& db_constructor, const std::string& log_postfix)
|
|
{
|
|
RedirectLogs(id, log_postfix, [id, &db_constructor]() {
|
|
CommonPart(db_constructor);
|
|
UniquePart(id, db_constructor);
|
|
});
|
|
}
|
|
|
|
template <class TDbConstructor>
|
|
static void ReadWorkItem(unsigned int id,
|
|
const TDbConstructor& db_constructor,
|
|
const std::string& log_postfix)
|
|
{
|
|
RedirectLogs(id, log_postfix, [&db_constructor]() { ReadCommonPart(db_constructor); });
|
|
}
|
|
|
|
template <class TDbConstructor>
|
|
static void FillForReading(const TDbConstructor& db_constructor)
|
|
{
|
|
decltype(db_constructor()) db = db_constructor();
|
|
CommonPartSection(0u, common_part_size, [&db]() -> decltype(db) { return db; });
|
|
}
|
|
|
|
template <class TDbConstructor>
|
|
static void ValidateCommonPart(const TDbConstructor& db_constructor)
|
|
{
|
|
decltype(db_constructor()) db = db_constructor();
|
|
const auto cp = common_part();
|
|
|
|
for(auto i = 0u; i < common_part_size; i++)
|
|
{
|
|
const auto key = std::to_string(i / ids_per_key);
|
|
const auto id = std::to_string(i % ids_per_key);
|
|
const auto data = cp[i];
|
|
TestData read(TestData::NoInit{});
|
|
|
|
ASSERT_TRUE(db.Load(key, id, read));
|
|
ASSERT_EQ(read, data);
|
|
}
|
|
}
|
|
|
|
private:
|
|
template <class TWorker>
|
|
static void RedirectLogs(unsigned int id, const std::string& log_postfix, const TWorker& worker)
|
|
{
|
|
std::ofstream log;
|
|
std::ofstream log_err;
|
|
std::streambuf *cout_buf = nullptr, *cerr_buf = nullptr;
|
|
|
|
if(thread_logs_root().has_value())
|
|
{
|
|
// NOLINTBEGIN (bugprone-unchecked-optional-access)
|
|
const auto out_path = thread_logs_root().value() /
|
|
("thread-" + std::to_string(id) + "_" + log_postfix + ".log");
|
|
const auto err_path = thread_logs_root().value() /
|
|
("thread-" + std::to_string(id) + "_" + log_postfix + "-err.log");
|
|
// NOLINTEND (bugprone-unchecked-optional-access)
|
|
|
|
fs::remove(out_path);
|
|
fs::remove(err_path);
|
|
|
|
log.open(out_path);
|
|
log_err.open(err_path);
|
|
cout_buf = std::cout.rdbuf();
|
|
cerr_buf = std::cerr.rdbuf();
|
|
std::cout.rdbuf(log.rdbuf());
|
|
std::cerr.rdbuf(log_err.rdbuf());
|
|
}
|
|
|
|
worker();
|
|
|
|
if(thread_logs_root().has_value())
|
|
{
|
|
std::cout.rdbuf(cout_buf);
|
|
std::cerr.rdbuf(cerr_buf);
|
|
}
|
|
}
|
|
|
|
template <class TDbConstructor>
|
|
static void ReadCommonPart(const TDbConstructor& db_constructor)
|
|
{
|
|
MIOPEN_LOG_CUSTOM(
|
|
LoggingLevel::Default, "Test", "Common part. Section with common db instance.");
|
|
{
|
|
decltype(db_constructor()) db = db_constructor();
|
|
ReadCommonPartSection(0u, common_part_size / 2, [&db]() -> decltype(db) { return db; });
|
|
}
|
|
|
|
MIOPEN_LOG_CUSTOM(
|
|
LoggingLevel::Default, "Test", "Common part. Section with separate db instances.");
|
|
ReadCommonPartSection(
|
|
common_part_size / 2,
|
|
common_part_size,
|
|
[&db_constructor]() -> decltype(db_constructor()) { return db_constructor(); });
|
|
}
|
|
|
|
template <class TDbGetter>
|
|
static void
|
|
ReadCommonPartSection(unsigned int start, unsigned int end, const TDbGetter& db_getter)
|
|
{
|
|
const auto cp = common_part();
|
|
|
|
for(auto i = start; i < end; i++)
|
|
{
|
|
const auto key = std::to_string(i / ids_per_key);
|
|
const auto id = std::to_string(i % ids_per_key);
|
|
const auto data = cp[i];
|
|
TestData read(TestData::NoInit{});
|
|
|
|
ASSERT_TRUE(db_getter().Load(key, id, read));
|
|
ASSERT_EQ(read, data);
|
|
}
|
|
}
|
|
|
|
template <class TDbConstructor>
|
|
static void CommonPart(const TDbConstructor& db_constructor)
|
|
{
|
|
MIOPEN_LOG_CUSTOM(
|
|
LoggingLevel::Default, "Test", "Common part. Section with common db instance.");
|
|
{
|
|
decltype(db_constructor()) db = db_constructor();
|
|
CommonPartSection(0u, common_part_size / 2, [&db]() -> decltype(db) { return db; });
|
|
}
|
|
|
|
MIOPEN_LOG_CUSTOM(
|
|
LoggingLevel::Default, "Test", "Common part. Section with separate db instances.");
|
|
CommonPartSection(
|
|
common_part_size / 2,
|
|
common_part_size,
|
|
[&db_constructor]() -> decltype(db_constructor()) { return db_constructor(); });
|
|
}
|
|
|
|
template <class TDbGetter>
|
|
static void CommonPartSection(unsigned int start, unsigned int end, const TDbGetter& db_getter)
|
|
{
|
|
const auto cp = common_part();
|
|
|
|
for(auto i = start; i < end; i++)
|
|
{
|
|
const auto key = std::to_string(i / ids_per_key);
|
|
const auto id = std::to_string(i % ids_per_key);
|
|
const auto data = cp[i];
|
|
|
|
db_getter().Update(key, id, data);
|
|
}
|
|
}
|
|
|
|
template <class TDbConstructor>
|
|
static void UniquePart(unsigned int id, const TDbConstructor& db_constructor)
|
|
{
|
|
Random rnd(123123 + id);
|
|
|
|
MIOPEN_LOG_CUSTOM(
|
|
LoggingLevel::Default, "Test", "Unique part. Section with common db instance.");
|
|
{
|
|
decltype(db_constructor()) db = db_constructor();
|
|
UniquePartSection(rnd, 0, unique_part_size / 2, [&db]() -> decltype(db) { return db; });
|
|
}
|
|
|
|
MIOPEN_LOG_CUSTOM(
|
|
LoggingLevel::Default, "Test", "Unique part. Section with separate db instances.");
|
|
UniquePartSection(
|
|
rnd,
|
|
unique_part_size / 2,
|
|
unique_part_size,
|
|
[&db_constructor]() -> decltype(db_constructor()) { return db_constructor(); });
|
|
}
|
|
|
|
template <class TDbGetter>
|
|
static void
|
|
UniquePartSection(Random& rnd, unsigned int start, unsigned int end, const TDbGetter& db_getter)
|
|
{
|
|
for(auto i = start; i < end; i++)
|
|
{
|
|
auto key = LimitedRandom(rnd, common_part_size / ids_per_key + 2);
|
|
auto id = LimitedRandom(rnd, ids_per_key + 1);
|
|
TestData data(rnd);
|
|
|
|
db_getter().Update(std::to_string(key), std::to_string(id), data);
|
|
}
|
|
}
|
|
|
|
static std::mt19937::result_type LimitedRandom(Random& rnd, std::mt19937::result_type min)
|
|
{
|
|
std::mt19937::result_type key;
|
|
|
|
do
|
|
key = rnd.Next();
|
|
while(key < min);
|
|
|
|
return key;
|
|
}
|
|
|
|
static const std::vector<TestData>& common_part_init()
|
|
{
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
static std::vector<TestData> data(common_part_size, TestData{TestData::NoInit{}});
|
|
|
|
for(auto i = 0u; i < common_part_size; i++)
|
|
data[i] = TestData::Seeded<common_part_seed>();
|
|
|
|
return data;
|
|
}
|
|
};
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
unsigned int DBMultiThreadedTestWork::threads_count = 16;
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
unsigned int DBMultiThreadedTestWork::common_part_size = 32;
|
|
// NOLINTNEXTLINE (cppcoreguidelines-avoid-non-const-global-variables)
|
|
unsigned int DBMultiThreadedTestWork::unique_part_size = 32;
|
|
|
|
// multi-threaded test classes
|
|
template <class TDb>
|
|
class DbMultiThreadedTest : public DbTest
|
|
{
|
|
public:
|
|
DbMultiThreadedTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for multithreaded write access...");
|
|
|
|
std::shared_mutex mutex;
|
|
std::vector<std::thread> threads;
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Initializing test data...");
|
|
DBMultiThreadedTestWork::Initialize();
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Launching test threads...");
|
|
threads.reserve(DBMultiThreadedTestWork::threads_count);
|
|
const auto c = [this]()
|
|
MIOPEN_RETURNS(GetDbInstance<TDb>(DbKinds::PerfDb, temp_file, false));
|
|
|
|
{
|
|
std::unique_lock<std::shared_mutex> lock(mutex);
|
|
|
|
for(auto i = 0u; i < DBMultiThreadedTestWork::threads_count; i++)
|
|
{
|
|
auto thread_body = [c, &mutex, i]() {
|
|
std::shared_lock<std::shared_mutex> lock_(mutex);
|
|
DBMultiThreadedTestWork::WorkItem(i, c, "mt");
|
|
};
|
|
|
|
threads.emplace_back(thread_body);
|
|
}
|
|
}
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Waiting for test threads...");
|
|
for(auto& thread : threads)
|
|
thread.join();
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Validating results...");
|
|
DBMultiThreadedTestWork::ValidateCommonPart(c);
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Validation passed...");
|
|
}
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbMultiThreadedReadTest : public DbTest
|
|
{
|
|
public:
|
|
DbMultiThreadedReadTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for multithreaded read access...");
|
|
|
|
std::shared_mutex mutex;
|
|
std::vector<std::thread> threads;
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Initializing test data...");
|
|
const auto c = [this]()
|
|
MIOPEN_RETURNS(GetDbInstance<TDb>(DbKinds::PerfDb, temp_file, false));
|
|
DBMultiThreadedTestWork::FillForReading(c);
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Launching test threads...");
|
|
threads.reserve(DBMultiThreadedTestWork::threads_count);
|
|
|
|
{
|
|
std::unique_lock<std::shared_mutex> lock(mutex);
|
|
|
|
for(auto i = 0u; i < DBMultiThreadedTestWork::threads_count; i++)
|
|
{
|
|
threads.emplace_back([c, &mutex, i]() {
|
|
std::shared_lock<std::shared_mutex> lock_(mutex);
|
|
DBMultiThreadedTestWork::ReadWorkItem(i, c, "mt");
|
|
});
|
|
}
|
|
}
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Waiting for test threads...");
|
|
for(auto& thread : threads)
|
|
thread.join();
|
|
}
|
|
};
|
|
|
|
// multi-process test classes
|
|
template <class TDb>
|
|
class DbMultiProcessTest : public DbTest
|
|
{
|
|
public:
|
|
DbMultiProcessTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for multiprocess write access...");
|
|
|
|
std::vector<ProcessAsync> children{};
|
|
const auto lock_file_path = LockFilePath(temp_file);
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Initializing test data...");
|
|
DBMultiThreadedTestWork::Initialize();
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Launching test processes...");
|
|
{
|
|
auto& file_lock = miopen::LockFile::Get(lock_file_path);
|
|
std::shared_lock<miopen::LockFile> lock(file_lock);
|
|
|
|
auto id = 0;
|
|
|
|
// clang-format off
|
|
for(auto i = 0u; i < DBMultiThreadedTestWork::threads_count; ++i)
|
|
{
|
|
// Use --gtest_filter to run only the child worker test in spawned process
|
|
auto args =
|
|
std::string{"--reset-sharding"
|
|
" --gtest_filter="} + ArgsHelper::child_worker_filter +
|
|
" --" + ArgsHelper::write_arg +
|
|
" --" + ArgsHelper::id_arg + " " + std::to_string(id++) +
|
|
" --" + ArgsHelper::path_arg + " " + temp_file.Path() +
|
|
" --" + ArgsHelper::db_class_arg + " " + ArgsHelper::db_class::Get<TDb>();
|
|
|
|
if(thread_logs_root().has_value())
|
|
{
|
|
// NOLINTNEXTLINE (bugprone-unchecked-optional-access)
|
|
args += std::string{" --"} + ArgsHelper::logs_path_arg + " " + thread_logs_root().value();
|
|
}
|
|
|
|
if(full_set())
|
|
args += " --all";
|
|
|
|
children.emplace_back(exe_path(), args);
|
|
}
|
|
// clang-format on
|
|
}
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Waiting for test processes...");
|
|
for(auto&& child : children)
|
|
{
|
|
ASSERT_EQ(child.Wait(), 0);
|
|
}
|
|
|
|
{
|
|
std::error_code ec;
|
|
fs::remove(lock_file_path, ec);
|
|
}
|
|
|
|
const auto c = [this]()
|
|
MIOPEN_RETURNS(GetDbInstance<TDb>(DbKinds::PerfDb, temp_file, false));
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Validating results...");
|
|
DBMultiThreadedTestWork::ValidateCommonPart(c);
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Validation passed...");
|
|
}
|
|
|
|
static void WorkItem(unsigned int id, const fs::path& db_path, bool write)
|
|
{
|
|
{
|
|
auto& file_lock = miopen::LockFile::Get(LockFilePath(db_path));
|
|
std::lock_guard<miopen::LockFile> lock(file_lock);
|
|
}
|
|
|
|
const auto c = [&db_path]()
|
|
MIOPEN_RETURNS(GetDbInstance<TDb>(DbKinds::PerfDb, db_path, false));
|
|
|
|
if(write)
|
|
DBMultiThreadedTestWork::WorkItem(id, c, "mp");
|
|
else
|
|
DBMultiThreadedTestWork::ReadWorkItem(id, c, "mp");
|
|
}
|
|
};
|
|
|
|
template <class TDb>
|
|
class DbMultiProcessReadTest : public DbTest
|
|
{
|
|
public:
|
|
DbMultiProcessReadTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing " << ArgsHelper::db_class::Get<TDb>()
|
|
<< " for multiprocess read access...");
|
|
|
|
std::vector<ProcessAsync> children{};
|
|
const auto lock_file_path = LockFilePath(temp_file);
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Initializing test data...");
|
|
const auto c = [this]()
|
|
MIOPEN_RETURNS(GetDbInstance<TDb>(DbKinds::PerfDb, temp_file, false));
|
|
DBMultiThreadedTestWork::FillForReading(c);
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Launching test processes...");
|
|
{
|
|
auto& file_lock = miopen::LockFile::Get(lock_file_path);
|
|
std::shared_lock<miopen::LockFile> lock(file_lock);
|
|
|
|
auto id = 0;
|
|
|
|
// clang-format off
|
|
for(auto i = 0u; i < DBMultiThreadedTestWork::threads_count; ++i)
|
|
{
|
|
// Use --gtest_filter to run only the child worker test in spawned process
|
|
// Note: no --write flag for read test
|
|
auto args =
|
|
std::string{"--reset-sharding"
|
|
" --gtest_filter="} + ArgsHelper::child_worker_filter +
|
|
" --" + ArgsHelper::id_arg + " " + std::to_string(id++) +
|
|
" --" + ArgsHelper::path_arg + " " + temp_file +
|
|
" --" + ArgsHelper::db_class_arg + " " + ArgsHelper::db_class::Get<TDb>();
|
|
|
|
if(thread_logs_root().has_value())
|
|
{
|
|
// NOLINTNEXTLINE (bugprone-unchecked-optional-access)
|
|
args += std::string{" --"} + ArgsHelper::logs_path_arg + " " + thread_logs_root().value();
|
|
}
|
|
|
|
if(full_set())
|
|
args += " --all";
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", exe_path() + " " + args);
|
|
children.emplace_back(exe_path(), args);
|
|
}
|
|
// clang-format on
|
|
}
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Waiting for test processes...");
|
|
for(auto&& child : children)
|
|
{
|
|
ASSERT_EQ(child.Wait(), 0);
|
|
}
|
|
|
|
{
|
|
std::error_code ec;
|
|
fs::remove(lock_file_path, ec);
|
|
}
|
|
}
|
|
|
|
static void WorkItem(unsigned int id, const fs::path& db_path)
|
|
{
|
|
{
|
|
auto& file_lock = miopen::LockFile::Get(LockFilePath(db_path));
|
|
std::lock_guard<miopen::LockFile> lock(file_lock);
|
|
}
|
|
|
|
const auto c = [&db_path]() { return TDb(db_path, false); };
|
|
|
|
DBMultiThreadedTestWork::WorkItem(id, c, "mp");
|
|
}
|
|
};
|
|
|
|
class DbMultiFileTest : public DbTest
|
|
{
|
|
protected:
|
|
DbMultiFileTest(TempFile& temp_file_) : DbTest(temp_file_) {}
|
|
|
|
fs::path user_db_path = temp_file.Path() + ".user";
|
|
|
|
void ResetDb()
|
|
{
|
|
DbTest::ResetDb();
|
|
user_db_path = temp_file.Path() + ".user";
|
|
}
|
|
|
|
private:
|
|
#if MIOPEN_EMBED_DB
|
|
TestRordbEmbedFsOverrideLock rordb_embed_fs_override;
|
|
#endif
|
|
};
|
|
|
|
template <bool merge_records>
|
|
class DbMultiFileReadTest : public DbMultiFileTest
|
|
{
|
|
public:
|
|
DbMultiFileReadTest(TempFile& temp_file_) : DbMultiFileTest(temp_file_) {}
|
|
|
|
void Run()
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Running multifile read test" << (merge_records ? " with merge" : "")
|
|
<< "...");
|
|
|
|
MergedAndMissing();
|
|
|
|
ResetDb();
|
|
ReadUser();
|
|
|
|
ResetDb();
|
|
ReadInstalled();
|
|
|
|
ResetDb();
|
|
ReadConflict();
|
|
}
|
|
|
|
private:
|
|
static const std::array<std::pair<const std::string, TestData>, 1>& single_item_data()
|
|
{
|
|
static const std::array<std::pair<const std::string, TestData>, 1> data{
|
|
{{id0(), value2()}}};
|
|
|
|
return data;
|
|
}
|
|
|
|
void MergedAndMissing() const
|
|
{
|
|
RawWrite(temp_file, key(), common_data());
|
|
RawWrite(user_db_path, key(), single_item_data());
|
|
|
|
static const std::array<std::pair<const std::string, TestData>, 2> merged_data{{
|
|
{id1(), value1()},
|
|
{id0(), value2()},
|
|
}};
|
|
|
|
MultiFileDb<ReadonlyRamDb, RamDb, merge_records> db(
|
|
DbKinds::PerfDb, temp_file, user_db_path);
|
|
if(merge_records)
|
|
ValidateSingleEntry(key(), merged_data, db);
|
|
else
|
|
ValidateSingleEntry(key(), single_item_data(), db);
|
|
|
|
const TestData invalid_key(100, 200);
|
|
auto record1 = db.FindRecord(invalid_key);
|
|
ASSERT_FALSE(record1);
|
|
}
|
|
|
|
void ReadUser() const
|
|
{
|
|
RawWrite(user_db_path, key(), single_item_data());
|
|
MultiFileDb<ReadonlyRamDb, RamDb, merge_records> db(
|
|
DbKinds::PerfDb, temp_file, user_db_path);
|
|
ValidateSingleEntry(key(), single_item_data(), db);
|
|
}
|
|
|
|
void ReadInstalled() const
|
|
{
|
|
RawWrite(temp_file, key(), single_item_data());
|
|
MultiFileDb<ReadonlyRamDb, RamDb, merge_records> db(
|
|
DbKinds::PerfDb, temp_file, user_db_path);
|
|
ValidateSingleEntry(key(), single_item_data(), db);
|
|
}
|
|
|
|
void ReadConflict() const
|
|
{
|
|
RawWrite(temp_file, key(), single_item_data());
|
|
ReadUser();
|
|
}
|
|
};
|
|
|
|
class DbMultiFileWriteTest : public DbMultiFileTest
|
|
{
|
|
public:
|
|
DbMultiFileWriteTest(TempFile& temp_file_) : DbMultiFileTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Running multifile write test...");
|
|
|
|
DbRecord record(DbKinds::PerfDb, key());
|
|
ASSERT_TRUE(record.SetValues(id0(), value0()));
|
|
ASSERT_TRUE(record.SetValues(id1(), value1()));
|
|
|
|
{
|
|
MultiFileDb<ReadonlyRamDb, RamDb, true> db(DbKinds::PerfDb, temp_file, user_db_path);
|
|
|
|
ASSERT_TRUE(db.StoreRecord(record));
|
|
}
|
|
|
|
std::string read;
|
|
ASSERT_FALSE(std::getline(std::ifstream(temp_file.Path()), read).good());
|
|
ASSERT_TRUE(std::getline(std::ifstream(user_db_path), read).good());
|
|
|
|
auto db = MultiFileDb<ReadonlyRamDb, RamDb, true>{DbKinds::PerfDb, temp_file, user_db_path};
|
|
ValidateSingleEntry(key(), common_data(), db);
|
|
}
|
|
};
|
|
|
|
class DbMultiFileOperationsTest : public DbMultiFileTest
|
|
{
|
|
public:
|
|
DbMultiFileOperationsTest(TempFile& temp_file_) : DbMultiFileTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
PrepareDb();
|
|
UpdateTest();
|
|
LoadTest();
|
|
RemoveTest();
|
|
RemoveRecordTest();
|
|
}
|
|
|
|
void PrepareDb() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Running multifile operations test...");
|
|
|
|
{
|
|
DbRecord record(DbKinds::PerfDb, key());
|
|
ASSERT_TRUE(record.SetValues(id0(), value0()));
|
|
ASSERT_TRUE(record.SetValues(id1(), value2()));
|
|
|
|
PlainTextDb db(DbKinds::PerfDb, temp_file);
|
|
ASSERT_TRUE(db.StoreRecord(record));
|
|
}
|
|
}
|
|
|
|
void UpdateTest() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Update test...");
|
|
|
|
{
|
|
MultiFileDb<ReadonlyRamDb, RamDb, true> db(DbKinds::PerfDb, temp_file, user_db_path);
|
|
ASSERT_TRUE(db.Update(key(), id1(), value1()));
|
|
}
|
|
|
|
{
|
|
PlainTextDb db(DbKinds::PerfDb, user_db_path);
|
|
TestData read(TestData::NoInit{});
|
|
ASSERT_FALSE(db.Load(key(), id0(), read));
|
|
ASSERT_TRUE(db.Load(key(), id1(), read));
|
|
ASSERT_EQ(read, value1());
|
|
}
|
|
|
|
{
|
|
PlainTextDb db(DbKinds::PerfDb, temp_file);
|
|
ValidateData(db, value2());
|
|
}
|
|
}
|
|
|
|
void LoadTest() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Load test...");
|
|
|
|
MultiFileDb<ReadonlyRamDb, RamDb, true> db(DbKinds::PerfDb, temp_file, user_db_path);
|
|
ValidateData(db, value1());
|
|
}
|
|
|
|
void RemoveTest() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Remove test...");
|
|
|
|
MultiFileDb<ReadonlyRamDb, RamDb, true> db(DbKinds::PerfDb, temp_file, user_db_path);
|
|
ASSERT_FALSE(db.Remove(key(), id0()));
|
|
ASSERT_TRUE(db.Remove(key(), id1()));
|
|
|
|
ValidateData(db, value2());
|
|
}
|
|
|
|
void RemoveRecordTest() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Remove record test...");
|
|
|
|
MultiFileDb<ReadonlyRamDb, RamDb, true> db(DbKinds::PerfDb, temp_file, user_db_path);
|
|
ASSERT_TRUE(db.Update(key(), id1(), value1()));
|
|
ASSERT_TRUE(db.RemoveRecord(key()));
|
|
|
|
ValidateData(db, value2());
|
|
}
|
|
|
|
template <class TDb>
|
|
void ValidateData(TDb& db, const TestData& id1Value) const
|
|
{
|
|
TestData read(TestData::NoInit{});
|
|
ASSERT_TRUE(db.Load(key(), id0(), read));
|
|
ASSERT_EQ(read, value0());
|
|
ASSERT_TRUE(db.Load(key(), id1(), read));
|
|
ASSERT_EQ(read, id1Value);
|
|
}
|
|
};
|
|
|
|
class DbMultiFileMultiThreadedReadTest : public DbMultiFileTest
|
|
{
|
|
public:
|
|
DbMultiFileMultiThreadedReadTest(TempFile& temp_file_) : DbMultiFileTest(temp_file_) {}
|
|
|
|
void Run()
|
|
{
|
|
MIOPEN_LOG_CUSTOM(
|
|
LoggingLevel::Default, "Test", "Testing db for multifile multithreaded read access...");
|
|
|
|
std::shared_mutex mutex;
|
|
std::vector<std::thread> threads;
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Initializing test data...");
|
|
const auto c = [this]() {
|
|
return MultiFileDb<ReadonlyRamDb, RamDb, true>(
|
|
DbKinds::PerfDb, temp_file, user_db_path);
|
|
};
|
|
ResetDb();
|
|
DBMultiThreadedTestWork::FillForReading(c);
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Launching test threads...");
|
|
threads.reserve(DBMultiThreadedTestWork::threads_count);
|
|
|
|
{
|
|
std::unique_lock<std::shared_mutex> lock(mutex);
|
|
|
|
for(auto i = 0u; i < DBMultiThreadedTestWork::threads_count; i++)
|
|
{
|
|
threads.emplace_back([c, &mutex, i]() {
|
|
std::shared_lock<std::shared_mutex> lock_(mutex);
|
|
DBMultiThreadedTestWork::ReadWorkItem(i, c, "mt");
|
|
});
|
|
}
|
|
}
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Waiting for test threads...");
|
|
for(auto& thread : threads)
|
|
thread.join();
|
|
}
|
|
};
|
|
|
|
class DbMultiFileMultiThreadedTest : public DbMultiFileTest
|
|
{
|
|
public:
|
|
DbMultiFileMultiThreadedTest(TempFile& temp_file_) : DbMultiFileTest(temp_file_) {}
|
|
|
|
void Run() const
|
|
{
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default,
|
|
"Test",
|
|
"Testing db for multifile multithreaded write access...");
|
|
|
|
std::shared_mutex mutex;
|
|
std::vector<std::thread> threads;
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Initializing test data...");
|
|
DBMultiThreadedTestWork::Initialize();
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Launching test threads...");
|
|
threads.reserve(DBMultiThreadedTestWork::threads_count);
|
|
const auto c = [this]() {
|
|
return MultiFileDb<ReadonlyRamDb, RamDb, true>(
|
|
DbKinds::PerfDb, temp_file, user_db_path);
|
|
};
|
|
|
|
{
|
|
std::unique_lock<std::shared_mutex> lock(mutex);
|
|
|
|
for(auto i = 0u; i < DBMultiThreadedTestWork::threads_count; i++)
|
|
{
|
|
threads.emplace_back([c, &mutex, i]() {
|
|
std::shared_lock<std::shared_mutex> _lock(mutex);
|
|
DBMultiThreadedTestWork::WorkItem(i, c, "mt");
|
|
});
|
|
}
|
|
}
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Waiting for test threads...");
|
|
for(auto& thread : threads)
|
|
thread.join();
|
|
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Validating results...");
|
|
DBMultiThreadedTestWork::ValidateCommonPart(c);
|
|
MIOPEN_LOG_CUSTOM(LoggingLevel::Default, "Test", "Validation passed...");
|
|
}
|
|
};
|
|
|
|
std::string GetArg(const std::string& name)
|
|
{
|
|
const auto& args = testing::internal::GetArgvs();
|
|
const auto target = "--" + name;
|
|
|
|
for(size_t i = 1; i + 1 < args.size(); ++i)
|
|
{
|
|
if(args[i] == target)
|
|
return args[i + 1];
|
|
}
|
|
return "";
|
|
}
|
|
|
|
bool HasFlag(const std::string& name)
|
|
{
|
|
const auto& args = testing::internal::GetArgvs();
|
|
const auto target = "--" + name;
|
|
|
|
for(size_t i = 1; i < args.size(); ++i)
|
|
{
|
|
if(args[i] == target)
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool IsChildProcessMode() { return !GetArg(ArgsHelper::id_arg).empty(); }
|
|
|
|
class CPU_PerfDb_ChildWorker_NONE : public ::testing::Test
|
|
{
|
|
protected:
|
|
void SetUp() override
|
|
{
|
|
if(!IsChildProcessMode())
|
|
{
|
|
GTEST_SKIP() << "Not in child process mode (no --" << ArgsHelper::id_arg << " arg)";
|
|
}
|
|
}
|
|
};
|
|
|
|
// cppcheck false positive: TEST_F inside anonymous namespace with complex fixture definitions.
|
|
// Known issue: https://trac.cppcheck.net/ticket/9160 (wontfix).
|
|
// Same workaround used in conv_ai_3d_heuristics.cpp.
|
|
// cppcheck-suppress syntaxError
|
|
TEST_F(CPU_PerfDb_ChildWorker_NONE, ProcessWork)
|
|
{
|
|
std::string logs_root = GetArg(ArgsHelper::logs_path_arg);
|
|
if(!logs_root.empty())
|
|
thread_logs_root() = logs_root;
|
|
|
|
if(HasFlag("all"))
|
|
{
|
|
full_set() = true;
|
|
DBMultiThreadedTestWork::threads_count = 32;
|
|
DBMultiThreadedTestWork::common_part_size = 128;
|
|
DBMultiThreadedTestWork::unique_part_size = 128;
|
|
}
|
|
|
|
std::string db_class = GetArg(ArgsHelper::db_class_arg);
|
|
std::string db_path = GetArg(ArgsHelper::path_arg);
|
|
int id = std::stoi(GetArg(ArgsHelper::id_arg));
|
|
bool write = HasFlag(ArgsHelper::write_arg);
|
|
|
|
if(db_class == ArgsHelper::db_class::db)
|
|
{
|
|
DbMultiProcessTest<PlainTextDb>::WorkItem(id, db_path, write);
|
|
}
|
|
else if(db_class == ArgsHelper::db_class::ramdb)
|
|
{
|
|
DbMultiProcessTest<RamDb>::WorkItem(id, db_path, write);
|
|
}
|
|
}
|
|
|
|
class CPU_PerfDb_NONE : public ::testing::Test
|
|
{
|
|
protected:
|
|
TempFile temp_file{"miopen.tests.perfdb"};
|
|
|
|
static void SetUpTestSuite()
|
|
{
|
|
// Save exe path for spawning child processes
|
|
const auto& args = testing::internal::GetArgvs();
|
|
if(!args.empty())
|
|
{
|
|
exe_path() = fs::absolute(args[0]);
|
|
#ifdef _WIN32
|
|
// CTest may invoke without .exe extension, but CreateProcess
|
|
// with explicit lpApplicationName requires it
|
|
if(!exe_path().has_extension())
|
|
exe_path().replace_extension(".exe");
|
|
#endif
|
|
}
|
|
}
|
|
|
|
template <class TDb>
|
|
void DbTests()
|
|
{
|
|
DbFindTest<TDb>{temp_file}.Run();
|
|
DbStoreTest<TDb>{temp_file}.Run();
|
|
DbUpdateTest<TDb>{temp_file}.Run();
|
|
DbRemoveTest<TDb>{temp_file}.Run();
|
|
DbReadTest<TDb>{temp_file}.Run();
|
|
DbWriteTest<TDb>{temp_file}.Run();
|
|
DbOperationsTest<TDb>{temp_file}.Run();
|
|
DbParallelTest<TDb>{temp_file}.Run();
|
|
|
|
DbMultiThreadedReadTest<TDb>{temp_file}.Run();
|
|
DbMultiProcessReadTest<TDb>{temp_file}.Run();
|
|
DbMultiThreadedTest<TDb>{temp_file}.Run();
|
|
DbMultiProcessTest<TDb>{temp_file}.Run();
|
|
}
|
|
|
|
void MultiFileDbTests()
|
|
{
|
|
if(!DisableUserDbFileIO)
|
|
{
|
|
DbMultiFileReadTest<true>{temp_file}.Run();
|
|
DbMultiFileReadTest<false>{temp_file}.Run();
|
|
DbMultiFileWriteTest{temp_file}.Run();
|
|
}
|
|
DbMultiFileOperationsTest{temp_file}.Run();
|
|
DbMultiFileMultiThreadedReadTest{temp_file}.Run();
|
|
DbMultiFileMultiThreadedTest{temp_file}.Run();
|
|
}
|
|
};
|
|
|
|
TEST_F(CPU_PerfDb_NONE, RamDb_AllTests) { DbTests<RamDb>(); }
|
|
|
|
TEST_F(CPU_PerfDb_NONE, PlainTextDb_AllTests) { DbTests<PlainTextDb>(); }
|
|
|
|
TEST_F(CPU_PerfDb_NONE, MultiFileDb_AllTests) { MultiFileDbTests(); }
|
|
|
|
} // anonymous namespace
|