ROCm CI Service Account
232 lignes
8.5 KiB
C++
232 lignes
8.5 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 WARRANNTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNNESS 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 INN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR INN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
*/
|
|
|
|
/**
|
|
|
|
Testcase Scenarios
|
|
------------------
|
|
Functional:
|
|
1) Start stream capture and get capture info. Verify api is success, capture status is hipStreamCaptureStatusActive
|
|
and identifier returned is valid/non-zero.
|
|
2) End stream capture and get capture info. Verify api is success, capture status is hipStreamCaptureStatusNone
|
|
and identifier is not returned/updated by api.
|
|
3) Begin capture on hipStreamPerThread and get capture info. Verify api is success, capture status is hipStreamCaptureStatusActive
|
|
and identifier returned is valid/non-zero.
|
|
4) End capture on hipStreamPerThread, get capture info. Verify api is success, capture status is hipStreamCaptureStatusNone
|
|
and identifier is not returned/updated by api.
|
|
5) Perform multiple captures and verify the identifier returned is unique.
|
|
|
|
Argument Validation/Negative:
|
|
1) Pass pId as nullptr and verify api doesn’t crash and returns success.
|
|
2) Pass pCaptureStatus as nullptr and verify api doesn’t crash and returns error code.
|
|
|
|
*/
|
|
|
|
#include <hip_test_common.hh>
|
|
#include <hip_test_checkers.hh>
|
|
#include <hip_test_kernels.hh>
|
|
|
|
constexpr size_t N = 1000000;
|
|
constexpr int LAUNCH_ITERS = 1;
|
|
|
|
/**
|
|
* Validates stream capture info, launches graph and verify results
|
|
*/
|
|
void validateStreamCaptureInfo(hipStream_t mstream) {
|
|
hipStream_t stream1{nullptr}, stream2{nullptr}, streamForLaunch{nullptr};
|
|
hipEvent_t memsetEvent1, memsetEvent2, forkStreamEvent;
|
|
hipGraph_t graph{nullptr};
|
|
hipGraphExec_t graphExec{nullptr};
|
|
constexpr unsigned blocks = 512;
|
|
constexpr unsigned threadsPerBlock = 256;
|
|
size_t Nbytes = N * sizeof(float);
|
|
float *A_d, *C_d;
|
|
float *A_h, *C_h;
|
|
A_h = reinterpret_cast<float*>(malloc(Nbytes));
|
|
C_h = reinterpret_cast<float*>(malloc(Nbytes));
|
|
REQUIRE(A_h != nullptr);
|
|
REQUIRE(C_h != nullptr);
|
|
HIP_CHECK(hipMalloc(&A_d, Nbytes));
|
|
HIP_CHECK(hipMalloc(&C_d, Nbytes));
|
|
REQUIRE(A_d != nullptr);
|
|
REQUIRE(C_d != nullptr);
|
|
HIP_CHECK(hipStreamCreate(&streamForLaunch));
|
|
|
|
// Initialize input buffer
|
|
for (size_t i = 0; i < N; ++i) {
|
|
A_h[i] = 3.146f + i; // Pi
|
|
}
|
|
|
|
// Create cross stream dependencies.
|
|
// memset operations are done on stream1 and stream2
|
|
// and they are joined back to mainstream
|
|
HIP_CHECK(hipStreamCreate(&stream1));
|
|
HIP_CHECK(hipStreamCreate(&stream2));
|
|
HIP_CHECK(hipEventCreate(&memsetEvent1));
|
|
HIP_CHECK(hipEventCreate(&memsetEvent2));
|
|
HIP_CHECK(hipEventCreate(&forkStreamEvent));
|
|
|
|
HIP_CHECK(hipStreamBeginCapture(mstream, hipStreamCaptureModeGlobal));
|
|
HIP_CHECK(hipEventRecord(forkStreamEvent, mstream));
|
|
HIP_CHECK(hipStreamWaitEvent(stream1, forkStreamEvent, 0));
|
|
HIP_CHECK(hipStreamWaitEvent(stream2, forkStreamEvent, 0));
|
|
HIP_CHECK(hipMemsetAsync(A_d, 0, Nbytes, stream1));
|
|
HIP_CHECK(hipEventRecord(memsetEvent1, stream1));
|
|
HIP_CHECK(hipMemsetAsync(C_d, 0, Nbytes, stream2));
|
|
HIP_CHECK(hipEventRecord(memsetEvent2, stream2));
|
|
HIP_CHECK(hipStreamWaitEvent(mstream, memsetEvent1, 0));
|
|
HIP_CHECK(hipStreamWaitEvent(mstream, memsetEvent2, 0));
|
|
HIP_CHECK(hipMemcpyAsync(A_d, A_h, Nbytes, hipMemcpyHostToDevice, mstream));
|
|
hipLaunchKernelGGL(HipTest::vector_square, dim3(blocks),
|
|
dim3(threadsPerBlock), 0, mstream, A_d, C_d, N);
|
|
|
|
hipStreamCaptureStatus captureStatus{hipStreamCaptureStatusNone};
|
|
unsigned long long capSequenceID = 0; // NOLINT
|
|
HIP_CHECK(hipStreamGetCaptureInfo(mstream, &captureStatus, &capSequenceID));
|
|
|
|
// verify capture status is active and sequence id is valid
|
|
REQUIRE(captureStatus == hipStreamCaptureStatusActive);
|
|
REQUIRE(capSequenceID > 0);
|
|
|
|
// End capture and verify graph is returned
|
|
HIP_CHECK(hipMemcpyAsync(C_h, C_d, Nbytes, hipMemcpyDeviceToHost, mstream));
|
|
HIP_CHECK(hipStreamEndCapture(mstream, &graph));
|
|
REQUIRE(graph != nullptr);
|
|
|
|
// verify capture status is inactive and sequence id is not updated
|
|
capSequenceID = 0;
|
|
HIP_CHECK(hipStreamGetCaptureInfo(mstream, &captureStatus, &capSequenceID));
|
|
REQUIRE(captureStatus == hipStreamCaptureStatusNone);
|
|
REQUIRE(capSequenceID == 0);
|
|
|
|
HIP_CHECK(hipGraphInstantiate(&graphExec, graph, nullptr, nullptr, 0));
|
|
REQUIRE(graphExec != nullptr);
|
|
|
|
// Replay the recorded sequence multiple times
|
|
for (int i = 0; i < LAUNCH_ITERS; i++) {
|
|
HIP_CHECK(hipGraphLaunch(graphExec, streamForLaunch));
|
|
}
|
|
|
|
HIP_CHECK(hipStreamSynchronize(streamForLaunch));
|
|
|
|
HIP_CHECK(hipGraphDestroy(graph));
|
|
HIP_CHECK(hipStreamDestroy(streamForLaunch));
|
|
HIP_CHECK(hipStreamDestroy(stream1));
|
|
HIP_CHECK(hipStreamDestroy(stream2));
|
|
HIP_CHECK(hipEventDestroy(forkStreamEvent));
|
|
HIP_CHECK(hipEventDestroy(memsetEvent1));
|
|
HIP_CHECK(hipEventDestroy(memsetEvent2));
|
|
HIP_CHECK(hipFree(A_d));
|
|
HIP_CHECK(hipFree(C_d));
|
|
|
|
// Validate the computation
|
|
for (size_t i = 0; i < N; i++) {
|
|
if (C_h[i] != A_h[i] * A_h[i]) {
|
|
INFO("A and C not matching at " << i << " C_h[i] " << C_h[i]
|
|
<< " A_h[i] " << A_h[i]);
|
|
REQUIRE(false);
|
|
}
|
|
}
|
|
free(A_h);
|
|
free(C_h);
|
|
}
|
|
|
|
/**
|
|
* Basic Functional Test for stream capture and getting capture info.
|
|
* Regular/custom stream is used for stream capture.
|
|
*/
|
|
TEST_CASE("Unit_hipStreamGetCaptureInfo_BasicFunctional") {
|
|
hipStream_t streamForCapture;
|
|
|
|
HIP_CHECK(hipStreamCreate(&streamForCapture));
|
|
validateStreamCaptureInfo(streamForCapture);
|
|
HIP_CHECK(hipStreamDestroy(streamForCapture));
|
|
}
|
|
|
|
/**
|
|
* Test performs stream capture on hipStreamPerThread and validates
|
|
* capture info.
|
|
*/
|
|
TEST_CASE("Unit_hipStreamGetCaptureInfo_hipStreamPerThread") {
|
|
validateStreamCaptureInfo(hipStreamPerThread);
|
|
}
|
|
|
|
/**
|
|
* Test starts stream capture on multiple streams and verifies uniqueness of
|
|
* identifiers returned.
|
|
*/
|
|
TEST_CASE("Unit_hipStreamGetCaptureInfo_UniqueID") {
|
|
constexpr int numStreams = 100;
|
|
hipStream_t streams[numStreams]{};
|
|
hipStreamCaptureStatus captureStatus{hipStreamCaptureStatusNone};
|
|
std::vector<int> idlist;
|
|
unsigned long long capSequenceID{}; // NOLINT
|
|
hipGraph_t graph{nullptr};
|
|
|
|
for (int i = 0; i < numStreams; i++) {
|
|
HIP_CHECK(hipStreamCreate(&streams[i]));
|
|
HIP_CHECK(hipStreamBeginCapture(streams[i], hipStreamCaptureModeGlobal));
|
|
HIP_CHECK(hipStreamGetCaptureInfo(streams[i], &captureStatus,
|
|
&capSequenceID));
|
|
REQUIRE(captureStatus == hipStreamCaptureStatusActive);
|
|
REQUIRE(capSequenceID > 0);
|
|
idlist.push_back(capSequenceID);
|
|
}
|
|
|
|
for (int i = 0; i < numStreams; i++) {
|
|
for (int j = i+1; j < numStreams; j++) {
|
|
if (idlist[i] == idlist[j]) {
|
|
INFO("Same identifier returned for stream "
|
|
<< i << " and stream " << j);
|
|
REQUIRE(false);
|
|
}
|
|
}
|
|
}
|
|
|
|
for (int i = 0; i < numStreams; i++) {
|
|
HIP_CHECK(hipStreamEndCapture(streams[i], &graph));
|
|
HIP_CHECK(hipGraphDestroy(graph));
|
|
HIP_CHECK(hipStreamDestroy(streams[i]));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Argument validation/Negative tests for api
|
|
*/
|
|
TEST_CASE("Unit_hipStreamGetCaptureInfo_ArgValidation") {
|
|
hipError_t ret;
|
|
hipStream_t stream;
|
|
hipStreamCaptureStatus captureStatus;
|
|
unsigned long long capSequenceID; // NOLINT
|
|
HIP_CHECK(hipStreamCreate(&stream));
|
|
|
|
SECTION("Capture ID location as nullptr") {
|
|
ret = hipStreamGetCaptureInfo(stream, &captureStatus, nullptr);
|
|
// Capture ID is optional
|
|
REQUIRE(ret == hipSuccess);
|
|
}
|
|
|
|
SECTION("Capture Status location as nullptr") {
|
|
ret = hipStreamGetCaptureInfo(stream, nullptr, &capSequenceID);
|
|
REQUIRE(ret == hipErrorInvalidValue);
|
|
}
|
|
|
|
HIP_CHECK(hipStreamDestroy(stream));
|
|
}
|