blob: be92505c18016a035673b46bcd5c369209740ff3 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
diff --git a/CMakeLists.txt b/CMakeLists.txt
index e98d74d..de8740e 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -55,7 +55,7 @@ macro(add_library NAME)
endif()
endmacro()
find_package(hip)
-
+set( Tensile_TEST_LOCAL_PATH "" CACHE PATH "Use local Tensile directory instead of fetching a GitHub branch" )
if(CMAKE_CXX_COMPILER MATCHES ".*/hcc$")
set(TENSILE_USE_LLVM ON CACHE BOOL "Use LLVM for parsing config files.")
set(TENSILE_USE_MSGPACK OFF CACHE BOOL "Use msgpack for parsing config files.")
@@ -79,11 +79,30 @@ endif()
# set(MIOPEN_TENSILE_SRC dev)
# endif()
# Use the virtual-env setup and download package from specified repo:
-virtualenv_install(wheel)
-virtualenv_install("git+https://github.com/ROCmSoftwarePlatform/Tensile.git@${MIOPEN_TENSILE_TAG}")
-list(APPEND CMAKE_PREFIX_PATH ${VIRTUALENV_HOME_DIR})
-find_package(Tensile REQUIRED COMPONENTS HIP)
+option(BUILD_USING_LOCAL_TENSILE "Build as a shared library" ON )
+
+if (WIN32)
+ set( Tensile_ROOT "${CMAKE_BINARY_DIR}/virtualenv/Lib/site-packages/Tensile" )
+endif()
+
+include(virtualenv)
+if (BUILD_USING_LOCAL_TENSILE)
+ virtualenv_install(${Tensile_TEST_LOCAL_PATH})
+ message (STATUS "using local Tensile from ${Tensile_TEST_LOCAL_PATH}, copied to ${Tensile_ROOT}")
+else()
+ # Use the virtual-env setup and download package from specified repot:
+ set( tensile_fork "ROCmSoftwarePlatform" CACHE STRING "Tensile fork to use" )
+ virtualenv_install("git+https://github.com/${tensile_fork}/Tensile.git@${MIOPEN_TENSILE_TAG}")
+ message (STATUS "using GIT Tensile fork=${tensile_fork} from branch=${MIOPEN_TENSILE_TAG}")
+endif()
+message(STATUS "Adding ${VIRTUALENV_HOME_DIR} to CMAKE_PREFIX_PATH")
+list(APPEND CMAKE_PREFIX_PATH ${VIRTUALENV_HOME_DIR})
+if (TENSILE_VERSION)
+ find_package(Tensile ${TENSILE_VERSION} EXACT REQUIRED HIP LLVM PATHS "${INSTALLED_TENSILE_PATH}")
+else()
+ find_package(Tensile 4.28.0 EXACT REQUIRED HIP LLVM PATHS "${INSTALLED_TENSILE_PATH}")
+endif()
set_target_properties( TensileHost PROPERTIES POSITION_INDEPENDENT_CODE ON )
|
