From d75c364864caa6f24b5785d70ed7a5f11ccd96e9 Mon Sep 17 00:00:00 2001
From: Bertan Dogancay <111835151+BertanDogancay@users.noreply.github.com>
Date: Mon, 29 Jan 2024 14:08:59 -0700
Subject: [PATCH] Do not use LL128 when disabled (#1066)

---
 cmake/Generator.cmake | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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}")