Skip to content
Merged
Show file tree
Hide file tree
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -176,20 +176,21 @@ if( BUILD_WITH_TENSILE )
set_property( CACHE Tensile_COMPILER PROPERTY STRINGS hcc hipcc)

include(virtualenv)

if (Tensile_TEST_LOCAL_PATH)
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" )
set( tensile_tag a472399c5e97ac13b171092c2eb61d87d3a08d53 CACHE STRING "Tensile tag to download" )
set( tensile_tag develop CACHE STRING "Tensile tag to download" )
virtualenv_install("git+https://github.com/${tensile_fork}/Tensile.git@${tensile_tag}")
message (STATUS "using GIT Tensile fork=${tensile_fork} from branch=${tensile_tag}")
endif()
list(APPEND CMAKE_PREFIX_PATH ${VIRTUALENV_HOME_DIR})
set( Tensile_ROOT "${VIRTUALENV_HOME_DIR}/bin" CACHE STRING "Local path of Tensile" )
set( Tensile_TensileConfig ${VIRTUALENV_HOME_DIR}/cmake/TensileConfig.cmake)
# set( Tensile_ROOT "${VIRTUALENV_HOME_DIR}/lib/python3.5/site-packages/Tensile" CACHE STRING "Local path of Tensile" )
find_package(Tensile REQUIRED)

# set( Tensile_TensileConfig ${VIRTUALENV_HOME_DIR}/cmake/TensileConfig.cmake)

endif()

Expand Down
10 changes: 2 additions & 8 deletions library/src/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -36,14 +36,9 @@ if( BUILD_WITH_TENSILE )
list( APPEND package_targets Tensile )
endif( )

# HACK: We include the config file directly because find_package(Tensile) is broken
#include( ${Tensile_TensileConfig} )
execute_process(COMMAND "${VIRTUALENV_HOME_DIR}/bin/TensileGetPath" OUTPUT_VARIABLE INSTALLED_TENSILE_PATH)
message(STATUS "TENSILE_PATH: ${INSTALLED_TENSILE_PATH}")
find_package(Tensile 4.11.0 EXACT REQUIRED HIP LLVM OpenMP PATHS "${INSTALLED_TENSILE_PATH}")

set( Tensile_RUNTIME_LANGUAGE "HIP" )
message( STATUS "AMDGPU_TARGETS=${AMDGPU_TARGETS}" )
set( ${CMAKE_CURRENT_SOURCE_DIR}/blas3/Tensile/Logic/${Tensile_LOGIC} )
TensileCreateLibraryCmake(
${CMAKE_CURRENT_SOURCE_DIR}/blas3/Tensile/Logic/${Tensile_LOGIC}
${Tensile_RUNTIME_LANGUAGE}
Expand All @@ -52,8 +47,7 @@ if( BUILD_WITH_TENSILE )
${Tensile_MERGE_FILES}
${Tensile_SHORT_FILENAMES}
${Tensile_PRINT_DEBUG}
Tensile_ROOT ${Tensile_ROOT}
)
)

# Create a unique name for Tensile compiled for rocBLAS
set_target_properties( Tensile PROPERTIES OUTPUT_NAME tensile-rocblas CXX_EXTENSIONS NO )
Expand Down