hash.cuh

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
 
#pragma once
#include <cuda_runtime.h>
#include <cuda/std/array>
#include <cuda/std/span>
#include <cuda/std/tuple>
#include <concepts>

template <typename Hash>
concept Hashable = requires(Hash hash, cuda::std::span<const int4, 4> msg) {
  { hash.Hash(msg) } -> std::same_as<cuda::std::array<int4, 2>>;
};

template <typename Hash>
concept HasHash4 = Hashable<Hash> &&
    requires(Hash hash, cuda::std::span<const cuda::std::array<int4, 4>, 4> msgs) {
  { hash.Hash4(msgs) } -> std::same_as<cuda::std::array<cuda::std::array<int4, 2>, 4>>;
};

template <typename Hash>
concept XorHashable = requires(Hash hash, cuda::std::tuple<int4, const int4> msg) {
  { hash.Hash(msg) } -> std::same_as<cuda::std::array<int4, 4>>;
};

template <typename Hash>
concept CcrHashable = requires(Hash hash, int4 block) {
  { hash.Hash(block) } -> std::same_as<int4>;
};
 
// -----------------------------------------------------------------------------
 
namespace fss::detail {
 
template <typename Hash>
cuda::std::array<int4, 8> HashBatch(Hash &hash, const cuda::std::array<int4, 8> &blocks) {
  cuda::std::array<int4, 8> out;
  for (int i = 0; i < 8; ++i) out[i] = hash.Hash(blocks[i]);
  return out;
}
 
}  // namespace fss::detail