Skip to content

Commit

Permalink
Add SHA1 testing
Browse files Browse the repository at this point in the history
  • Loading branch information
mborland committed Oct 18, 2024
1 parent e45a1ff commit 4d9352e
Show file tree
Hide file tree
Showing 2 changed files with 111 additions and 0 deletions.
1 change: 1 addition & 0 deletions test/nvcc_jamfile
Original file line number Diff line number Diff line change
Expand Up @@ -10,3 +10,4 @@ project : requirements
;

run test_md5_nvcc.cu ;
run test_sha1_nvcc.cu ;
110 changes: 110 additions & 0 deletions test/test_sha1_nvcc.cu
Original file line number Diff line number Diff line change
@@ -0,0 +1,110 @@
// Copyright Matt Borland 2024
// Use, modification and distribution are subject to the
// Boost Software License, Version 1.0. (See accompanying file
// LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

#include <boost/crypt/hash/sha1.hpp>
#include "cuda_managed_ptr.hpp"
#include "stopwatch.hpp"
#include "generate_random_strings.hpp"
#include <iostream>
#include <iomanip>
#include <exception>
#include <memory>

#include <cuda_runtime.h>

using digest_type = boost::crypt::array<boost::crypt::uint8_t, 20>;

// The kernel function
__global__ void cuda_test(char** in, digest_type* out, int numElements)
{
int i = blockIdx.x * blockDim.x + threadIdx.x;

if (i < numElements)
{
out[i] = boost::crypt::sha1(in[i]);
}
}

int main()
{
try
{
// Error code to check return values for CUDA calls
cudaError_t err = cudaSuccess;

// Print the vector length to be used, and compute its size
constexpr int numElements = 50000;
constexpr std::size_t elementSize = 64;

std::cout << "[Vector operation on " << numElements << " elements]" << std::endl;

// Allocate the managed input vector A
char** input_vector1 = new char*[numElements];

// Allocate the managed output vector C
cuda_managed_ptr<digest_type> output_vector(numElements);

for (int i = 0; i < numElements; ++i)
{
input_vector1[i] = new char[elementSize];
if (input_vector1[i] == nullptr)
{
throw std::runtime_error("Failed to allocate memory for input_vector1");
}
boost::crypt::generate_random_string(input_vector1[i], elementSize);
}

// Launch the Vector Add CUDA Kernel
int threadsPerBlock = 256;
int blocksPerGrid =(numElements + threadsPerBlock - 1) / threadsPerBlock;
std::cout << "CUDA kernel launch with " << blocksPerGrid << " blocks of " << threadsPerBlock << " threads" << std::endl;

watch w;
cuda_test<<<blocksPerGrid, threadsPerBlock>>>(input_vector1, output_vector.get(), numElements);
cudaDeviceSynchronize();
std::cout << "CUDA kernal done in " << w.elapsed() << "s" << std::endl;

err = cudaGetLastError();
if (err != cudaSuccess)
{
std::cerr << "Failed to launch vectorAdd kernel (error code " << cudaGetErrorString(err) << ")!" << std::endl;
return EXIT_FAILURE;
}

// Verify that the result vector is correct
std::vector<digest_type> results;
results.reserve(numElements);
w.reset();
for(int i = 0; i < numElements; ++i)
{
results.emplace_back(boost::crypt::sha1(input_vector1[i]));
}
double t = w.elapsed();

// check the results
for(int i = 0; i < numElements; ++i)
{
if (output_vector[i][0] != results[i][0])
{
std::cerr << "Result verification failed at element " << i << "!" << std::endl;
return EXIT_FAILURE;
}
}

std::cout << "Test PASSED with calculation time: " << t << "s" << std::endl;
std::cout << "Done\n";

// Cleanup all the memory we allocated
for (int i = 0; i < numElements; ++i)
{
delete[] input_vector1[i];
}
delete[] input_vector1;
}
catch (const std::exception& e)
{
std::cerr << "Terminated with exception: " << e.what() << std::endl;
}
}

0 comments on commit 4d9352e

Please sign in to comment.