fix version

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Sep 22, 2023 · 1a2d487 · 1a2d487
1 parent ee83aec
commit 1a2d487
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/source/lib/src/gpu/CMakeLists.txt b/source/lib/src/gpu/CMakeLists.txt
@@ -59,10 +59,10 @@ elseif(USE_ROCM_TOOLKIT)
   set(CMAKE_HIP_STANDARD 14)
   add_definitions("-DCUB_IGNORE_DEPRECATED_CPP_DIALECT")
 
-  message(STATUS "HIP major version is " ${HIP_VERSION_MAJOR})
+  message(STATUS "HIP major version is " ${hip_VERSION_MAJOR})
 
-  set(CMAKE_HIP_FLAGS -fno-gpu-rdc; ${CMAKE_HIP_FLAGS}) # --amdgpu-target=gfx906
-  if(HIP_VERSION VERSION_LESS 3.5.1)
+  set(CMAKE_HIP_FLAGS -fno-gpu-rdc ${CMAKE_HIP_FLAGS}) # --amdgpu-target=gfx906
+  if(hip_VERSION VERSION_LESS 3.5.1)
     set(CMAKE_HIP_FLAGS -hc; ${CMAKE_HIP_FLAGS})
   endif()