agodavar
222 строки
8.3 KiB
C++
222 строки
8.3 KiB
C++
/*
|
|
Copyright (c) 2015-2016 Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
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.
|
|
*/
|
|
|
|
/* HIT_START
|
|
* BUILD: %t %s ../test_common.cpp EXCLUDE_HIP_PLATFORM nvidia
|
|
* TEST: %t
|
|
* HIT_END
|
|
*/
|
|
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <algorithm>
|
|
#include <stdlib.h>
|
|
#include <iostream>
|
|
#include <random>
|
|
#include "hip/hip_runtime.h"
|
|
#include <hip/device_functions.h>
|
|
|
|
#define HIP_ASSERT(x) (assert((x) == hipSuccess))
|
|
|
|
#define TEST_DEBUG (0)
|
|
|
|
|
|
// CPU implementation of bitextract
|
|
template <typename T>
|
|
T bit_extract(T src0, unsigned int src1, unsigned int src2) {
|
|
unsigned int bits = sizeof(T) * 8;
|
|
T offset = src1 & (bits - 1);
|
|
T width = src2 & (bits - 1);
|
|
if (width == 0) {
|
|
return 0;
|
|
} else {
|
|
return (src0 << (bits - width - offset)) >> (bits - width);
|
|
}
|
|
}
|
|
|
|
__global__ void HIP_kernel(unsigned int* out32, unsigned int* in32_0,
|
|
unsigned int* in32_1, unsigned int* in32_2,
|
|
unsigned long long int* out64, unsigned long long int* in64_0,
|
|
unsigned int* in64_1, unsigned int* in64_2) {
|
|
int x = blockDim.x * blockIdx.x + threadIdx.x;
|
|
|
|
out32[x] = __bitextract_u32(in32_0[x], in32_1[x], in32_2[x]);
|
|
out64[x] = __bitextract_u64(in64_0[x], in64_1[x], in64_2[x]);
|
|
}
|
|
|
|
|
|
using namespace std;
|
|
|
|
int main() {
|
|
unsigned int* hostOut32;
|
|
unsigned int* hostSrc032;
|
|
unsigned int* hostSrc132;
|
|
unsigned int* hostSrc232;
|
|
unsigned long long int* hostOut64;
|
|
unsigned long long int* hostSrc064;
|
|
unsigned int* hostSrc164;
|
|
unsigned int* hostSrc264;
|
|
|
|
unsigned int* deviceOut32;
|
|
unsigned int* deviceSrc032;
|
|
unsigned int* deviceSrc132;
|
|
unsigned int* deviceSrc232;
|
|
unsigned long long int* deviceOut64;
|
|
unsigned long long int* deviceSrc064;
|
|
unsigned int* deviceSrc164;
|
|
unsigned int* deviceSrc264;
|
|
|
|
hipDeviceProp_t devProp;
|
|
hipGetDeviceProperties(&devProp, 0);
|
|
cout << " System minor " << devProp.minor << endl;
|
|
cout << " System major " << devProp.major << endl;
|
|
cout << " agent prop name " << devProp.name << endl;
|
|
|
|
cout << "hip Device prop succeeded " << endl;
|
|
|
|
unsigned int wave_size = devProp.warpSize;
|
|
unsigned int num_waves_per_block = 2;
|
|
unsigned int num_threads_per_block = wave_size * num_waves_per_block;
|
|
unsigned int num_blocks = 2;
|
|
unsigned int NUM = num_threads_per_block * num_blocks;
|
|
|
|
int i;
|
|
int errors;
|
|
|
|
hostOut32 = (unsigned int*)malloc(NUM * sizeof(unsigned int));
|
|
hostSrc032 = (unsigned int*)malloc(NUM * sizeof(unsigned int));
|
|
hostSrc132 = (unsigned int*)malloc(NUM * sizeof(unsigned int));
|
|
hostSrc232 = (unsigned int*)malloc(NUM * sizeof(unsigned int));
|
|
|
|
hostOut64 = (unsigned long long int*)malloc(NUM * sizeof(unsigned long long int));
|
|
hostSrc064 = (unsigned long long int*)malloc(NUM * sizeof(unsigned long long int));
|
|
hostSrc164 = (unsigned int*)malloc(NUM * sizeof(unsigned int));
|
|
hostSrc264 = (unsigned int*)malloc(NUM * sizeof(unsigned int));
|
|
|
|
// initialize the input data
|
|
std::random_device rd;
|
|
std::uniform_int_distribution<uint32_t> uint32_src0_dist;
|
|
std::uniform_int_distribution<uint32_t> uint32_src12_dist(0,31);
|
|
std::uniform_int_distribution<uint64_t> uint64_src0_dist;
|
|
std::uniform_int_distribution<uint32_t> uint64_src12_dist(0,63);
|
|
for (i = 0; i < NUM; i++) {
|
|
hostOut32[i] = 0;
|
|
hostSrc032[i] = uint32_src0_dist(rd);
|
|
hostSrc132[i] = uint32_src12_dist(rd);
|
|
hostSrc232[i] = uint32_src12_dist(rd);
|
|
if (hostSrc132[i] + hostSrc232[i] > 32)
|
|
hostSrc232[i] = 32 - hostSrc132[i];
|
|
hostOut64[i] = 0;
|
|
hostSrc064[i] = uint64_src0_dist(rd);
|
|
hostSrc164[i] = uint64_src12_dist(rd);
|
|
hostSrc264[i] = uint64_src12_dist(rd);
|
|
}
|
|
|
|
HIP_ASSERT(hipMalloc((void**)&deviceOut32, NUM * sizeof(unsigned int)));
|
|
HIP_ASSERT(hipMalloc((void**)&deviceSrc032, NUM * sizeof(unsigned int)));
|
|
HIP_ASSERT(hipMalloc((void**)&deviceSrc132, NUM * sizeof(unsigned int)));
|
|
HIP_ASSERT(hipMalloc((void**)&deviceSrc232, NUM * sizeof(unsigned int)));
|
|
|
|
HIP_ASSERT(hipMalloc((void**)&deviceOut64, NUM * sizeof(unsigned long long int)));
|
|
HIP_ASSERT(hipMalloc((void**)&deviceSrc064, NUM * sizeof(unsigned long long int)));
|
|
HIP_ASSERT(hipMalloc((void**)&deviceSrc164, NUM * sizeof(unsigned int)));
|
|
HIP_ASSERT(hipMalloc((void**)&deviceSrc264, NUM * sizeof(unsigned int)));
|
|
|
|
HIP_ASSERT(hipMemcpy(deviceSrc032, hostSrc032, NUM * sizeof(unsigned int), hipMemcpyHostToDevice));
|
|
HIP_ASSERT(hipMemcpy(deviceSrc132, hostSrc132, NUM * sizeof(unsigned int), hipMemcpyHostToDevice));
|
|
HIP_ASSERT(hipMemcpy(deviceSrc232, hostSrc232, NUM * sizeof(unsigned int), hipMemcpyHostToDevice));
|
|
|
|
HIP_ASSERT(hipMemcpy(deviceSrc064, hostSrc064, NUM * sizeof(unsigned long long int),
|
|
hipMemcpyHostToDevice));
|
|
HIP_ASSERT(hipMemcpy(deviceSrc164, hostSrc164, NUM * sizeof(unsigned int), hipMemcpyHostToDevice));
|
|
HIP_ASSERT(hipMemcpy(deviceSrc264, hostSrc264, NUM * sizeof(unsigned int), hipMemcpyHostToDevice));
|
|
|
|
|
|
hipLaunchKernelGGL(HIP_kernel, dim3(num_blocks), dim3(num_threads_per_block),
|
|
0, 0,
|
|
deviceOut32, deviceSrc032, deviceSrc132, deviceSrc232,
|
|
deviceOut64, deviceSrc064, deviceSrc164, deviceSrc264);
|
|
|
|
|
|
HIP_ASSERT(hipMemcpy(hostOut32, deviceOut32, NUM * sizeof(unsigned int), hipMemcpyDeviceToHost));
|
|
HIP_ASSERT(hipMemcpy(hostOut64, deviceOut64,
|
|
NUM * sizeof(unsigned long long int), hipMemcpyDeviceToHost));
|
|
|
|
// verify the results
|
|
errors = 0;
|
|
for (i = 0; i < NUM; i++) {
|
|
if (hostOut32[i] != bit_extract<uint32_t>(hostSrc032[i], hostSrc132[i], hostSrc232[i])) {
|
|
errors++;
|
|
#if TEST_DEBUG
|
|
cout << "device: " << hostOut32[i] << " host: "
|
|
<< bit_extract<uint32_t>(hostSrc032[i], hostSrc132[i], hostSrc232[i])
|
|
<< " " << hostSrc032[i] << " " << hostSrc132[i] << " " << hostSrc232[i] << "\n";
|
|
#endif
|
|
}
|
|
}
|
|
if (errors != 0) {
|
|
cout << "__bitextract_u32() FAILED\n" << endl;
|
|
return -1;
|
|
} else {
|
|
cout << "__bitextract_u32() checked!" << endl;
|
|
}
|
|
errors = 0;
|
|
for (i = 0; i < NUM; i++) {
|
|
if (hostOut64[i] != bit_extract<uint64_t>(hostSrc064[i], hostSrc164[i], hostSrc264[i])) {
|
|
errors++;
|
|
#if TEST_DEBUG
|
|
cout << "device: " << hostOut64[i] << " host: "
|
|
<< bit_extract<uint64_t>(hostSrc064[i], hostSrc164[i], hostSrc264[i])
|
|
<< " " << hostSrc064[i] << " " << hostSrc164[i] << " " << hostSrc264[i] << "\n";
|
|
#endif
|
|
}
|
|
}
|
|
if (errors != 0) {
|
|
cout << "__bitextract_u64() FAILED" << endl;
|
|
return -1;
|
|
} else {
|
|
cout << "__bitextract_u64() checked!" << endl;
|
|
}
|
|
|
|
cout << "__bitextract_u32() and __bitextract_u64() PASSED!" << endl;
|
|
|
|
HIP_ASSERT(hipFree(deviceOut32));
|
|
HIP_ASSERT(hipFree(deviceSrc032));
|
|
HIP_ASSERT(hipFree(deviceSrc132));
|
|
HIP_ASSERT(hipFree(deviceSrc232));
|
|
HIP_ASSERT(hipFree(deviceOut64));
|
|
HIP_ASSERT(hipFree(deviceSrc064));
|
|
HIP_ASSERT(hipFree(deviceSrc164));
|
|
HIP_ASSERT(hipFree(deviceSrc264));
|
|
|
|
free(hostOut32);
|
|
free(hostSrc032);
|
|
free(hostSrc132);
|
|
free(hostSrc232);
|
|
free(hostOut64);
|
|
free(hostSrc064);
|
|
free(hostSrc164);
|
|
free(hostSrc264);
|
|
|
|
return errors;
|
|
}
|