Patching smi path for tests.

This commit is contained in:
Renjith Ravindran Kannath 2023-08-30 17:04:22 -07:00
parent dea3164ec0
commit ec463dbde6
2 changed files with 37 additions and 0 deletions

View File

@ -0,0 +1,31 @@
From 099ac638f41d9224f649fe23a64783bb408a2b09 Mon Sep 17 00:00:00 2001
From: Sreenivasa Murthy Kolam <sreenivasamurthy.kolam@amd.com>
Date: Wed, 30 Aug 2023 09:41:15 +0000
Subject: [PATCH] add rocm-smi-lib path for building tests
---
test/CMakeLists.txt | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 85f98d0..269f517 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -69,11 +69,12 @@ function(add_rocwmma_test TEST_TARGET TEST_SOURCE)
list(APPEND TEST_SOURCE ${ARGN})
add_executable(${TEST_TARGET} ${TEST_SOURCE})
- target_link_libraries(${TEST_TARGET} rocwmma gtest)
+ target_link_libraries(${TEST_TARGET} rocwmma gtest ${ROCM_SMI_DIR}/lib)
target_link_libraries(${TEST_TARGET} OpenMP::OpenMP_CXX "-L${HIP_CLANG_ROOT}/lib" "-Wl,-rpath=${HIP_CLANG_ROOT}/lib")
target_include_directories(${TEST_TARGET} PRIVATE
${CMAKE_CURRENT_SOURCE_DIR}
- ${ROCWMMA_TEST_INCLUDE_DIRS})
+ ${ROCWMMA_TEST_INCLUDE_DIRS}
+ ${ROCM_SMI_DIR}/include)
# Add support to include extended test coverage
if(ROCWMMA_BUILD_EXTENDED_TESTS)
--
2.39.3

View File

@ -83,6 +83,8 @@ class Rocwmma(CMakePackage):
for tgt in itertools.chain(["auto"], amdgpu_targets):
depends_on("rocblas amdgpu_target={0}".format(tgt), when="amdgpu_target={0}".format(tgt))
patch("0001-add-rocm-smi-lib-path-for-building-tests.patch", when="@5.6.0")
def setup_build_environment(self, env):
env.set("CXX", self.spec["hip"].hipcc)
@ -107,5 +109,9 @@ def cmake_args(self):
tgt = self.spec.variants["amdgpu_target"]
if "auto" not in tgt:
args.append(self.define_from_variant("AMDGPU_TARGETS", "amdgpu_target"))
if self.spec.satisfies("@5.6.0:"):
args.append(
self.define("ROCM_SMI_DIR", self.spec["rocm-smi-lib"].prefix)
)
return args