diff --git a/cmake/Generator.cmake b/cmake/Generator.cmake index 782dfb0980..a38cd80690 100644 --- a/cmake/Generator.cmake +++ b/cmake/Generator.cmake @@ -154,7 +154,7 @@ function(gen_device_table) foreach(func IN LISTS FUNC_LIST) string(FIND "${func}" "LL128" IS_LL128) if(NOT IS_LL128 EQUAL -1) - file(APPEND ${DEVICE_TABLE_H_FILE} "#if defined(__gfx90a__)\n") + file(APPEND ${DEVICE_TABLE_H_FILE} "#if defined(__gfx90a__) && defined(ENABLE_LL128)\n") endif() if(ENABLE_IFC) file(APPEND ${DEVICE_TABLE_H_FILE} "__device__ void ${func}();\n") @@ -173,7 +173,7 @@ function(gen_device_table) foreach(func ${FUNC_LIST}) string(FIND "${func}" "LL128" IS_LL128) if(NOT IS_LL128 EQUAL -1) - file(APPEND ${DEVICE_TABLE_H_FILE} "#if defined(__gfx90a__)\n") + file(APPEND ${DEVICE_TABLE_H_FILE} "#if defined(__gfx90a__) && defined(ENABLE_LL128)\n") file(APPEND ${DEVICE_TABLE_H_FILE} " ${func},\n") file(APPEND ${DEVICE_TABLE_H_FILE} "#else\n") string(REPLACE "LL128" "LL" func "${func}") @@ -375,7 +375,7 @@ function(gen_collectives) file(WRITE ${FILE_PATH} "#include \"${COLL_LOWER}.h\"\n#include \"common.h\"\n#include \"collectives.h\"\n") string(FIND "${list_name}" "LL128" IS_LL128) if(NOT IS_LL128 EQUAL -1) - file(APPEND ${FILE_PATH} "#if defined(__gfx90a__)\n") + file(APPEND ${FILE_PATH} "#if defined(__gfx90a__) && defined(ENABLE_LL128)\n") endif() foreach(IMPL IN LISTS IMPL_LIST) file(APPEND ${FILE_PATH} "${IMPL}")