Merge branch 'develop_v2.10' into release_v2.10

e8c89225 · wenjh · 72295c6e · dc86f372 · e8c89225 · e8c89225
Commit e8c89225 authored Jan 07, 2026 by wenjh
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/cpp/CMakeLists.txt tests/cpp/CMakeLists.txt +1 -1

transformer_engine/common/CMakeLists.txt transformer_engine/common/CMakeLists.txt +1 -1

No files found.
--- a/tests/cpp/CMakeLists.txt
+++ b/tests/cpp/CMakeLists.txt
@@ -19,7 +19,7 @@ if(((EXISTS "/opt/dtk/") OR (EXISTS $ENV{ROCM_PATH})) AND NOT (EXISTS "/bin/nvcc
  add_definitions(-DNDEBUG)
  add_definitions(-DUSE_ROCM)
  if(NOT DEFINED ENV{NVTE_ROCM_ARCH})
-    SET(CMAKE_HIP_ARCHITECTURES gfx906;gfx926;gfx928;gfx936)
+    SET(CMAKE_HIP_ARCHITECTURES gfx906;gfx926;gfx928;gfx936;gfx938)
  else()
    SET(CMAKE_HIP_ARCHITECTURES $ENV{NVTE_ROCM_ARCH})
  endif()

--- a/transformer_engine/common/CMakeLists.txt
+++ b/transformer_engine/common/CMakeLists.txt
@@ -125,7 +125,7 @@ else()
  SET(CMAKE_CXX_COMPILER "${ROCM_PATH}/bin/hipcc")  
  
  if(NOT DEFINED ENV{NVTE_ROCM_ARCH})
-    SET(CMAKE_HIP_ARCHITECTURES gfx906;gfx926;gfx928;gfx936)
+    SET(CMAKE_HIP_ARCHITECTURES gfx906;gfx926;gfx928;gfx936;gfx938)
  else()
    SET(CMAKE_HIP_ARCHITECTURES $ENV{NVTE_ROCM_ARCH})
  endif()