mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-13 01:36:06 +00:00
Update test CMakeLists to add new tests automatically and add Jenkins stage for tests (#88)
* add docker file and make default target buildable * add Jenkinsfile * remove empty env block * fix package stage * remove render group from docker run * clean up Jenkins file * add cppcheck as dev dependency * update cmake file * Add profiler build stage * add hip_version config file for reduction operator * correct jenkins var name * Build release instead of debug * Update test CMakeLists.txt reorg test dir add test stage * reduce compile threads to prevent compiler crash * add optional debug stage, update second test * remove old test target * fix tests to return proper results and self review * Fix package name and make test run without args * change Dockerfile to ues rocm4.3.1 * remove parallelism from build * Lower paralellism Co-authored-by: Chao Liu <chao.liu2@amd.com>
This commit is contained in:
172
test/magic_number_division.cpp
Normal file
172
test/magic_number_division.cpp
Normal file
@@ -0,0 +1,172 @@
|
||||
#include <iostream>
|
||||
#include <numeric>
|
||||
#include <initializer_list>
|
||||
#include <cstdlib>
|
||||
#include <stdlib.h>
|
||||
#include <half.hpp>
|
||||
#include "config.hpp"
|
||||
#include "print.hpp"
|
||||
#include "device.hpp"
|
||||
#include "host_tensor.hpp"
|
||||
#include "host_tensor_generator.hpp"
|
||||
#include "device_tensor.hpp"
|
||||
|
||||
__global__ void gpu_magic_number_division(uint32_t magic_multiplier,
|
||||
uint32_t magic_shift,
|
||||
const int32_t* p_dividend,
|
||||
int32_t* p_result,
|
||||
uint64_t num)
|
||||
{
|
||||
uint64_t global_thread_num = blockDim.x * gridDim.x;
|
||||
|
||||
uint64_t global_thread_id = blockIdx.x * blockDim.x + threadIdx.x;
|
||||
|
||||
for(uint64_t data_id = global_thread_id; data_id < num; data_id += global_thread_num)
|
||||
{
|
||||
p_result[data_id] =
|
||||
ck::MagicDivision::DoMagicDivision(p_dividend[data_id], magic_multiplier, magic_shift);
|
||||
}
|
||||
}
|
||||
|
||||
__global__ void
|
||||
gpu_naive_division(int32_t divisor, const int32_t* p_dividend, int32_t* p_result, uint64_t num)
|
||||
{
|
||||
uint64_t global_thread_num = blockDim.x * gridDim.x;
|
||||
|
||||
uint64_t global_thread_id = blockIdx.x * blockDim.x + threadIdx.x;
|
||||
|
||||
for(uint64_t data_id = global_thread_id; data_id < num; data_id += global_thread_num)
|
||||
{
|
||||
p_result[data_id] = p_dividend[data_id] / divisor;
|
||||
}
|
||||
}
|
||||
|
||||
__host__ void cpu_magic_number_division(uint32_t magic_multiplier,
|
||||
uint32_t magic_shift,
|
||||
const int32_t* p_dividend,
|
||||
int32_t* p_result,
|
||||
uint64_t num)
|
||||
{
|
||||
for(uint64_t data_id = 0; data_id < num; ++data_id)
|
||||
{
|
||||
p_result[data_id] =
|
||||
ck::MagicDivision::DoMagicDivision(p_dividend[data_id], magic_multiplier, magic_shift);
|
||||
}
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
T check_error(const std::vector<T>& ref, const std::vector<T>& result)
|
||||
{
|
||||
T error = 0;
|
||||
T max_diff = 0;
|
||||
T ref_value = 0, result_value = 0;
|
||||
|
||||
for(std::size_t i = 0; i < ref.size(); ++i)
|
||||
{
|
||||
T diff = std::abs(ref[i] - result[i]);
|
||||
error += diff;
|
||||
|
||||
if(max_diff < diff)
|
||||
{
|
||||
max_diff = diff;
|
||||
ref_value = ref[i];
|
||||
result_value = result[i];
|
||||
}
|
||||
}
|
||||
|
||||
return max_diff;
|
||||
}
|
||||
|
||||
int main(int, char*[])
|
||||
{
|
||||
uint64_t num_divisor = 4096;
|
||||
uint64_t num_dividend = 1L << 16;
|
||||
|
||||
std::vector<int32_t> divisors_host(num_divisor);
|
||||
std::vector<int32_t> dividends_host(num_dividend);
|
||||
|
||||
// generate divisor
|
||||
for(uint64_t i = 0; i < num_divisor; ++i)
|
||||
{
|
||||
divisors_host[i] = i + 1;
|
||||
}
|
||||
|
||||
// generate dividend
|
||||
for(uint64_t i = 0; i < num_divisor; ++i)
|
||||
{
|
||||
dividends_host[i] = i;
|
||||
}
|
||||
|
||||
DeviceMem dividends_dev_buf(sizeof(int32_t) * num_dividend);
|
||||
DeviceMem naive_result_dev_buf(sizeof(int32_t) * num_dividend);
|
||||
DeviceMem magic_result_dev_buf(sizeof(int32_t) * num_dividend);
|
||||
|
||||
std::vector<int32_t> naive_result_host(num_dividend);
|
||||
std::vector<int32_t> magic_result_host(num_dividend);
|
||||
std::vector<int32_t> magic_result_host2(num_dividend);
|
||||
|
||||
dividends_dev_buf.ToDevice(dividends_host.data());
|
||||
|
||||
bool pass = true;
|
||||
|
||||
for(std::size_t i = 0; i < num_divisor; ++i)
|
||||
{
|
||||
// run naive division on GPU
|
||||
gpu_naive_division<<<1024, 256>>>(
|
||||
divisors_host[i],
|
||||
static_cast<const int32_t*>(dividends_dev_buf.GetDeviceBuffer()),
|
||||
static_cast<int32_t*>(naive_result_dev_buf.GetDeviceBuffer()),
|
||||
num_dividend);
|
||||
|
||||
// calculate magic number
|
||||
uint32_t magic_multiplier, magic_shift;
|
||||
|
||||
ck::tie(magic_multiplier, magic_shift) =
|
||||
ck::MagicDivision::CalculateMagicNumbers(divisors_host[i]);
|
||||
|
||||
// run magic division on GPU
|
||||
gpu_magic_number_division<<<1024, 256>>>(
|
||||
magic_multiplier,
|
||||
magic_shift,
|
||||
static_cast<const int32_t*>(dividends_dev_buf.GetDeviceBuffer()),
|
||||
static_cast<int32_t*>(magic_result_dev_buf.GetDeviceBuffer()),
|
||||
num_dividend);
|
||||
|
||||
naive_result_dev_buf.FromDevice(naive_result_host.data());
|
||||
magic_result_dev_buf.FromDevice(magic_result_host.data());
|
||||
|
||||
int32_t max_diff = check_error(naive_result_host, magic_result_host);
|
||||
|
||||
if(max_diff != 0)
|
||||
{
|
||||
pass = false;
|
||||
continue;
|
||||
}
|
||||
|
||||
cpu_magic_number_division(magic_multiplier,
|
||||
magic_shift,
|
||||
dividends_host.data(),
|
||||
magic_result_host2.data(),
|
||||
num_dividend);
|
||||
|
||||
max_diff = check_error(naive_result_host, magic_result_host2);
|
||||
|
||||
if(max_diff != 0)
|
||||
{
|
||||
pass = false;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
if(pass)
|
||||
{
|
||||
std::cout << "test magic number division: Pass" << std::endl;
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
std::cout << "test magic number division: Fail" << std::endl;
|
||||
return -1;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user