AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-459479 - Add compressed codeobj test

Browse Source 
Add simple tests to verify compressed code
objects.

Change-Id: Iae148c3c928e18247624937512918dbb3cbc462d
This commit is contained in:
taosang2
2024-04-29 14:06:42 -04:00
committed by Tao Sang
parent 01bfe61293
commit 6e1c4c2655
5 changed files with 126 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
catch/unit/module/hipModuleLoadData.cc
+13
View File
@@ -36,6 +36,18 @@ TEST_CASE("Unit_hipModuleLoadData_Positive_Basic") {
HIP_CHECK(hipModuleUnload(module));
}
#if defined(__HIP_PLATFORM_AMD__)
SECTION("Load compiled module from file with compressed code objects") {
const auto loaded_module = LoadModuleIntoBuffer("copyKernelCompressed.code");
HIP_CHECK(hipModuleLoadData(&module, loaded_module.data()));
REQUIRE(module != nullptr);
hipFunction_t kernel = nullptr;
HIP_CHECK(hipModuleGetFunction(&kernel, module, "copy_ker"));
REQUIRE(kernel != nullptr);
HIP_CHECK(hipModuleUnload(module));
}
#endif
SECTION("Load RTCd module") {
const auto rtc = CreateRTCCharArray(R"(extern "C" __global__ void kernel() {})");
HIP_CHECK(hipModuleLoadData(&module, rtc.data()));
@@ -65,6 +77,7 @@ TEST_CASE("Unit_hipModuleLoadData_Negative_Image_Is_An_Empty_String") {
HIP_CHECK_ERROR(hipModuleLoadData(&module, ""), hipErrorInvalidImage);
}
/**
* @addtogroup hipModuleLoad hipModuleGetFunction
* @{