ROCm CI Service Account
d9b40f34a4
Added tests for AtomicAdd API for coherent/non-coherent memory with and without unsafeAtomics flags Change-Id: I1937d7e936ac062b8d93feb59a50a6dd8ae9feca
92 خطوط
3.7 KiB
C++
92 خطوط
3.7 KiB
C++
/*
|
|
Copyright (c) 2022 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.
|
|
*/
|
|
/*
|
|
AtomicAdd on CoarseGrainMemory
|
|
1. The following test scenario verifies
|
|
atomicAdd on CoarseGrain memory without any unsafeatomics flag
|
|
This testcase works only on gfx90a.
|
|
*/
|
|
|
|
#include<hip_test_checkers.hh>
|
|
#include<hip_test_common.hh>
|
|
|
|
#define INC_VAL 10
|
|
#define INITIAL_VAL 5
|
|
|
|
template<typename T>
|
|
static __global__ void AtomicCheck(T* Ad, T* result) {
|
|
T inc_val = 10;
|
|
*result = atomicAdd(Ad, inc_val);
|
|
}
|
|
|
|
/*atomicAdd API for the coarse grained memory variable
|
|
without any flag
|
|
Input: Ad{5}, INC_VAL{10}
|
|
Output: atomicAdd API would work and the 0/P is INITIAL_VAL + INC_VAL
|
|
Generate the assembly file and check whether
|
|
global_atomic_cmpswap instruction is generated
|
|
or not */
|
|
|
|
TEMPLATE_TEST_CASE("Unit_AtomicAdd_NonCoherentwithoutflag", "",
|
|
float, double) {
|
|
hipDeviceProp_t prop;
|
|
int device;
|
|
HIP_CHECK(hipGetDevice(&device));
|
|
HIP_CHECK(hipGetDeviceProperties(&prop, device));
|
|
std::string gfxName(prop.gcnArchName);
|
|
if ((gfxName == "gfx90a" || gfxName.find("gfx90a:")) == 0) {
|
|
if (prop.canMapHostMemory != 1) {
|
|
SUCCEED("Does not support HostPinned Memory");
|
|
} else {
|
|
TestType *A_h{nullptr}, *result{nullptr};
|
|
TestType *A_d{nullptr}, *result_d{nullptr};
|
|
HIP_CHECK(hipHostMalloc(reinterpret_cast<void**>(&A_h), sizeof(TestType),
|
|
hipHostMallocNonCoherent));
|
|
A_h[0] = INITIAL_VAL;
|
|
HIP_CHECK(hipHostMalloc(reinterpret_cast<void**>(&result),
|
|
sizeof(TestType),
|
|
hipHostMallocNonCoherent));
|
|
result[0] = INITIAL_VAL;
|
|
HIP_CHECK(hipHostGetDevicePointer(reinterpret_cast<void**>(&A_d),
|
|
A_h, 0));
|
|
HIP_CHECK(hipHostGetDevicePointer(reinterpret_cast<void**>(&result_d),
|
|
result, 0));
|
|
hipLaunchKernelGGL(AtomicCheck<TestType>,
|
|
dim3(1), dim3(1),
|
|
0, 0, A_d,
|
|
result_d);
|
|
HIP_CHECK(hipDeviceSynchronize());
|
|
bool testResult;
|
|
REQUIRE(A_h[0] == INITIAL_VAL + INC_VAL);
|
|
REQUIRE(result[0] == INITIAL_VAL);
|
|
testResult = HipTest::assemblyFile_Verification<TestType>(
|
|
"AtomicAdd_NonCoherent_withoutflag-hip-amdgcn(.*)\\.s",
|
|
"global_atomic_cmpswap");
|
|
REQUIRE(testResult == true);
|
|
HIP_CHECK(hipHostFree(A_h));
|
|
HIP_CHECK(hipHostFree(result));
|
|
}
|
|
} else {
|
|
SUCCEED("Memory model feature is only supported for gfx90a, Hence"
|
|
"skipping the testcase for this GPU " << device);
|
|
}
|
|
}
|