Skip to content

Commit

Permalink
[ADAPTERS] HIP specific CTS
Browse files Browse the repository at this point in the history
  • Loading branch information
Petr Vesely authored and veselypeta committed Aug 2, 2023
1 parent fb05463 commit 661d171
Show file tree
Hide file tree
Showing 6 changed files with 138 additions and 0 deletions.
4 changes: 4 additions & 0 deletions test/conformance/adapters/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -6,3 +6,7 @@
if(UR_BUILD_ADAPTER_CUDA)
add_subdirectory(cuda)
endif()

if(UR_BUILD_ADAPTER_HIP)
add_subdirectory(hip)
endif()
23 changes: 23 additions & 0 deletions test/conformance/adapters/hip/CMakeLists.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
# Copyright (C) 2023 Intel Corporation
# Part of the Unified-Runtime Project, under the Apache License v2.0 with LLVM Exceptions.
# See LICENSE.TXT
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

add_conformance_test_with_devices_environment(adapter-hip
hip_fixtures.h
hip_urContextGetNativeHandle.cpp
hip_urDeviceGetNativeHandle.cpp
hip_urEventGetNativeHandle.cpp
)
target_link_libraries(test-adapter-hip PRIVATE rocmdrv)

if(UR_HIP_PLATFORM STREQUAL "AMD")
target_compile_definitions(test-adapter-hip PRIVATE __HIP_PLATFORM_AMD__)
else()
target_compile_definitions(test-adapter-hip PRIVATE __HIP_PLATFORM_NVIDIA__)
endif()

set_tests_properties(adapter-hip PROPERTIES
LABELS "conformance:hip"
ENVIRONMENT "UR_ADAPTERS_FORCE_LOAD=\"$<TARGET_FILE:ur_adapter_hip>\""
)
45 changes: 45 additions & 0 deletions test/conformance/adapters/hip/hip_fixtures.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
// Copyright (C) 2022-2023 Intel Corporation
// Part of the Unified-Runtime Project, under the Apache License v2.0 with LLVM Exceptions.
// See LICENSE.TXT
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#ifndef UR_TEST_CONFORMANCE_ADAPTERS_HIP_FIXTURES_H_INCLUDED
#define UR_TEST_CONFORMANCE_ADAPTERS_HIP_FIXTURES_H_INCLUDED
#include <hip/hip_runtime.h>
// #include <hip/hip_runtime_api.h>
#include <uur/fixtures.h>

namespace uur {

struct ResultHip {
constexpr ResultHip(hipError_t result) noexcept : value(result) {}

inline bool operator==(const ResultHip &rhs) const noexcept {
return rhs.value == value;
}

hipError_t value;
};

} // namespace uur

#ifndef ASSERT_EQ_RESULT_HIP
#define ASSERT_EQ_RESULT_HIP(EXPECTED, ACTUAL) \
ASSERT_EQ(uur::ResultHip(EXPECTED), uur::ResultHip(ACTUAL))
#endif // ASSERT_EQ_RESULT_HIP

#ifndef ASSERT_SUCCESS_HIP
#define ASSERT_SUCCESS_HIP(ACTUAL) ASSERT_EQ_RESULT_HIP(hipSuccess, ACTUAL)
#endif // ASSERT_SUCCESS_HIP

#ifndef EXPECT_EQ_RESULT_HIP
#define EXPECT_EQ_RESULT_HIP(EXPECTED, ACTUAL) \
EXPECT_EQ_RESULT_HIP(uur::ResultHip(EXPECTED), uur::ResultHip(ACTUAL))
#endif // EXPECT_EQ_RESULT_HIP

#ifndef EXPECT_SUCCESS_HIP
#define EXPECT_SUCCESS_HIP(ACTUAL) \
EXPECT_EQ_RESULT_HIP(UR_RESULT_SUCCESS, ACTUAL)
#endif // EXPECT_EQ_RESULT_HIP

#endif // UR_TEST_CONFORMANCE_ADAPTERS_HIP_FIXTURES_H_INCLUDED
16 changes: 16 additions & 0 deletions test/conformance/adapters/hip/hip_urContextGetNativeHandle.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
// Copyright (C) 2022-2023 Intel Corporation
// Part of the Unified-Runtime Project, under the Apache License v2.0 with LLVM Exceptions.
// See LICENSE.TXT
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include "hip_fixtures.h"

using urHipContextGetNativeHandleTest = uur::urContextTest;
UUR_INSTANTIATE_DEVICE_TEST_SUITE_P(urHipContextGetNativeHandleTest);

TEST_P(urHipContextGetNativeHandleTest, Success) {
ur_native_handle_t native_context = nullptr;
ASSERT_SUCCESS(urContextGetNativeHandle(context, &native_context));
hipCtx_t hip_context = reinterpret_cast<hipCtx_t>(native_context);
std::ignore = hip_context;
}
20 changes: 20 additions & 0 deletions test/conformance/adapters/hip/hip_urDeviceGetNativeHandle.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
// Copyright (C) 2022-2023 Intel Corporation
// Part of the Unified-Runtime Project, under the Apache License v2.0 with LLVM Exceptions.
// See LICENSE.TXT
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include "hip_fixtures.h"

using urHipGetDeviceNativeHandle = uur::urDeviceTest;
UUR_INSTANTIATE_DEVICE_TEST_SUITE_P(urHipGetDeviceNativeHandle);

TEST_P(urHipGetDeviceNativeHandle, Success) {
ur_native_handle_t native_handle;
ASSERT_SUCCESS(urDeviceGetNativeHandle(device, &native_handle));

hipDevice_t hip_device = *reinterpret_cast<hipDevice_t *>(&native_handle);

char hip_device_name[256];
ASSERT_SUCCESS_HIP(
hipDeviceGetName(hip_device_name, sizeof(hip_device_name), hip_device));
}
30 changes: 30 additions & 0 deletions test/conformance/adapters/hip/hip_urEventGetNativeHandle.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
// Copyright (C) 2022-2023 Intel Corporation
// Part of the Unified-Runtime Project, under the Apache License v2.0 with LLVM Exceptions.
// See LICENSE.TXT
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include "hip_fixtures.h"

using urHipEventGetNativeHandleTest = uur::urQueueTest;
UUR_INSTANTIATE_DEVICE_TEST_SUITE_P(urHipEventGetNativeHandleTest);

TEST_P(urHipEventGetNativeHandleTest, Success) {
constexpr size_t buffer_size = 1024;
ur_mem_handle_t mem = nullptr;
ASSERT_SUCCESS(urMemBufferCreate(context, UR_MEM_FLAG_READ_WRITE,
buffer_size, nullptr, &mem));

ur_event_handle_t event = nullptr;
uint8_t pattern = 6;
ASSERT_SUCCESS(urEnqueueMemBufferFill(queue, mem, &pattern, sizeof(pattern),
0, buffer_size, 0, nullptr, &event));

ur_native_handle_t native_event = nullptr;
ASSERT_SUCCESS(urEventGetNativeHandle(event, &native_event));
hipEvent_t hip_event = reinterpret_cast<hipEvent_t>(native_event);

ASSERT_SUCCESS_HIP(hipEventSynchronize(hip_event));

ASSERT_SUCCESS(urEventRelease(event));
ASSERT_SUCCESS(urMemRelease(mem));
}

0 comments on commit 661d171

Please sign in to comment.