File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -114,8 +114,6 @@ if(NOT __MAGMA_INCLUDED)
114
114
set (__MAGMA_EXTERN_PREFIX "${CMAKE_CURRENT_BINARY_DIR} /magma" )
115
115
set (__MAGMA_INSTALL_DIR "${PROJECT_SOURCE_DIR} /torch" )
116
116
117
- message (STATUS "MAGMA building for GPU_TARGETS=$ENV{PYTORCH_ROCM_ARCH} " )
118
-
119
117
cmake_host_system_information (RESULT N_LOGICAL_CORES QUERY NUMBER_OF_LOGICAL_CORES )
120
118
121
119
ExternalProject_Add (magma_external
@@ -127,9 +125,10 @@ if(NOT __MAGMA_INCLUDED)
127
125
CONFIGURE_COMMAND ${CMAKE_COMMAND} -E copy <SOURCE_DIR>/make.inc-examples/make.inc.hip-gcc-mkl <SOURCE_DIR>/make.inc
128
126
COMMAND ${CMAKE_COMMAND} -E chdir <SOURCE_DIR> make -f make.gen.hipMAGMA -j ${N_LOGICAL_CORES}
129
127
BUILD_COMMAND ${CMAKE_COMMAND} -E env MKLROOT=${MKLROOT}
130
- ${CMAKE_COMMAND} -E chdir <SOURCE_DIR> make lib/libmagma.so -j ${N_LOGICAL_CORES} MKLROOT=${MKLROOT} GPU_TARGET= " $ENV{PYTORCH_ROCM_ARCH} "
128
+ ${CMAKE_COMMAND} -E chdir <SOURCE_DIR> make lib/libmagma.so -j ${N_LOGICAL_CORES} MKLROOT=${MKLROOT} GPU_TARGET=$ENV{PYTORCH_ROCM_ARCH}
131
129
INSTALL_COMMAND ${CMAKE_COMMAND} -E copy <SOURCE_DIR>/lib/libmagma.so <INSTALL_DIR>/lib/
132
130
COMMAND ${CMAKE_COMMAND} -E copy_directory <SOURCE_DIR>/include <INSTALL_DIR>/include/magma
131
+ LIST_SEPARATOR ";" # Helps to avoid cmake splitting ENV{PYTORCH_ROCM_ARCH} in to spaces. Can be anything except spaces.
133
132
USES_TERMINAL_DOWNLOAD TRUE
134
133
USES_TERMINAL_CONFIGURE TRUE
135
134
USES_TERMINAL_BUILD TRUE
You can’t perform that action at this time.
0 commit comments