9da4a23561
* Add more GPU architectures support * Update layout.py * Optimize performance, Add SM90 support, Add 1D2D SM100 support * Add fmtlib submodule at commit 553ec11 --------- Co-authored-by: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
36 lines
918 B
C++
36 lines
918 B
C++
#pragma once
|
|
|
|
#include <string>
|
|
|
|
namespace deep_gemm {
|
|
|
|
static uint64_t fnv1a(const std::string& data, const uint64_t& seed) {
|
|
uint64_t h = seed;
|
|
const uint64_t& prime = 0x100000001b3ull;
|
|
for (const char& c: data) {
|
|
h ^= static_cast<uint8_t>(c);
|
|
h *= prime;
|
|
}
|
|
return h;
|
|
}
|
|
|
|
static std::string get_hex_digest(const std::string& data) {
|
|
const auto& state_0 = fnv1a(data, 0xc6a4a7935bd1e995ull);
|
|
const auto& state_1 = fnv1a(data, 0x9e3779b97f4a7c15ull);
|
|
|
|
// Split-mix 64
|
|
const auto& split_mix = [](uint64_t z) {
|
|
z = (z ^ (z >> 30)) * 0xbf58476d1ce4e5b9ull;
|
|
z = (z ^ (z >> 27)) * 0x94d049bb133111ebull;
|
|
return z ^ (z >> 31);
|
|
};
|
|
|
|
std::ostringstream oss;
|
|
oss << std::hex << std::setfill('0')
|
|
<< std::setw(16) << split_mix(state_0)
|
|
<< std::setw(16) << split_mix(state_1);
|
|
return oss.str();
|
|
}
|
|
|
|
} // namespace deep_gemm
|